我想把下面的字符串转换成分类形式或一个热编码。
string1 = "Interstitial markings are diffusely prominent throughout both lungs. Heart size is normal. Pulmonary XXXX normal."
st1 = string1.split()
我正在使用下面的代码,但它生成错误。
from numpy import array
from numpy import argmax
from keras.utils import to_categorical
# define example
data = array(st1)
print(data)
encoded = to_categorical(data)
print(encoded)
# invert encoding
inverted = argmax(encoded[0])
print(inverted)
误差
['Interstitial' 'markings' 'are' 'diffusely' 'prominent' 'throughout' 'both' 'lungs.' 'Heart' 'size' 'is' 'normal.' 'Pulmonary' 'XXXX''normal.']
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-15-b034d9393342> in <module>
5 data = array(st1)
6 print(data)
----> 7 encoded = to_categorical(data)
8 print(encoded)
9 # invert encoding
/usr/local/lib/python3.7/dist-packages/keras/utils/np_utils.py in to_categorical(y, num_classes, dtype)
60 [0. 0. 0. 0.]
61 """
---> 62 y = np.array(y, dtype='int')
63 input_shape = y.shape
64 if input_shape and input_shape[-1] == 1 and len(input_shape) > 1:
ValueError: invalid literal for int() with base 10: 'Interstitial'
2条答案
按热度按时间2nbm6dog1#
从逻辑上讲,错误是将str类型转换为int。
喜欢
这是因为
keras仅支持对已进行整数编码的数据进行单热编码。
在这种情况下,可以按如下所示使用
LabelEncoder
。给出编号
xmq68pz92#
Tensorflow已经清楚地提到了here,即
tf.keras.utils.to_categorical
是用于将类向量(整数)转换为二进制类矩阵的。您的
data
变量包含字符串类型元素,这与integer
不同,因此出现错误。