public class MLDeeplearningModel extends Object implements Externalizable
Constructor and Description |
---|
MLDeeplearningModel() |
MLDeeplearningModel(hex.deeplearning.DeepLearningModel dlModel) |
Modifier and Type | Method and Description |
---|---|
hex.deeplearning.DeepLearningModel |
getDeepLearningModel()
Returns the model
|
String |
getURIStringForLocation(String loc) |
double[] |
predict(water.fvec.Frame inputs) |
double |
predict(org.apache.spark.mllib.linalg.Vector input)
Predicts the label of a given input
|
void |
readExternal(ObjectInput in) |
void |
setDeepLearningModel(hex.deeplearning.DeepLearningModel model)
Set the model
|
void |
setStorageLocation(String location) |
void |
writeExternal(ObjectOutput out) |
public MLDeeplearningModel()
public MLDeeplearningModel(hex.deeplearning.DeepLearningModel dlModel)
public void setStorageLocation(String location)
public void setDeepLearningModel(hex.deeplearning.DeepLearningModel model)
model
- modelpublic hex.deeplearning.DeepLearningModel getDeepLearningModel()
public double predict(org.apache.spark.mllib.linalg.Vector input)
input
- input to predict as a vectorpublic double[] predict(water.fvec.Frame inputs)
public void writeExternal(ObjectOutput out) throws IOException
writeExternal
in interface Externalizable
IOException
public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException
readExternal
in interface Externalizable
IOException
ClassNotFoundException
Copyright © 2015 WSO2, Inc.. All Rights Reserved.