nnlib
GPU-accelerated, C/C++ neural network library.
|
This is the complete list of members for Network, including all inherited members.
add(size_t numNeurons, const std::string &activation="linear") | Network | |
forward(const sTensor &batch) | Network | |
layers | Network | private |
location | Network | private |
Network(size_t inputSize, bool useGPU=true, long long seed=NO_SEED) | Network | explicit |
previousSize | Network | private |
processEpoch(std::vector< sTensor > &batches, std::vector< sTensor > &targets, std::vector< sTensor > &targetsOnHost, float learningRate, Loss *loss, std::vector< Metric * > &metrics) | Network | private |
seed | Network | private |
train(sTensor &X, sTensor &y, int epochs, size_t batchSize, float learningRate, Loss *loss, std::vector< Metric * > &metrics) | Network |