Я готовлю ввод для подачи в нейросеть Keras для решения проблемы мультикласса как:
encoder = LabelEncoder()
encoder.fit(y)
encoded_Y = encoder.transform(y)
# convert integers to dummy variables (i.e. one hot encoded)
dummy_y = np_utils.to_categorical(encoded_Y)
X_train, X_test, y_train, y_test = train_test_split(X, dummy_y, test_size=0.2, random_state=42)
X_train, X_val, y_train, y_val = train_test_split(X_train, y_train, test_size=0.06, random_state=42)
После обучения модели я пытаюсь запустить следующие строки, чтобы получить прогноз, отражающий исходные имена классов :
y_pred = model.predict_classes(X_test)
y_pred = encoder.inverse_transform(y_pred)
y_test = np.argmax(y_test, axis = 1)
y_test = encoder.inverse_transform(y_test)
Однако я получаю удивительно низкий уровень точности (0,36), в отличие от обучения и проверки, которые достигают 0,98. Это правильный способ преобразования классов обратно в исходные метки?
Я вычисляю точность как:
# For training
history.history['acc']
# For testing
accuracy_score(y_test, y_pred)