Python可视化神器Yellowbrick使用

前言

文的文字及图片来源于网络,仅供学习、交流使用,不具有任何商业用途,版权归原作者所有,如有问题请及时联系我们以作处理。

作者： 沂水寒城

ps：如有需要python学习资料的小伙伴可以加点击下方链接自行获取

在机器学习、数据挖掘领域里面，接触到数据处理分析的人来说，数据探索是非常重要的一部分工作，而数据可视化会成为数据分析工程师完成数据探索工作的有力工具。本文主要是介绍一款我日常使用较多的可视化利器yellowbrick，这是一款基于sklearn+matplotlib模块构建的更加高级的可视化工具，能够更加方便地完成很多数据探索、分词与展示工作。

学习使用一个模块最好的方式就是学习它提供的api，下面先给出来几个比较好的参考地址：

1）官方文档地址（英文）

https://www.scikit-yb.org/en/latest/

2）官方文档地址（中文）

http://www.scikit-yb.org/zh/latest/

yellowbrick是由一套被称为"visualizers"组成的可视化诊断工具组成的集合，其由scikit-learn api延伸而来，对模型选择过程其指导作用。总之，yellowbrick结合了scikit-learn和matplotlib并且最好得传承了scikit-learn文档，对你的模型进行可视化！

想要了解yellowbrick就必须先了解visualizers，它是estimators从数据中学习得的对象，其主要任务是产生可对模型选择过程有更深入了解的视图。从scikit-learn来看，当可视化数据空间或者封装一个模型estimator时，其和转换器（transformers）相似，就像"modelcv" (比如ridgecv,lassocv)的工作原理一样。yellowbrick的主要目标是创建一个和scikit-learn类似的有意义的api。

yellowbrick中最受欢迎的visualizers包括：

如此强大的可视化工具，安装方式却很简单，使用下面的命令：

pip install yellowbrick

如果需要升级最新版本的则可以使用下面的命令：

pip install –u yellowbrick

安装完成后，我们就可以进行使用了。该模块提供了几个常用的可用于实验使用的数据集，如下所示：

进入到对应数据集文件夹下面，都会有三个文件，对于bikeshare如下：

其中：bikeshare.csv为数据集文件，如：

meta.json为字段元信息文件，如：

readme.md为介绍说明文件，如：

基于共享单车数据集，简单的数据分析工作实现如下：

 1 def testfunc5(savepath='results/bikeshare_rank2d.png'):
 2     '''
 3     共享单车数据集预测
 4     '''
 5     data=pd.read_csv('bikeshare/bikeshare.csv')
 6     x=data[["season", "month", "hour", "holiday", "weekday", "workingday",
 7             "weather", "temp", "feelslike", "humidity", "windspeed"
 8           ]]
 9     y=data["riders"]
10     visualizer=rank2d(algorithm="pearson")
11     visualizer.fit_transform(x)
12     visualizer.poof(outpath=savepath)
13 ​
14 ​
15 def testfunc6(savepath='results/bikeshare_temperate_feelslike_relation.png'):
16     '''
17     进一步考察相关性
18     '''
19     data=pd.read_csv('bikeshare/bikeshare.csv')
20     x=data[["season", "month", "hour", "holiday", "weekday", "workingday",
21             "weather", "temp", "feelslike", "humidity", "windspeed"]]
22     y=data["riders"]
23     visualizer=jointplotvisualizer(feature='temp', target='feelslike')
24     visualizer.fit(x['temp'], x['feelslike'])
25     visualizer.poof(outpath=savepath)
26 ​
27 ​
28 def testfunc7(savepath='results/bikeshare_linearregression_residualsplot.png'):
29     '''
30     基于共享单车数据使用线性回归模型预测
31     '''
32     data = pd.read_csv('bikeshare/bikeshare.csv')
33     x=data[["season", "month", "hour", "holiday", "weekday", "workingday",
34             "weather", "temp", "feelslike", "humidity", "windspeed"]]
35     y=data["riders"]
36     x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.3)
37     visualizer=residualsplot(linearregression())
38     visualizer.fit(x_train, y_train)
39     visualizer.score(x_test, y_test)
40     visualizer.poof(outpath=savepath)
41 ​
42 ​
43 def testfunc8(savepath='results/bikeshare_ridgecv_alphaselection.png'):
44     '''
45     基于共享单车数据使用alphaselection
46     '''
47     data=pd.read_csv('bikeshare/bikeshare.csv')
48     x=data[["season", "month", "hour", "holiday", "weekday", "workingday",
49             "weather", "temp", "feelslike", "humidity", "windspeed"]]
50     y=data["riders"]
51     alphas=np.logspace(-10, 1, 200)
52     visualizer=alphaselection(ridgecv(alphas=alphas))
53     visualizer.fit(x, y)
54     visualizer.poof(outpath=savepath)
55 ​
56 ​
57 def testfunc9(savepath='results/bikeshare_ridge_predictionerror.png'):
58     '''
59     基于共享单车数据绘制预测错误图
60     '''
61     data=pd.read_csv('bikeshare/bikeshare.csv')
62     x=data[["season", "month", "hour", "holiday", "weekday", "workingday",
63             "weather", "temp", "feelslike", "humidity", "windspeed"]]
64     y=data["riders"]
65     x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.3)
66     visualizer=predictionerror(ridge(alpha=3.181))
67     visualizer.fit(x_train, y_train)
68     visualizer.score(x_test, y_test)
69     visualizer.poof(outpath=savepath)

 

