TSNE降维学习

在学习使用Cora数据集时,输入的初始为2708, 1433, 输入一共有2708个点,每个节点有1433个特征。测试集的大小为1000, 1433最后的输出为1000, 7,表示每个点的类别。

我们要的就是查看这个1000, 7的预测结果。想要将他展示在一个平面上是比较困难的,因为平面是2维的。所以需要用TSNE来将1000, 7降维至1000, 2

使用过程如下:

python 复制代码
# tsne visualize
    # TSNE 用于降维
    tsne = TSNE()
    out = tsne.fit_transform(predict)   # out: [1000, 2]    predict:[1000, 7]
    fig = plt.figure()
    for i in range(7):
        indices = tensor_y == i
        x, y = out[indices].T
        plt.scatter(x, y, label=str(i))

    plt.legend(loc=0)
    plt.savefig('tsne.png')
    plt.show()
相关推荐
AOwhisky6 小时前
Redis 学习笔记(第三期):持久化与主从复制
运维·数据库·redis·笔记·学习·云计算
xxie1237946 小时前
return与print
开发语言·python
秋96 小时前
从 Python 后端工程师转型 AI Engineer(AI 工程化)的完整补课清单(2026实战版)
开发语言·人工智能·python
程序员二叉7 小时前
【Java】 异常高频面试题精讲 | 易错点+对比总结
java·开发语言·面试
慕木沐7 小时前
Google ADK Java 1.0版本 核心机制与实战 Demo
java·开发语言·python
Tbisnic7 小时前
AI大模型学习第十一天:技术选型、安全防护与金融实战
python·学习·ai·大模型·提示词工程
Roann_seo%7 小时前
C++文件操作完全指南:从文本读写到二进制文件处理
开发语言·c++
hboot8 小时前
AI工程师第一课 - Python
前端·后端·python
huangdong_8 小时前
淘宝商品SKU图自动分类技术深度解析:从DOM解析到智能归档
开发语言·javascript·ecmascript
阿正的梦工坊8 小时前
【Rust】12-借用检查器与非词法生命周期
开发语言·后端·rust