:param mode: mode 有picture和data和result三种模式, picture模式将训练结果输出成折线图， result模式将最后一个epoch的结果输出成日志文件 data模式记录每一个epoch的结果并输出成日志文件 :modelname: 神经网络的模型名 :classnum :待分类数 :return:
(epochs, modelname, classnum, mode, train_data_dir, validation_data_dir)
| 9 | |
| 10 | |
| 11 | def runClassification(epochs, modelname, classnum, mode, train_data_dir, validation_data_dir): |
| 12 | ''' |
| 13 | |
| 14 | :param mode: mode 有picture和data和result三种模式, |
| 15 | picture模式将训练结果输出成折线图， |
| 16 | result模式将最后一个epoch的结果输出成日志文件 |
| 17 | data模式记录每一个epoch的结果并输出成日志文件 |
| 18 | |
| 19 | :modelname: 神经网络的模型名 |
| 20 | :classnum :待分类数 |
| 21 | :return: |
| 22 | ''' |
| 23 | for i in range(run_times): |
| 24 | Classification(epochs=epochs, modelname=modelname, classnum=classnum, train_data_dir=train_data_dir, |
| 25 | validation_data_dir=validation_data_dir, mode=mode).run() |
| 26 | |
| 27 | |
| 28 | if __name__ == '__main__': |
no test coverage detected