Набор данных содержит около 9500 обучающих выборок и 2500 тестовых выборок.
Архитектура моей модели:
Код: Выделить всё
model = Sequential([
Dense(128, activation='relu', input_shape=(X_train.shape[1],)),
Dropout(0.4),
Dense(64, activation='relu', kernel_regularizer=tf.keras.regularizers.l2(0.001)),
Dropout(0.3),
Dense(1, activation='sigmoid')
])
- Оптимизатор: Адам
- Потери: двоичная кроссэнтропия
- Эпохи: 30
- Размер пакета: 32
Подробнее здесь: https://stackoverflow.com/questions/798 ... egularizat
Мобильная версия