/DLToolbox

深度学习实验工具箱,专为科研设计

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

DLToolbox

深度学习实验工具箱,专为科研设计。拥有统一模型接口定义,高度可扩展,不仅适用于深度学习,传统机器学习算法,sklearn库等也可以很好的结合进来,实现自动化实验

目前进展

分类器

分类器部分基本完善,拥有灵活的训练参数配置,训练过程控制和自定义,自动绘制多类ROC曲线,自动输出各项参数等功能。

下一步目标

由于初期过去针对分类器,整体系统设计较为混乱,未来将,重新设计DataSource接口,使其规范化,并引入数据流管道的概念,方便未来以更复杂的形式训练网络,例如多出口GAN等