Что выводит keras model.predict?Python

Программы на Python
Anonymous
Что выводит keras model.predict?

Сообщение Anonymous »

Я построил модель LSTM (см. ниже) и обучил ее. Моя функция потерь — это двоичная перекрестная энтропия, поскольку я занимаюсь двоичной классификацией. Данные обучения y представляют собой набор из 0 и 1.
Когда я запускаю model.predict(x_test_scaled), я получаю набор одиночных серий со значениями в диапазоне от 0 до 1. Я предполагаю, что это вероятность, но это вероятность того, что выходные данные = 0 или что выходные данные = 1?
model = Sequential()

model.add(LSTM(units=512, input_shape = (X_train.shape[1],X_train.shape[2]), return_sequences=True))
model.add(Dropout(0.3))

model.add(LSTM(units=512, return_sequences=False))
model.add(Dropout(0.3))

model.add(Dense(264), activation = 'tanh')

model.add(Dense(1))


Подробнее здесь: https://stackoverflow.com/questions/657 ... ict-output

Вернуться в «Python»