hex.genmodel.algos.deepwater.caffe

Class DeepwaterCaffeBackend

java.lang.Object

hex.genmodel.algos.deepwater.caffe.DeepwaterCaffeBackend

All Implemented Interfaces:

deepwater.backends.BackendTrain

public class DeepwaterCaffeBackend
extends java.lang.Object
implements deepwater.backends.BackendTrain

This backend forward requests to a docker images running the python Caffe interface. C.f h20-docker/caffe for more information.

Field Summary

Fields

Modifier and Type	Field and Description
static java.lang.String	CAFFE_DIR
static java.lang.String	CAFFE_H2O_DIR

Constructor Summary

Constructors

Constructor and Description
DeepwaterCaffeBackend()

Method Summary

Modifier and Type	Method and Description
deepwater.backends.BackendModel	buildNet(deepwater.datasets.ImageDataSet dataset, deepwater.backends.RuntimeOptions opts, deepwater.backends.BackendParams bparms, int num_classes, java.lang.String name)
void	delete(deepwater.backends.BackendModel m)
void	deleteSavedModel(java.lang.String model_path)
void	deleteSavedParam(java.lang.String param_path)
float[]	extractLayer(deepwater.backends.BackendModel m, java.lang.String name, float[] data)
java.lang.String	listAllLayers(deepwater.backends.BackendModel m)
float[]	loadMeanImage(deepwater.backends.BackendModel m, java.lang.String path)
void	loadParam(deepwater.backends.BackendModel m, java.lang.String param_path)
float[]	predict(deepwater.backends.BackendModel m, float[] data)
byte[]	readBytes(java.io.File file)
void	saveModel(deepwater.backends.BackendModel m, java.lang.String model_path)
void	saveParam(deepwater.backends.BackendModel m, java.lang.String param_path)
void	setParameter(deepwater.backends.BackendModel m, java.lang.String name, float value)
java.lang.String	toJson(deepwater.backends.BackendModel m)
float[]	train(deepwater.backends.BackendModel m, float[] data, float[] label)
void	writeBytes(java.io.File file, byte[] payload)

Methods inherited from class java.lang.Object

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

Field Detail

CAFFE_DIR

public static final java.lang.String CAFFE_DIR

See Also:

Constant Field Values

CAFFE_H2O_DIR

public static final java.lang.String CAFFE_H2O_DIR

See Also:

Constant Field Values

Constructor Detail

DeepwaterCaffeBackend

public DeepwaterCaffeBackend()

Method Detail

delete

public void delete(deepwater.backends.BackendModel m)

Specified by:

delete in interface deepwater.backends.BackendTrain

buildNet

public deepwater.backends.BackendModel buildNet(deepwater.datasets.ImageDataSet dataset,
                                                deepwater.backends.RuntimeOptions opts,
                                                deepwater.backends.BackendParams bparms,
                                                int num_classes,
                                                java.lang.String name)

Specified by:

buildNet in interface deepwater.backends.BackendTrain

saveModel

public void saveModel(deepwater.backends.BackendModel m,
                      java.lang.String model_path)

Specified by:

saveModel in interface deepwater.backends.BackendTrain

loadParam

public void loadParam(deepwater.backends.BackendModel m,
                      java.lang.String param_path)

Specified by:

loadParam in interface deepwater.backends.BackendTrain

saveParam

public void saveParam(deepwater.backends.BackendModel m,
                      java.lang.String param_path)

Specified by:

saveParam in interface deepwater.backends.BackendTrain

loadMeanImage

public float[] loadMeanImage(deepwater.backends.BackendModel m,
                             java.lang.String path)

Specified by:

loadMeanImage in interface deepwater.backends.BackendTrain

toJson

public java.lang.String toJson(deepwater.backends.BackendModel m)

Specified by:

toJson in interface deepwater.backends.BackendTrain

setParameter

public void setParameter(deepwater.backends.BackendModel m,
                         java.lang.String name,
                         float value)

Specified by:

setParameter in interface deepwater.backends.BackendTrain

train

public float[] train(deepwater.backends.BackendModel m,
                     float[] data,
                     float[] label)

Specified by:

train in interface deepwater.backends.BackendTrain

predict

public float[] predict(deepwater.backends.BackendModel m,
                       float[] data)

Specified by:

predict in interface deepwater.backends.BackendTrain

deleteSavedModel

public void deleteSavedModel(java.lang.String model_path)

Specified by:

deleteSavedModel in interface deepwater.backends.BackendTrain

deleteSavedParam

public void deleteSavedParam(java.lang.String param_path)

Specified by:

deleteSavedParam in interface deepwater.backends.BackendTrain

listAllLayers

public java.lang.String listAllLayers(deepwater.backends.BackendModel m)

Specified by:

listAllLayers in interface deepwater.backends.BackendTrain

extractLayer

public float[] extractLayer(deepwater.backends.BackendModel m,
                            java.lang.String name,
                            float[] data)

Specified by:

extractLayer in interface deepwater.backends.BackendTrain

writeBytes

public void writeBytes(java.io.File file,
                       byte[] payload)
                throws java.io.IOException

Specified by:

writeBytes in interface deepwater.backends.BackendTrain

Throws:

java.io.IOException

readBytes

public byte[] readBytes(java.io.File file)
                 throws java.io.IOException

Specified by:

readBytes in interface deepwater.backends.BackendTrain

Throws:

java.io.IOException