public static double CalcMSE(XGBRegressor regr) { var target = GetKlaThickness(LoadedDatas); var pred = regr.Predict(GetReflectivity(LoadedDatas)); return(MSE(pred, target)); }
// 나중에는 세팅 , 데이터 , 로드 여부 순으로 모델 만들기 public static XGBRegressor CreateModel(List <IpsDataSet> datas) { LoadedDatas = datas; Regr = new XGBRegressor(); Regr.Fit(GetReflectivity(datas), GetKlaThickness(datas)); return(Regr); }
[TestMethod] public void Predict() { var dataTrain = TestUtils.GetRegressorDataTrain(); var labelsTrain = TestUtils.GetRegressorLabelsTrain(); var dataTest = TestUtils.GetRegressorDataTest(); var xgbr = new XGBRegressor(); xgbr.Fit(dataTrain, labelsTrain); var preds = xgbr.Predict(dataTest); Assert.IsTrue(TestUtils.RegressorPredsCorrect(preds)); }
public void TestRegressorSaveAndLoadWithParameters() { var dataTrain = TestUtils.GetRegressorDataTrain(); var labelsTrain = TestUtils.GetRegressorLabelsTrain(); var dataTest = TestUtils.GetRegressorDataTest(); var xgbr = new XGBRegressor(); xgbr.Fit(dataTrain, labelsTrain); var preds1 = xgbr.Predict(dataTest); xgbr.SaveModelToFile(TEST_FILE); var xgbr2 = BaseXgbModel.LoadRegressorFromFile(TEST_FILE); var preds2 = xgbr2.Predict(dataTest); Assert.IsTrue(TestUtils.AreEqual(preds1, preds2)); }
public static XGBRegressor LoadModel(string path) // Excuted when ScanAutorun is fired { Regr = LoadRegressorFromFile(path); return(Regr); }
public static void Reset() { Regr = new XGBRegressor(); LoadedDatas = new List <IpsDataSet>(); }