valueMLmodel.h

#include <iostream>
#include "RLDataset.h"
#include <math.h>
#include "LinearNet.h"
#include <torch/torch.h>

#ifndef VALUEMLMODEL_H
#define VALUEMLMODEL_H

using std::vector;

class valueMLmodel {

    public:
        valueMLmodel();
        valueMLmodel(unsigned int ds);

        torch::Tensor tensState;
        
        std::shared_ptr<LinearNet> linearNet;
         unsigned int dim;    

        void learn(RLDataset& dataSet, unsigned int numberOfEpochs, unsigned int batch_size, float alpha);
        
        vector<float> calcValueEstimates(state* s);

        torch::Tensor vectorToTensor(vector<vector<bool>>& vec);    

        vector<float> tensorToVector(torch::Tensor& tens);    
};
#endif