当前位置:   article > 正文

读取csv数据,并做简单t-SNE处理_t-sne如何引进文件数据

t-sne如何引进文件数据
  1. from sklearn.manifold import TSNE
  2. import matplotlib.pyplot as plt
  3. import pandas as pd
  4. import numpy as np
  5. from sklearn import manifold
  6. import xlrd
  7. #将csv格式文件的内容转换为矩阵,不保留第一行(标签)
  8. def load_csv(path):
  9. data_read = pd.read_csv(path)
  10. list = data_read.values.tolist()
  11. data = np.array(list)
  12. print(data.shape)
  13. # print(data)
  14. return data
  15. #载入我的csv格式的数据集
  16. data = load_csv(r"C:\\Users\\Cai\\Desktop\\test\\hcvdat0.csv")
  17. #提取分组信息(最后一列),并将字符串转为数字
  18. target = data[:,12]
  19. target_T = np.array(target)
  20. target_T = [int(x) for x in target_T]
  21. tsne = TSNE(n_components=2,learning_rate=100).fit_transform(data[:,1:11])
  22. plt.figure(figsize=(12, 6))
  23. plt.subplot(121)
  24. plt.scatter(tsne[:, 0], tsne[:, 1], s=10, c= target_T)
  25. plt.colorbar()
  26. plt.show()

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/202247
推荐阅读
相关标签
  

闽ICP备14008679号