Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Call stop_train after training.

  • Loading branch information...
commit 57bcf5b3348923f8099db8bf0f17a1ecc494a3d6 1 parent f2fb797
@pluskid pluskid authored
Showing with 6 additions and 1 deletion.
  1. +6 −1 src/shogun/multiclass/tree/ConditionalProbabilityTree.cpp
View
7 src/shogun/multiclass/tree/ConditionalProbabilityTree.cpp
@@ -130,7 +130,12 @@ bool CConditionalProbabilityTree::train_machine(CFeatures* data)
}
m_feats->end_parser();
- // TODO: call stop_train for each sub-machine
+ for (int32_t i=0; i < m_machines->get_num_elements(); ++i)
+ {
+ COnlineLibLinear *lll = dynamic_cast<COnlineLibLinear *>(m_machines->get_element(i));
+ lll->stop_train();
+ SG_UNREF(lll);
+ }
return true;
}
Please sign in to comment.
Something went wrong with that request. Please try again.