Настройте среду
Импортировать модули, которые мы собираемся использовать
Установите наши данные
Мы используем Титаник: Машинное обучение после катастрофы в качестве примера набора данных.
- Прочтите набор данных
- Разберитесь с отсутствующим значением
- Перекодировать переменные
- Проверить распределение целевой переменной
Подготовить данные для моделирования
- Разделить данные (X, y)
- Преобразовать в массив NumPy
Построить, скомпилировать, подогнать модель
- Используйте последовательный API для построения вашей модели
- Укажите оптимизатор (rmsprop или Adam)
- Установите функцию потерь (binary_crossentropy)
- Подгоните модель (создайте новую переменную под названием «история», чтобы вы могли оценить кривые обучения)
- Обратные вызовы EarlyStopping для предотвращения переобучения (терпение 10)
Оцените модель
- Кривые обучения (потеря)
- Кривые обучения (точность)
- Матрица путаницы