Refactor Smoothing type to its own file

Add log to boost
This commit is contained in:
2025-03-08 14:04:08 +01:00
parent 81fd7df7f0
commit b987dcbcc4
9 changed files with 41 additions and 23 deletions

View File

@@ -46,8 +46,6 @@ namespace bayesnet {
std::string className;
std::map<std::string, std::vector<int>> states;
torch::Tensor dataset; // (n+1)xm tensor
status_t status = NORMAL;
std::vector<std::string> notes; // Used to store messages occurred during the fit process
void checkFitParameters();
virtual void buildModel(const torch::Tensor& weights) = 0;
void trainModel(const torch::Tensor& weights, const Smoothing_t smoothing) override;