mirror of
https://github.com/opencv/opencv.git
synced 2026-07-29 23:33:05 +04:00
ml: add checks of empty train data
This commit is contained in:
@@ -920,6 +920,7 @@ public:
|
||||
|
||||
bool train( const Ptr<TrainData>& trainData, int flags ) CV_OVERRIDE
|
||||
{
|
||||
CV_Assert(!trainData.empty());
|
||||
const int MAX_ITER = 1000;
|
||||
const double DEFAULT_EPSILON = FLT_EPSILON;
|
||||
|
||||
@@ -955,6 +956,7 @@ public:
|
||||
}
|
||||
int train_anneal(const Ptr<TrainData>& trainData)
|
||||
{
|
||||
CV_Assert(!trainData.empty());
|
||||
SimulatedAnnealingANN_MLP s(*this, trainData);
|
||||
trained = true; // Enable call to CalcError
|
||||
int iter = simulatedAnnealingSolver(s, params.initialT, params.finalT, params.coolingRatio, params.itePerStep, NULL, params.rEnergy);
|
||||
|
||||
Reference in New Issue
Block a user