Skip to content

Latest commit

 

History

History

cnn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

CNN实现

  1. 基础配置
# 训练验证数据集目录
path = '../../data/train-validation-set'
# 模型保存地址,最后接的是模型名字
model_path = '../../model/model.ckpt'
  1. 参数调试
...
# 训练集和验证集,训练集比例
ratio = 0.7
...

# 训练和测试数据,n_epoch是训练次数
n_epoch = 50
batch_size = 32

# 以及各层的参数
...
  1. 网络结构 网络结构

代码最后部分的是打印图标,包括 train_loss, train_acc, validation_loss, volidation_acc。
训练完所有的epoch后

  1. validation_loss趋于0(我们这个模型训练结果大概0.5)
  2. validation_acc趋于1.0(这个大概0.90)