Python字典实现单词One-Hot编码:简单易懂的代码示例
假设我们有一个单词列表'words',其中包含一些单词。我们可以使用Python中的字典来实现单词的One-Hot转换,代码如下:
# 创建单词列表
words = ['cat', 'dog', 'bird', 'fish']
# 创建空字典
onehot = {}
# 遍历单词列表
for i, word in enumerate(words):
# 创建一个全为0的列表
onehot_vector = [0] * len(words)
# 将当前单词的位置设置为1
onehot_vector[i] = 1
# 将单词和对应的onehot向量添加到字典中
onehot[word] = onehot_vector
# 打印结果
print(onehot)
运行结果如下所示:
{'cat': [1, 0, 0, 0], 'dog': [0, 1, 0, 0], 'bird': [0, 0, 1, 0], 'fish': [0, 0, 0, 1]}
在这个例子中,我们使用enumerate函数获取每个单词的索引,然后为每个单词创建一个全为0的onehot向量。接下来,我们将当前单词的位置设置为1,最后将单词和对应的onehot向量添加到字典中。
原文地址: https://www.cveoy.top/t/topic/oTx7 著作权归作者所有。请勿转载和采集!