public ResultExperimentClassification ExperimentClassification(object a)
{
InfaForExperimentClassification infa = (InfaForExperimentClassification)a;
ResultExperimentClassification result = new ResultExperimentClassification();
result.timeTrain = new Stopwatch();
result.timeTest = new Stopwatch();
result.timeСlassification = new Stopwatch();
RandomForest RF = new RandomForest();
RF.setNumIterations(100); //Количество деревьев
result.timeTrain.Start();
RF.buildClassifier(infa.trainDataSet); //Обучние
result.timeTrain.Stop();
result.timeTest.Start();
result.evaluation = evaluateModel(RF, infa.trainDataSet, infa.testDataSet); //Тестирование
result.timeTest.Stop();
result.timeСlassification.Start();
RF.classifyInstance(infa.testDataSet.instance(0)); //Замер времени классификации 1 экземпляра
result.timeСlassification.Stop();
return(result);
}
public RandomForest TrainClassification(object a)
{
InfaForExperimentClassification infa = (InfaForExperimentClassification)a;
RandomForest RF = new RandomForest();
RF.setNumIterations(100); //Количество деревьев
RF.buildClassifier(infa.trainDataSet); //Обучние
SerializationHelper.write(infa.fileNameModel, RF); //Сохранение модели
return(RF);
}
