serializatioin / deserialization and tests
This commit is contained in:
@@ -2,6 +2,8 @@
|
||||
|
||||
#include "./ActivationFunction.h"
|
||||
|
||||
#include <cassert>
|
||||
|
||||
namespace NeuralNetwork {
|
||||
namespace ActivationFunction {
|
||||
|
||||
@@ -17,12 +19,17 @@ namespace ActivationFunction {
|
||||
return new Linear(lambda);
|
||||
}
|
||||
|
||||
virtual std::string stringify() const override {
|
||||
return "{ \"class\": \"NeuralNetwork::ActivationFunction::Linear\", \"lamba\" : "+std::to_string(lambda)+"}";
|
||||
virtual SimpleJSON::Type::Object serialize() const override {
|
||||
return {{"class", "NeuralNetwork::ActivationFunction::Linear"}, {"lambda", lambda}};
|
||||
}
|
||||
|
||||
static std::unique_ptr<Linear> deserialize(const SimpleJSON::Type::Object &obj) {
|
||||
return std::unique_ptr<Linear>(new Linear(obj["lambda"].as<double>()));
|
||||
}
|
||||
|
||||
protected:
|
||||
float lambda;
|
||||
NEURAL_NETWORK_REGISTER_ACTIVATION_FUNCTION(NeuralNetwork::ActivationFunction::Linear, Linear::deserialize)
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user