Create directory if missing when saving the model

Author: Jamie Matthews

NsfwSpy.Train/Program.cs

@@ -14,7 +14,7 @@ static void Main(string[] args)
             var projectDirectory = Path.GetFullPath(Path.Combine(AppContext.BaseDirectory, "../../../"));
             var workspaceRelativePath = Path.Combine(projectDirectory, "Workspace");
             var saveModelPath = Path.Combine(projectDirectory, "Models");
-            var assetsPath = @"D:\NsfwSpy\Images";
+            var assetsPath = @"E:\NsfwSpy\Images";
 
             var mlContext = new MLContext();
 
@@ -61,6 +61,9 @@ static void Main(string[] args)
 
             var trainedModel = trainingPipeline.Fit(trainSet);
 
+            if (!Directory.Exists(saveModelPath))
+                Directory.CreateDirectory(saveModelPath);
+
             saveModelPath = Path.Combine(saveModelPath, "NsfwSpyModel.zip");
             mlContext.Model.Save(trainedModel, trainSet.Schema, saveModelPath);