one-hot实现代码
时间: 2023-12-29 21:03:50 浏览: 65
手动实现One-Hot编码完整实现Python版
5星 · 资源好评率100%
以下是一个简单的 Python 代码实现 one-hot 编码:
```python
def one_hot_encode(sequence, n_unique):
encoding = []
for value in sequence:
vector = [0 for _ in range(n_unique)]
vector[value] = 1
encoding.append(vector)
return encoding
```
这个函数的参数有两个。第一个参数是需要编码的序列,它可以是数字、字符串或其他类型。第二个参数 n_unique 是序列中唯一值的数量。
函数的实现方式很简单。它首先创建一个空列表 encoding,然后遍历序列中的每个值。对于每个值,它创建一个长度为 n_unique 的零向量,然后将向量中对应的位置设置为 1。最后,它将编码向量添加到 encoding 列表中,并返回该列表。
阅读全文