/
sat1988
/
MachineVision
Обзор
Документация
Войти
/
sat1988
/
MachineVision
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
task06/train.py
24 строки
815 B
Oleg Chorakaev
Добавлен 6 пример
26 июл 2025, 16:11
26 июл 2025, 16:11
edf09e1
Код
Авторство
О чём код?
import keras import numpy as np from keras.src.layers import Flatten, Dense x_test = np.load('data/x_test.npy') y_test = np.load('data/y_test.npy') x_train = np.load('data/x_train.npy') y_train = np.load('data/y_train.npy') model = keras.Sequential([ Flatten(input_shape=(28, 28, 1)), Dense(128, activation='relu'), Dense(10, activation='softmax') ]) print(model.summary()) x_train = x_train / 255 x_test = x_test / 255 y_train_cat = keras.utils.to_categorical(y_train, 10) y_test_cat = keras.utils.to_categorical(y_test, 10) model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) model.fit(x_train, y_train_cat, batch_size=100, epochs=10, validation_split=0.2) print("На тестовых данных:") model.evaluate(x_test, y_test_cat) model.save('models/Dense.h5')