python 计算onehot 并实现绘图

最新推荐文章于 2024-09-27 10:11:28 发布

guofeng314

最新推荐文章于 2024-09-27 10:11:28 发布

阅读量89

点赞数

文章标签： python 机器学习数据挖掘

本文链接：https://blog.csdn.net/guofeng314/article/details/131269351

版权

from sklearn.feature_extraction.text import CountVectorizer
import seaborn as sns
import matplotlib.pyplot as plt
corpus = ['time flies like an arrow', 'Fruit flies like a banana']
one_hot_vectorizer = CountVectorizer(binary=True)
one_hot=one_hot_vectorizer.fit_transform(corpus).toarray()
print(one_hot)
fs = one_hot_vectorizer.get_feature_names_out();
print(fs)
sns.heatmap(one_hot, annot=True, cbar=False, xticklabels=fs, yticklabels=['s1','s2'])
plt.show()