ilusha/Modules
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

NumRec Fixes

Browse source
This commit is contained in:
IlyaShurupov 2023-10-25 20:41:48 +03:00 • committed by Ilya Shurupov
parent 90ce453b60
commit 4804e51ad7
5 changed files with 181 additions and 148 deletions
Download patch file Download diff file Expand all files Collapse all files

8
DataAnalysis/public/FullyConnectedNN.hpp → DataAnalysis/public/FCNN.hpp
View file

@ -4,7 +4,8 @@
#include "DataAnalysisCommon.hpp"
namespace tp {
class FullyConnectedNN {
// Fully connected neural network
class FCNN {
struct Layer {
@ -32,12 +33,15 @@ namespace tp {
};
public:
FullyConnectedNN() = default;
FCNN() = default;
explicit FCNN(const Buffer<halni>& description);
void initializeRandom(const Buffer<halni>& description);
void evaluate(const Buffer<halnf>& input, Buffer<halnf>& output);
halnf calcCost(const Buffer<halnf>& output);
void clearGrad();
void calcGrad(const Buffer<halnf>& output);
void applyGrad(halnf step);