首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Microsoft.ML KMeans get clusters

Microsoft.ML KMeans get clusters
EN

Stack Overflow用户
提问于 2019-09-08 03:12:12
回答 1查看 97关注 0票数 0

我想使用Microsoft.ML聚类KMeans对颜色进行量化。我有一个带有颜色的数据集,关键是获取关于训练过的聚类的信息。

有没有办法从模型/上下文中获取关于经过训练的集群的信息?

代码:

代码语言:javascript
复制
var mlContext = new MLContext();

// color is System.Drawing.Color array    
var trainingColors = colors
                    .Select(s => new ColorTrainingModel() { R = s.R, G = s.G, B = s.B })
                    .ToArray();

var trainingData = mlContext.Data.LoadFromEnumerable(trainingColors);

var kmeans = mlContext.Clustering.Trainers.KMeans(
        featureColumnName: "Features",
        numberOfClusters: 4);

var pipeline = mlContext.Transforms.Concatenate( 
        "Features",
        nameof(ColorTrainingModel.R), 
        nameof(ColorTrainingModel.G), 
        nameof(ColorTrainingModel.B)
    )

    .Append(kmeans);

var model = pipeline.Fit(trainingData);

// kinda result I am looking for
var colorPalette = model.Clusters.Select(s => new { R = s[0], G = s[1], ...
EN

回答 1

Stack Overflow用户

发布于 2019-09-08 12:41:13

得到了一个解决方案,它基本上与Microsoft.ML无关,但是,它使用KMeans并执行以下操作:

使用Accord.NET框架:http://accord-framework.net/docs/html/T_Accord_MachineLearning_KMeans.htm

代码语言:javascript
复制
var observations = colors
                    .Select(s => new double[] { s.R, s.G, s.B })
                    .ToArray();

var kmeans = new Accord.MachineLearning.KMeans(k: colorCount);

var clusters = kmeans.Learn(observations);

var palette = new List<System.Drawing.Color>();

foreach (var c in clusters)
{
    var col = System.Drawing.Color.FromArgb(

        (int)Math.Round(c.Centroid[0], 0),  //R
        (int)Math.Round(c.Centroid[1], 0),  //G
        (int)Math.Round(c.Centroid[2], 0)   //B
    );

    palette.Add(col);
}

return palette.ToArray();
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/57836752

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档