apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Ansh Babbar
2023-11-29 02:26:37 +05:30
parent a602816ce7
commit 48456f2ca3
1 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/logistic_regression/logistic_regression_main.cpp
+15 -15
View File
@@ -309,6 +309,21 @@ void BINDING_FUNCTION(util::Params& params, util::Timers& timers)
{
model->Lambda() = lambda;
// Did we want training accuracy?
if (params.Has("print_training_accuracy"))
{
timers.Start("lr_prediction");
arma::Row<size_t> predictions;
model->Classify(regressors, predictions);
const size_t correct = arma::accu(predictions == responses);
Log::Info << correct << " of " << responses.n_elem << " correct on training"
<< " set (" << (double(correct) / double(responses.n_elem) * 100) << ")."
<< endl;
timers.Stop("lr_prediction");
}
if (optimizerType == "sgd")
{
ens::SGD<> sgdOpt;
@@ -336,21 +351,6 @@ void BINDING_FUNCTION(util::Params& params, util::Timers& timers)
timers.Stop("logistic_regression_optimization");
}
}
// Did we want training accuracy?
if (params.Has("training") && params.Has("print_training_accuracy"))
{
timers.Start("lr_prediction");
arma::Row<size_t> predictions;
model->Classify(regressors, predictions);
const size_t correct = arma::accu(predictions == responses);
Log::Info << correct << " of " << responses.n_elem << " correct on training"
<< " set (" << (double(correct) / double(responses.n_elem) * 100) << ")."
<< endl;
timers.Stop("lr_prediction");
}
if (params.Has("test"))
{