Как я могу изменить функцию потерь, когда изменяется форма моих данных?
Поскольку мои данные слишком велики, я использую pd.read_csv(",chunksize=). Я использую categorical_crossentropy однако в качестве функции потерь на последнем отрезке у меня есть только одна цель. Так что я получаю ошибку:
Вы передаете целевой массив shape (2110, 1), используя в качестве потери categorical_crossentropy.
Что я уже пробовал:
Теперь я знаю, что могу использовать binary_crossentropy за этот последний кусок. Так вот что я сделал:
X_train, X_test, y_train, y_test = train_test_split(train_data, train_labels, shuffle=True, test_size=0.3) if y_train.shape[1] == 1: loss = 'binary_crossentropy' else: loss = 'categorical_crossentropy'
Когда я делаю это, я получаю ошибку:
IndexError: index 1 is out of bounds for axis 0 with size 1
Мои данные закодированы в одну горячую точку. Как я могу устранить эту ошибку? Спасибо