1 using Microsoft.ML.Runtime.Api;
 2
 3 namespace IrisClustering
 4 {
 5     public class IrisData
 6     {
 7         [Column("0")]
 8         public float SepalLength;
 9
10         [Column("1")]
11         public float SepalWidth;
12
13         [Column("2")]
14         public float PetalLength;
15
16         [Column("3")]
17         public float PetalWidth;
18     }
19
20     public class ClusterPrediction
21     {
22         [ColumnName("PredictedLabel")]
23         public uint PredictedClusterId;
24
25         [ColumnName("Score")]
26         public float[] Distances;
27     }
28 }

 1 using System;
 2 using System.IO;
 3 using System.Threading.Tasks;
 4 using Microsoft.ML.Legacy;
 5 using Microsoft.ML.Legacy.Data;
 6 using Microsoft.ML.Legacy.Trainers;
 7 using Microsoft.ML.Legacy.Transforms;
 8
 9 namespace IrisClustering
10 {
11     public static class Program
12     {
13         //数据文件路径
14         static readonly string _dataPath = Path.Combine(Environment.CurrentDirectory, "Data", "iris_data.txt");
15         //预测模型保存路径
16         static readonly string _modelPath = Path.Combine(Environment.CurrentDirectory, "Data", "IrisClusteringModel.zip");
17         //预测模型训练方法
18         private static PredictionModel<IrisData, ClusterPrediction> Train()
19        {
20             var pipeline = new LearningPipeline
21             {
22                 new TextLoader(_dataPath).CreateFrom<IrisData>(separator: ','), //加载数据
23                 new ColumnConcatenator( //指定数据列名
24                 "Features",      //特征 学习算法将通过ColumnConcatenator类将数据转化为特征列的值
25                 "SepalLength",   //花萼长度
26                 "SepalWidth",    //花萼宽度
27                 "PetalLength",   //花瓣长度
28                 "PetalWidth"),   //花瓣宽度
29                 new KMeansPlusPlusClusterer() { K = 3 }  //指定K-Means算法，这里因为数据中已经给出三个分类，所以我们指定K值为3
30             };
31             var model = pipeline.Train<IrisData, ClusterPrediction>();
32             return model;
33         }
34
35         //在Main方法前加async关键字是c#7.1及以后的新功能
36         //应当右键单击项目名称(IrisClustering) 属性->生成->高级->语言版本 中将语言改为c#7.1
37         private static async Task Main(string[] args)
38         {
39             PredictionModel<IrisData, ClusterPrediction> model = Train();
40             await model.WriteAsync(_modelPath);
41             var prediction = model.Predict(TestIrisData.Setosa);
42             Console.WriteLine($"Cluster:{prediction.PredictedClusterId}");
43             Console.WriteLine($"Distances:{string.Join(" ",prediction.Distances)}");
44             Console.Read();
45         }
46     }
47 }

 1 namespace IrisClustering
 2 {
 3     static class TestIrisData
 4     {
 5         internal static readonly IrisData Setosa = new IrisData
 6         {
 7             SepalLength = 5.1f,
 8             SepalWidth = 3.5f,
 9             PetalLength = 1.4f,
10             PetalWidth = 0.2f
11         };
12     }
13 }