Я хочу сделать прогноз с помощью нейросети Keras. Мои выходные данные имеют 3 разных значения -1,0,1. Когда я запускаю свой NN, я получаю ошибку:
ValueError: Error when checking target: expected dense_35 to have shape (3,) but got array with shape (1,)
Затем я пытаюсь сделать:
from tensorflow.python.keras.utils import to_categorical
results = to_categorical(results)
Но снова я получаю ту же ошибку:
ValueError: Error when checking target: expected dense_35 to have shape (3,) but got array with shape (2,)
Что я делаю неправильно? Это мой код:
features = df.iloc[:,-8:]
results = df.iloc[:,-9]
x_train, x_test, y_train, y_test = train_test_split(features, results, test_size=0.3, random_state=42)
model = Sequential()
model.add(Dense(64, input_dim = x_train.shape[1], activation = 'relu')) # input layer requires input_dim param
model.add(Dense(32, activation = 'relu'))
model.add(Dense(16, activation = 'relu'))
model.add(Dense(3, activation = 'softmax'))
model.compile(loss="categorical_crossentropy", optimizer= "adam", metrics=['accuracy'])
# call the function to fit to the data training the network)
es = EarlyStopping(monitor='val_loss', min_delta=0.001, patience=0, verbose=1, mode='auto')
model.fit(x_train, y_train, epochs = 10, shuffle = True, batch_size=128, validation_data=(x_test, y_test), verbose=2, callbacks=[es])