java实现one-hot
时间: 2023-07-07 20:17:13 浏览: 153
Word2Vec预训练字符串相似度-java实现,详细注释到行,附工程源码,预训练文件,F5直接运行
在 Java 中,可以通过以下方式实现 one-hot 编码:
1. 创建一个长度为 n 的零数组,其中 n 是类别数量。
```
int[] oneHot = new int[n];
```
2. 将要编码的类别的索引位置设置为 1。
```
oneHot[index] = 1;
```
例如,假设有三个类别:A,B,C。要将类别 B 编码为 one-hot 向量,则可以执行以下操作:
```
int[] oneHot = new int[3];
oneHot[1] = 1;
```
此时,oneHot 数组的值为 [0, 1, 0],表示类别 B。
阅读全文