Projekt

Obecné

Profil

« Předchozí | Další » 

Revize 60a60164

Přidáno uživatelem Roman Kalivoda před téměř 4 roky(ů)

Re #8955 refactoring

Zobrazit rozdíly:

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