首页 > 解决方案 > 我的 ML.Net 推荐算法对 R 平方和均方根值都返回 0

问题描述

我是机器学习的新手,但我有 C# 方面的经验,这就是为什么我想选择 Ml.net 作为让自己熟悉该主题的一种方式。

当我掌握这个概念时,我决定制作一个啤酒推荐模型,它将接受以前的评论并推荐一种新啤酒。

无论如何,当我运行程序时,下面 EvaluateModel 中返​​回的值每个都返回 0。我会喜欢另一双眼睛。谢谢!

这是训练模型

public static ITransformer BuildAndTrainModel(MLContext mlContext, IDataView trainingDataView)
        {
            IEstimator<ITransformer> estimator = mlContext.Transforms.Categorical.OneHotEncoding(outputColumnName: "abvEncoded", inputColumnName: "abv")
                                .Append(mlContext.Transforms.Categorical.OneHotEncoding(outputColumnName: "ibuEncoded", inputColumnName: "ibu"))
                                .Append(mlContext.Transforms.Conversion.MapValueToKey(outputColumnName: "beerIDEncoded", inputColumnName: "beerID"))
                                .Append(mlContext.Transforms.Categorical.OneHotEncoding(outputColumnName: "nameEncoded", inputColumnName: "name"))
                                .Append(mlContext.Transforms.Categorical.OneHotEncoding(outputColumnName: "styleEncoded", inputColumnName: "style"))
                                .Append(mlContext.Transforms.Categorical.OneHotEncoding(outputColumnName: "breweryIDEncoded", inputColumnName: "breweryID"))
                                .Append(mlContext.Transforms.Conversion.MapValueToKey(outputColumnName: "userIDEncoded", inputColumnName: "userID"))
                                .Append(mlContext.Transforms.Concatenate("Features", "abvEncoded", "ibuEncoded", "styleEncoded"));

                        var options = new MatrixFactorizationTrainer.Options
            {
                MatrixColumnIndexColumnName = "userIDEncoded",
                MatrixRowIndexColumnName = "beerIDEncoded",
                LabelColumnName = "Label",
                NumberOfIterations = 20,
                ApproximationRank = 100
            };

            var trainerEstimator = estimator.Append(mlContext.Recommendation().Trainers.MatrixFactorization(options));

            Console.WriteLine("=============== Training the model ===============");
            ITransformer model = trainerEstimator.Fit(trainingDataView);

            return model;

这是评估模型

public static void EvaluateModel(MLContext mlContext, IDataView testDataView, ITransformer model)
        {
            Console.WriteLine("=============== Evaluating the model ===============");
            var prediction = model.Transform(testDataView);

            var metrics = mlContext.Regression.Evaluate(prediction, labelColumnName: "Label", scoreColumnName: "Score");

            Console.WriteLine("Root Mean Squared Error : " + metrics.RootMeanSquaredError.ToString());
            Console.WriteLine("RSquared: " + metrics.RSquared.ToString());
        }
        public static void UseModelForSinglePrediction(MLContext mlContext, ITransformer model)
        {
            Console.WriteLine("=============== Making a prediction ===============");
            var predictionEngine = mlContext.Model.CreatePredictionEngine<Beers, BeerPrediction>(model);

            //Predict
            var beerSample = new Beers()
            {
                userID = 1,
                abv = ".08",
                ibu = "120",
                beerID = 379,
                name = "Heady Topper",
                style = "American",
                breweryID = 272,
                ounces = 16
             };
            var beerRatingPrediction = predictionEngine.Predict(beerSample);

            if (Math.Round(beerRatingPrediction.Score, 1) > 3.5)
            {
                Console.WriteLine("Beer " + beerSample.beerID + " is recommended for user " + beerSample.userID);
            }
            else
            {
                Console.WriteLine("Beer " + beerSample.beerID + " is not recommended for user " + beerSample.userID);
            }
        }

主要的

 static void Main(string[] args)
        {
            MLContext mLContext = new MLContext();

            (IDataView trainingDataView, IDataView testDataView) = LoadData(mLContext);

            ITransformer model = BuildAndTrainModel(mLContext, trainingDataView);

            EvaluateModel(mLContext, testDataView, model);

            UseModelForSinglePrediction(mLContext, model);

            SaveModel(mLContext, trainingDataView.Schema, model);

            Console.Read();

啤酒的对象等级

    class Beers
    {
        [LoadColumn(0)]
        public float userID { get; set; }

        [LoadColumn(1)]
        public string abv { get; set; }

        [LoadColumn(2)]
        public string ibu { get; set; }

        [LoadColumn(3)]
        public float beerID { get; set; }

        [LoadColumn(4)]
        public string name { get; set; }

        [LoadColumn(5)]
        public string style { get; set; }

        [LoadColumn(6)]
        public float breweryID { get; set; }

        [LoadColumn(7)]
        public float ounces { get; set; }

        [LoadColumn(8), ColumnName("Label")]
        public float rating { get; set; }




    }
}

标签: c#ml.net

解决方案


推荐阅读