public static void Execute() { Print.Header("Plot Regression Chart"); if (!Validate.DataIsLoaded() || !Validate.ModelIsTrained()) { return; } var predictions = SalaryPredictionService.GetPrediction(Program.TrainedModel, Program.Data.Take(100)).ToList(); const string fileName = "chart.png"; Plot.RegressionChart(predictions, fileName); Plot.ShowImage(fileName); }
public static void Execute() { Print.Header("Train Model"); if (!Validate.DataIsLoaded()) { return; } SplitData(Program.Data, out var trainingData, out var validationData); IEnumerable <string> previewData = null; Program.TrainedModel = ConsoleSpinner.Execute("Training", () => SalaryPredictionService.Train(trainingData, out previewData)); Print.PreviewTransformedData(previewData); var metrics = ConsoleSpinner.Execute("Evaluating", () => SalaryPredictionService.Evaluate(Program.TrainedModel, validationData)); Print.Metrics(metrics); }
public static void Execute() { Print.Header("Get Salary Prediction"); if (!Validate.DataIsLoaded() || !Validate.ModelIsTrained()) { return; } var age = ConsoleHelper.GetNumber("Enter Age of Employee:"); ConsoleHelper.WriteLine("Select ExperienceLevel of Employee", ConsoleColor.Cyan); var experienceLevel = ConsoleHelper.PickOption(ExperienceLevel.Values); var employee = new Employee(age, experienceLevel, 0); Print.Employee(employee); var prediction = SalaryPredictionService.GetPrediction(Program.TrainedModel, employee); Print.PredictedSalary(prediction); }