Implement 3 types of smoothing
This commit is contained in:
@@ -37,6 +37,7 @@ namespace bayesnet {
|
||||
void AODELd::trainModel(const torch::Tensor& weights)
|
||||
{
|
||||
for (const auto& model : models) {
|
||||
model->setSmoothing(smoothing);
|
||||
model->fit(Xf, y, features, className, states);
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user