bikeshare_rank2d.png特征相关性计算

bikeshare_linearregression_residualsplot.png使用线性回归模型预测

bikeshare_ridgecv_alphaselection.png使用alphaselection特征选择

bikeshare_ridge_predictionerror.png绘制预测错误图

除了可以直接对数据进行分析展示之外，yellowbrick同样可以对文本数据进行处理分析，下面我们基于爱好数据集进行简单的使用说明，具体代码实现如下所示：

 1 def hobbiesfreqdistvisualizer():
 2     '''
 3     文本可视化
 4     token 频率分布：绘制语料库中令牌的频率
 5     t-sne 语料库可视化：绘制更接近发现聚类的类似文档
 6     '''
 7     corpus=load_corpus("data/hobbies")
 8     vectorizer = countvectorizer()
 9     docs       = vectorizer.fit_transform(corpus.data)
10     features   = vectorizer.get_feature_names()
11     visualizer = freqdistvisualizer(features=features)
12     visualizer.fit(docs)
13     visualizer.poof(outpath='text_hobbies_freqdistvisualizer.png')
14     #去停用词
15     vectorizer = countvectorizer(stop_words='english')
16     docs       = vectorizer.fit_transform(corpus.data)
17     features   = vectorizer.get_feature_names()
18     visualizer = freqdistvisualizer(features=features)
19     visualizer.fit(docs)
20     visualizer.poof(outpath='text_hobbies_freqdistvisualizer_stop_words.png')
21 ​
22 ​
23 def hobbiesfreqdistvisualizer2():
24     '''
25     探索 烹饪和游戏 两种爱好的频度分布
26     （报错：没有label，应该为corpus.target）
27     '''
28     corpus=load_corpus("data/hobbies")
29     #烹饪爱好频度分布统计图
30     hobbies=defaultdict(list)
31     for text,label in zip(corpus.data,corpus.target):
32         hobbies[label].append(text)
33     vectorizer = countvectorizer(stop_words='english')
34     docs       = vectorizer.fit_transform(text for text in hobbies['cooking'])
35     features   = vectorizer.get_feature_names()
36     visualizer = freqdistvisualizer(features=features)
37     visualizer.fit(docs)
38     visualizer.poof(outpath='text_hobbies_cooking_freqdistvisualizer.png')
39     #游戏爱好频度分布统计图
40     hobbies=defaultdict(list)
41     for text,label in zip(corpus.data, corpus.target):
42         hobbies[label].append(text)
43     vectorizer = countvectorizer(stop_words='english')
44     docs       = vectorizer.fit_transform(text for text in hobbies['gaming'])
45     features   = vectorizer.get_feature_names()
46     visualizer = freqdistvisualizer(features=features)
47     visualizer.fit(docs)
48     visualizer.poof(outpath='text_hobbies_gaming_freqdistvisualizer.png')
49 ​
50 ​
51 def hobbiestsnevisualizer():
52     '''
53     t-sne语料库可视化
54     t分布随机邻域嵌入，t-sne。scikit-learn将此分解方法实现为sklearn.manifold.tsne转换器。
55      通过使用来自原始维度和分解维度的不同分布将高维文档向量分解为二维。 通过分解为2维或3维，
56      可以使用散点图来显示文档。
57     '''
58     corpus=load_corpus("data/hobbies")
59     tfidf  = tfidfvectorizer()
60     docs   = tfidf.fit_transform(corpus.data)
61     labels = corpus.target
62     tsne = tsnevisualizer()
63     tsne.fit(docs, labels)
64     tsne.poof(outpath='text_hobbies_tsnevisualizer.png')
65     #don't color points with their classes
66     tsne = tsnevisualizer(labels=["documents"])
67     tsne.fit(docs)
68     tsne.poof(outpath='text_hobbies_tsnevisualizer_nocolor.png')
69 ​
70 ​
71 def hobbiesclustertsnevisualizer():
72     '''
73     聚类应用
74     '''
75     corpus=load_corpus("data/hobbies")
76     tfidf  = tfidfvectorizer()
77     docs   = tfidf.fit_transform(corpus.data)
78     clusters=kmeans(n_clusters=5)
79     clusters.fit(docs)
80     tsne=tsnevisualizer()
81     tsne.fit(docs,["c{}".format(c) for c in clusters.labels_])
82     tsne.poof(outpath='text_hobbies_cluster_tsnevisualizer.png')

 

text_hobbies_freqdistvisualizer.png

text_hobbies_freqdistvisualizer_stop_words.png

text_hobbies_cooking_freqdistvisualizer.png

.