rmontanana/BayesNet
rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 3 Wiki Activity

Add info to output

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2023-11-22 16:38:50 +01:00
parent c2eb727fc7
commit bbe5302ab1
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/Platform/GridSearch.cc
View File

@ -54,19 +54,18 @@ namespace platform {
for (int nfold = 0; nfold < config.n_folds; nfold++) {
auto clf = Models::instance()->create(config.model);
auto [train, test] = fold->getFold(nfold);
// auto train_t = torch::tensor(train);
// auto test_t = torch::tensor(test);
// auto X_train = X.index({ "...", train_t });
// auto y_train = y.index({ train_t });
// auto X_test = X.index({ "...", test_t });
// auto y_test = y.index({ test_t });
auto train_t = torch::tensor(train);
auto test_t = torch::tensor(test);
auto X_train = X.index({ "...", train_t });
auto y_train = y.index({ train_t });
auto X_test = X.index({ "...", test_t });
auto y_test = y.index({ test_t });
showProgressFold(nfold + 1, getColor(clf->getStatus()), "a");
// Train model
// clf->fit(X_train, y_train, features, className, states);
showProgressFold(nfold + 1, getColor(clf->getStatus()), "b");
showProgressFold(nfold + 1, getColor(clf->getStatus()), "c");
sleep(1);
std::cout << "\b\b\b, " << flush;
std::cout << "\b\b\b, \b" << flush;
}
delete fold;
}
@ -89,7 +88,7 @@ namespace platform {
auto hyperparameters = platform::HyperParameters(datasets.getNames(), hyperparam_line);
processFile(dataset, datasets, hyperparameters);
}
std::cout << std::endl;
std::cout << "end." << std::endl;
}
// Save results
save();