Solve voting vector error

This commit is contained in:
2023-09-02 13:58:12 +02:00
parent 7806f961e2
commit 182b07ed90
11 changed files with 45 additions and 60 deletions

View File

@@ -182,6 +182,7 @@ namespace platform {
clf.reset();
}
cout << "end. " << flush;
delete fold;
}
result.setScoreTest(torch::mean(accuracy_test).item<double>()).setScoreTrain(torch::mean(accuracy_train).item<double>());
result.setScoreTestStd(torch::std(accuracy_test).item<double>()).setScoreTrainStd(torch::std(accuracy_train).item<double>());