public class ConvolutionalNetwork extends NeuralNetwork<BackpropagationTrainer> implements Serializable
ConvolutionalLayer
,
MaxPoolingLayer
,
BackpropagationTrainer
,
Serialized FormModifier and Type | Class and Description |
---|---|
static class |
ConvolutionalNetwork.Builder
Builder for a convolutional neural network.
|
Modifier and Type | Method and Description |
---|---|
static ConvolutionalNetwork.Builder |
builder()
Returns a builder for the
ConvolutionalNetwork |
List<Tensor> |
getDeltaWeights()
Returns delta weights for all layers.
|
List<Tensor> |
getLayersOutputs()
Returns outputs of all layers.
|
List<Tensor> |
getWeights()
Returns weights from all layers in this network as a list of tensors.
|
void |
setInput(Tensor input)
Sets network input and calculates entire network (triggers forward pass).
|
void |
setWeights(List<String> weights)
Sets network's weights for all layers.
|
applyWeightChanges, backward, getInputLayer, getL1RegSum, getL2RegSum, getLabel, getLayers, getLossFunction, getNormalizer, getOutput, getOutputLabel, getOutputLabels, getOutputLayer, getPreprocessing, getTrainer, load, predict, save, setLabel, setLossFunction, setNormalizer, setOutputError, setOutputLabels, setPreprocessing, setTrainer, test, toString, train
public void setInput(Tensor input)
NeuralNetwork
setInput
in class NeuralNetwork<BackpropagationTrainer>
input
- input tensorpublic static ConvolutionalNetwork.Builder builder()
ConvolutionalNetwork
public List<Tensor> getWeights()
public void setWeights(List<String> weights)
weights
- List of weights for all layers.public List<Tensor> getDeltaWeights()
Copyright © 2022. All rights reserved.