public void PredictProba()
{
var dataTrain = TestUtils.GetClassifierDataTrain();
var labelsTrain = TestUtils.GetClassifierLabelsTrain();
var dataTest = TestUtils.GetClassifierDataTest();
var xgbc = new XGBClassifier();
xgbc.Fit(dataTrain, labelsTrain);
var preds = xgbc.PredictProba(dataTest);
Assert.IsTrue(TestUtils.ClassifierPredsProbaCorrect(preds));
}
public void TestClassifierDump()
{
var dataTrain = TestUtils.GetClassifierDataTrain();
var labelsTrain = TestUtils.GetClassifierLabelsTrain();
var dataTest = TestUtils.GetClassifierDataTest();
var xgbc = new XGBClassifier();
xgbc.Fit(dataTrain, labelsTrain);
var preds1 = xgbc.PredictProba(dataTest);
var description = xgbc.DumpModelEx();
Console.WriteLine("Model Dumped: " + description);
}
public void TestClassifierSaveAndLoadWithParameters()
{
var dataTrain = TestUtils.GetClassifierDataTrain();
var labelsTrain = TestUtils.GetClassifierLabelsTrain();
var dataTest = TestUtils.GetClassifierDataTest();
var xgbc = new XGBClassifier(10, 0.01f, 50);
xgbc.Fit(dataTrain, labelsTrain);
var preds1 = xgbc.PredictProba(dataTest);
xgbc.SaveModelToFile(TEST_FILE);
var xgbc2 = BaseXgbModel.LoadClassifierFromFile(TEST_FILE);
var preds2 = xgbc2.PredictProba(dataTest);
Assert.IsTrue(TestUtils.AreEqual(preds1, preds2));
}
