Exemple #1
0
        public void SaveModelAsFile()
        {
            string       getPath        = Path.GetFullPath(Path.Combine(Environment.CurrentDirectory, @"..\..\..\"));
            string       _trainDataPath = getPath + "\\GoDiet\\DietData\\DataModel\\trainData\\weight_lr_train1.csv";
            MLContext    mlContext      = new MLContext(seed: 0);
            ITransformer model          = BinaryClassML.Train(mlContext, _trainDataPath);

            BinaryClassML.SaveModelAsFile(mlContext, model);
            Assert.AreEqual("Saved", BinaryClassML.check);
        }
Exemple #2
0
 public void SaveModelAsFileExceptionThrownTest()
 {
     try
     {
         string       getPath        = Path.GetFullPath(Path.Combine(Environment.CurrentDirectory, @"..\..\..\"));
         string       _trainDataPath = getPath + "\\DietData\\DataModel\\trainData\\weight_lr_train1.csv";
         MLContext    mlContext      = new MLContext(seed: 0);
         ITransformer model          = BinaryClassML.Train(mlContext, _trainDataPath);
         BinaryClassML.SaveModelAsFile(mlContext, model);
         Assert.Fail();
     }
     catch (Exception) { }
 }