diff --git a/src/Platform/Experiment.cc b/src/Platform/Experiment.cc index 480a9de..09674ab 100644 --- a/src/Platform/Experiment.cc +++ b/src/Platform/Experiment.cc @@ -181,11 +181,11 @@ namespace platform { item++; } cout << "end. " << flush; - delete fold; } result.setScoreTest(torch::mean(accuracy_test).item()).setScoreTrain(torch::mean(accuracy_train).item()); result.setScoreTestStd(torch::std(accuracy_test).item()).setScoreTrainStd(torch::std(accuracy_train).item()); result.setTrainTime(torch::mean(train_time).item()).setTestTime(torch::mean(test_time).item()); + result.setTestTimeStd(torch::std(test_time).item()).setTrainTimeStd(torch::std(train_time).item()); result.setNodes(torch::mean(nodes).item()).setLeaves(torch::mean(edges).item()).setDepth(torch::mean(num_states).item()); result.setDataset(fileName); addResult(result);