neuro_symbolic_toy_kb.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/disjunction_module.hpp"
#include "pulsatrix/linear_module.hpp"
#include "pulsatrix/negation_module.hpp"
#include "pulsatrix/satisfaction_loss.hpp"
#include "pulsatrix/sgd_optimizer.hpp"
#include "pulsatrix/tensor.hpp"
 
namespace pulsatrix {
 
class ToyKnowledgeBase {
public:
    explicit ToyKnowledgeBase(DeviceBackend* backend, float p = 2.0f);
 
    [[nodiscard]] float forward(const Tensor& x);
 
    void backward();
 
    float train_step(const Tensor& x, SGDOptimizer& optimizer);
 
    [[nodiscard]] const Tensor& last_a() const { return last_a_; }
 
    [[nodiscard]] const Tensor& last_b() const { return last_b_; }
 
    [[nodiscard]] const Tensor& last_rule() const { return last_rule_; }
 
    [[nodiscard]] LinearModule& predicate_a() { return linear_a_; }
 
    [[nodiscard]] LinearModule& predicate_b() { return linear_b_; }
 
    [[nodiscard]] NegationModule& negation_a() { return neg_a_; }
 
    [[nodiscard]] NegationModule& negation_b() { return neg_b_; }
 
    [[nodiscard]] DisjunctionModule& disjunction() { return disj_; }
 
private:
    DeviceBackend* backend_;
    LinearModule linear_a_;
    LinearModule linear_b_;
    NegationModule neg_a_;
    NegationModule neg_b_;
    DisjunctionModule disj_;
    SatisfactionLoss sat_loss_;
    Tensor last_a_;
    Tensor last_b_;
    Tensor last_rule_;
    bool has_forwarded_ = false;
};
 
}  // namespace pulsatrix