上传者: 38686231
|
上传时间: 2021-10-22 15:40:08
|
文件大小: 92KB
|
文件类型: -
本文参考了这篇文章:https://blog.csdn.net/weixin_36474809/article/details/89379727侵删
这篇文章的代码非常经典,很值得学习,如有错误请指正
utils.py
import numpy as np
import scipy.sparse as sp
import torch
'''
先将所有由字符串表示的标签数组用set保存,set的重要特征就是元素没有重复,
因此表示成set后可以直接得到所有标签的总数,随后为每个标签分配一个编号,创建一个单位矩阵,
单位矩阵的每一行对应一个one-hot向量,也就是np.identity(len(