3 Star 0 Fork 0

银梓萌/emotion-detection-keras-master

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
train.py 996 Bytes
一键复制 编辑 原始数据 按行查看 历史
靳康龄 提交于 2022-03-24 12:26 . train.py
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
nb_epoch = 20
batch_size = 128
# data generator
data_generator = ImageDataGenerator(
featurewise_center=False,
featurewise_std_normalization=False,
rotation_range=10,
width_shift_range=0.1,
height_shift_range=0.1,
zoom_range=.1,
horizontal_flip=True)
es = EarlyStopping(monitor='val_loss', patience = 10, mode = 'min', restore_best_weights=True)
history = model.fit_generator(data_generator.flow(x_train, y_train, batch_size),
steps_per_epoch=len(x_train) / batch_size,
epochs=nb_epoch,
verbose=1,
callbacks = [es],
validation_data=(x_eval, y_eval))
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yinzimeng11/emotion-detection-keras-master.git
git@gitee.com:yinzimeng11/emotion-detection-keras-master.git
yinzimeng11
emotion-detection-keras-master
emotion-detection-keras-master
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385