xor_training_example.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "pulsatrix/adam_optimizer.hpp"
#include "pulsatrix/linear_module.hpp"
#include "pulsatrix/metrics_sink.hpp"
#include "pulsatrix/mse_loss.hpp"
#include "pulsatrix/relu_module.hpp"
 
namespace pulsatrix {
 
class XorNetwork {
public:
    explicit XorNetwork(DeviceBackend* backend);
 
    [[nodiscard]] Tensor forward(const Tensor& input);
 
    float train_step(const Tensor& input, const Tensor& target, AdamOptimizer& optimizer, MetricsSink& sink,
                      int step);
 
    [[nodiscard]] const Tensor& linear1_weight() const { return linear1_.weight(); }
 
    [[nodiscard]] LinearModule& linear1() { return linear1_; }
    [[nodiscard]] LinearModule& linear2() { return linear2_; }
 
private:
    DeviceBackend* backend_;
    LinearModule linear1_;
    ReluModule relu_;
    LinearModule linear2_;
    MSELoss loss_;
};
 
}  // namespace pulsatrix