Revize 60a60164
Přidáno uživatelem Roman Kalivoda před téměř 4 roky(ů)
Server/ServerApp/Predictor/NaiveBayesClassifier.cs | ||
---|---|---|
40 | 40 |
{ |
41 | 41 |
this._trainingDataView = _mlContext.Data.LoadFromEnumerable(trainInput); |
42 | 42 |
var pipeline = _mlContext.Transforms.Conversion.MapValueToKey(nameof(ModelInput.Label)) |
43 |
.Append(_mlContext.Transforms.Concatenate("Features", new[] { "Temp" }))
|
|
43 |
.Append(_mlContext.Transforms.Concatenate("Features", new[] { nameof(ModelInput.Temp), nameof(ModelInput.Rain), nameof(ModelInput.Wind) }))
|
|
44 | 44 |
.Append(_mlContext.Transforms.NormalizeMinMax("Features", "Features")) |
45 | 45 |
.AppendCacheCheckpoint(_mlContext) |
46 | 46 |
.Append(_mlContext.MulticlassClassification.Trainers.NaiveBayes()) |
47 |
.Append(_mlContext.Transforms.Conversion.MapKeyToValue("PredictedLabel")); ;
|
|
47 |
.Append(_mlContext.Transforms.Conversion.MapKeyToValue(nameof(ModelOutput.PredictedLabel)));
|
|
48 | 48 |
|
49 | 49 |
this._trainedModel = pipeline.Fit(this._trainingDataView); |
50 | 50 |
this._predictionEngine = _mlContext.Model.CreatePredictionEngine<ModelInput, ModelOutput>(this._trainedModel); |
... | ... | |
53 | 53 |
|
54 | 54 |
public string Predict(ModelInput input) |
55 | 55 |
{ |
56 |
return this._predictionEngine.Predict(input).Prediction;
|
|
56 |
return this._predictionEngine.Predict(input).PredictedLabel;
|
|
57 | 57 |
} |
58 | 58 |
|
59 | 59 |
public void Evaluate(IEnumerable<ModelInput> modelInputs) |
60 | 60 |
{ |
61 | 61 |
var testDataView = this._mlContext.Data.LoadFromEnumerable(modelInputs); |
62 |
var testMetrics = _mlContext.MulticlassClassification.Evaluate(_trainedModel.Transform(testDataView)); |
|
62 |
var data = _trainedModel.Transform(testDataView); |
|
63 |
var testMetrics = _mlContext.MulticlassClassification.Evaluate(data); |
|
63 | 64 |
|
64 | 65 |
Console.WriteLine($"*************************************************************************************************************"); |
65 | 66 |
Console.WriteLine($"* Metrics for Multi-class Classification model - Test Data "); |
Také k dispozici: Unified diff
Re #8955 refactoring