Starting on the model.h, but need to make layer structs and structs for loss and optimizers
This commit is contained in:
@@ -4,13 +4,15 @@
|
||||
|
||||
#include "utils/vector.h"
|
||||
#include "utils/matrix.h"
|
||||
#include "modules/neural_networks/layers/Layer.h"
|
||||
|
||||
#include "random/random.h"
|
||||
|
||||
|
||||
namespace neural_networks{
|
||||
|
||||
template <typename T>
|
||||
struct Dropout_Layer{
|
||||
struct Dropout_Layer : Layer<T>{
|
||||
|
||||
// Store rate, we invert it as for example for dropout
|
||||
// of 0.1 we need a success rate of 0.9
|
||||
|
||||
Reference in New Issue
Block a user