one-hot 编码 (字符串和数字类型的标签)

在用Pytorch 验证标签的时候,建议使用ont-hot 编码,这样比较容易实现既定的功能。

直接上代码:  字符串类型  本代码将hello world 换做了矩阵

from numpy import argmax
# define input string
data = 'hello world'
print(data)
# define universe of possible input values
alphabet = 'abcdefghijklmnopqrstuvwxyz '
# define a mapping of chars to integers
char_to_int = dict((c, i) for i, c in enumerate(alphabet))
print("char_to_int:",char_to_int,len(char_to_int))
int_to_char = dict((i, c) for i, c in enumerate(alphabet))
print("int_to_char:",int_to_char)
# integer encode input data
integer_encoded = [char_to_int[char] for char in data]
print(integer_encoded,len(integer_encoded))   ## 寻找数据对应的编码
# one hot encode
onehot_encoded = list()
for value in integer_encoded:
       letter = [0 for _ in range(len(alphabet))]
       letter[value] = 1
       onehot_encoded.append(letter)
   #    print(letter,len(alphabet)
print(onehot_encoded)
# 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值