我有7个分类,我在我的y_列车上使用了标签编码器,即使我得到了这个错误,也将其转换为float。请调查一下这个问题
le = LabelEncoder()
yy_train=le.fit_transform(y_train)
yy_train=yy_train.astype(float)
model = Sequential()
model.add(Dense(186, input_shape=(180,), activation = 'relu'))
model.add(Dense(256, activation = 'relu'))
model.add(Dropout(0.6))
model.add(Dense(128, activation = 'relu'))
model.add(Dropout(0.5))
model.add(Dense(10, activation = 'softmax'))
model.compile(loss='categorical_crossentropy', metrics=['accuracy'], optimizer='adam')
history = model.fit(X_train, yy_train, batch_size=64, epochs=30)
您应该从类别数组中转换yy_train 指向指示类别的二进制值数组
例如
相关问题 更多 >
编程相关推荐