Solved Ld poor results

This commit is contained in:
2023-08-12 11:49:18 +02:00
parent 3a85481a5a
commit 405887f833
7 changed files with 13 additions and 9 deletions

View File

@@ -11,7 +11,7 @@ namespace bayesnet {
Xf = X_;
y = y_;
// Fills vectors Xv & yv with the data from tensors X_ (discretized) & y
fit_local_discretization(states, y);
states = fit_local_discretization(y);
// We have discretized the input data
// 1st we need to fit the model to build the normal SPODE structure, SPODE::fit initializes the base Bayesian network
SPODE::fit(dataset, features, className, states);
@@ -27,7 +27,7 @@ namespace bayesnet {
features = features_;
className = className_;
// Fills vectors Xv & yv with the data from tensors X_ (discretized) & y
fit_local_discretization(states, y);
states = fit_local_discretization(y);
// We have discretized the input data
// 1st we need to fit the model to build the normal SPODE structure, SPODE::fit initializes the base Bayesian network
SPODE::fit(dataset, features, className, states);