keras one-hot 编码和解码

最新推荐文章于 2026-03-27 17:58:20 发布

原创最新推荐文章于 2026-03-27 17:58:20 发布 · 1.3w 阅读

5 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#keras

收录于

本文介绍如何使用Keras库中的to_categorical函数对一维整数数组进行One-Hot编码，并演示了如何将编码后的数据解码回原始整数形式。通过实例展示了数据形状的变化以及编码和解码的过程。

参考
https://jovianlin.io/keras-one-hot-encode-decode-sequence-data/

import numpy as np
from keras.utils import to_categorical

data = np.array([1, 5, 3, 8])
print(data)

def encode(data):
    print('Shape of data (BEFORE encode): %s' % str(data.shape))
    encoded = to_categorical(data)
    print('Shape of data (AFTER  encode): %s\n' % str(encoded.shape))
    return encoded

encoded_data = encode(data)
print(encoded_data)

def decode(datum):
    return np.argmax(datum)

# 单个解
for i in range(encoded_data.shape[0]):
    datum = encoded_data[i]
    print('index: %d' % i)
    print('encoded datum: %s' % datum)
    decoded_datum = decode(encoded_data[i])
    print('decoded datum: %s' % decoded_datum)
    print()

# 全部解
print np.argmax(encoded_data,axis=1)