Keras中的fit函数训练集，验证集和测试集

最新推荐文章于 2023-04-19 13:24:53 发布

Study Hard and On

最新推荐文章于 2023-04-19 13:24:53 发布

阅读量9.7k

点赞数 6

文章标签：深度学习

本文链接：https://blog.csdn.net/cuiwei5201314/article/details/105929734

版权

Keras中的fit函数训练集，验证集和测试集

1.Keras fit函数history对象包含两个重要属性：
epoch：训练的轮数
history：它是一个字典，包含val_loss,val_acc,loss,acc四个key。
2. 关于训练集，验证集和测试集：
其中验证集是从训练集中抽取出来用于调参的，而测试集是和训练集无交集的，用于测试所选参数用于该模型的效果的。
在Keras中，验证集的划分只要在fit函数里设置validation_split的值就好了，这个对应了取训练集中百分之几的数据出来当做验证集。但由于shuffle是在validation _split之后执行的，所以如果一开始训练集没有shuffle的话，有可能使验证集全是负样本。测试集的使用只要在evaluate函数里设置就好了。
print model.evaluate（test_set_x，test_set_y ,batch_size=256）
这里注意evaluate和fit函数的默认batch_size都是32，自己记得修改。

出处：(https://cloud.tencent.com/developer/article/1437999)