This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
static void Main(string[] args) | |
{ | |
var trainingDataLocation = @"Data/winequality_white_train.csv"; | |
var testDataLocation = @"Data/winequality_white_test.csv"; | |
var modelEvaluator = new ModelEvaluator(); | |
var perceptronBinaryModel = new ModelBuilder(trainingDataLocation, new AveragedPerceptronBinaryClassifier()).BuildAndTrain(); | |
var perceptronBinaryMetrics = modelEvaluator.Evaluate(perceptronBinaryModel, testDataLocation); | |
PrintMetrics("Perceptron", perceptronBinaryMetrics); | |
var fastForestBinaryModel = new ModelBuilder(trainingDataLocation, new FastForestBinaryClassifier()).BuildAndTrain(); | |
var fastForestBinaryMetrics = modelEvaluator.Evaluate(fastForestBinaryModel, testDataLocation); | |
PrintMetrics("Fast Forest Binary", fastForestBinaryMetrics); | |
var fastTreeBinaryModel = new ModelBuilder(trainingDataLocation, new FastTreeBinaryClassifier()).BuildAndTrain(); | |
var fastTreeBinaryMetrics = modelEvaluator.Evaluate(fastTreeBinaryModel, testDataLocation); | |
PrintMetrics("Fast Tree Binary", fastTreeBinaryMetrics); | |
var linearSvmModel = new ModelBuilder(trainingDataLocation, new LinearSvmBinaryClassifier()).BuildAndTrain(); | |
var linearSvmMetrics = modelEvaluator.Evaluate(linearSvmModel, testDataLocation); | |
PrintMetrics("Linear SVM", linearSvmMetrics); | |
var logisticRegressionModel = new ModelBuilder(trainingDataLocation, new LogisticRegressionBinaryClassifier()).BuildAndTrain(); | |
var logisticRegressionMetrics = modelEvaluator.Evaluate(logisticRegressionModel, testDataLocation); | |
PrintMetrics("Logistic Regression Binary", logisticRegressionMetrics); | |
var sdcabModel = new ModelBuilder(trainingDataLocation, new StochasticDualCoordinateAscentBinaryClassifier()).BuildAndTrain(); | |
var sdcabMetrics = modelEvaluator.Evaluate(sdcabModel, testDataLocation); | |
PrintMetrics("Stochastic Dual Coordinate Ascent Binary", logisticRegressionMetrics); | |
VisualizeTenPredictionsForTheModel(fastForestBinaryModel, testDataLocation); | |
Console.ReadLine(); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
You're printing the logisticRegressionMetrics for the Stochastic Dual Coordinate Ascent Binary Classifier results