hex.genmodel.algos.deepwater.caffe

Class DeepwaterCaffeModel

java.lang.Object

hex.genmodel.algos.deepwater.caffe.DeepwaterCaffeModel

All Implemented Interfaces:

deepwater.backends.BackendModel

public class DeepwaterCaffeModel
extends java.lang.Object
implements deepwater.backends.BackendModel

Constructor Summary

Constructors

Constructor and Description
DeepwaterCaffeModel(int batch_size, int[] sizes, java.lang.String[] types, double[] dropout_ratios, long seed, boolean useGPU)
DeepwaterCaffeModel(java.lang.String graph, int[] input_shape, long seed, boolean useGPU)

Method Summary

Modifier and Type	Method and Description
void	loadParam(java.lang.String param_path)
float[]	predict(float[] data)
void	saveModel(java.lang.String model_path)
void	saveParam(java.lang.String param_path)
void	train(float[] data, float[] label)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DeepwaterCaffeModel

public DeepwaterCaffeModel(int batch_size,
                           int[] sizes,
                           java.lang.String[] types,
                           double[] dropout_ratios,
                           long seed,
                           boolean useGPU)

DeepwaterCaffeModel

public DeepwaterCaffeModel(java.lang.String graph,
                           int[] input_shape,
                           long seed,
                           boolean useGPU)

Method Detail

saveModel

public void saveModel(java.lang.String model_path)

saveParam

public void saveParam(java.lang.String param_path)

loadParam

public void loadParam(java.lang.String param_path)

train

public void train(float[] data,
                  float[] label)

predict

public float[] predict(float[] data)