Fix comment in AODELd

This commit is contained in:
2024-12-10 13:35:23 +01:00
parent e6501502d1
commit 43ceefd2c9

View File

@@ -20,7 +20,8 @@ namespace bayesnet {
// Fills std::vectors Xv & yv with the data from tensors X_ (discretized) & y // Fills std::vectors Xv & yv with the data from tensors X_ (discretized) & y
states = fit_local_discretization(y); states = fit_local_discretization(y);
// We have discretized the input data // We have discretized the input data
// 1st we need to fit the model to build the normal TAN structure, TAN::fit initializes the base Bayesian network // 1st we need to fit the model to build the normal AODE structure, Ensemble::fit
// calls buildModel to initialize the base models
Ensemble::fit(dataset, features, className, states, smoothing); Ensemble::fit(dataset, features, className, states, smoothing);
return *this; return *this;