文件名称:tfShell:可轻松扩展,构建,训练,测试和部署机器学习模型的类
文件大小:12KB
文件格式:ZIP
更新时间:2024-02-26 11:18:40
machine-learning tensorflow project-management ProjectmanagementPython
TF壳 该项目提供python类以自动构建,训练和测试tensorflow机器学习模型。 结构如下: 为您要训练的模型架构创建一个类。 例子在图表/ 此类具有init方法,该方法定义权重,偏差等。 此类还具有一个call方法,该方法将一些数据作为输入,并返回模型的输出 创建一个扩展了Models / BaseModel的模型类 在此类的初始化中,创建(1.)中的体系结构类并将其存储为属性 该课程跟踪学习超参数,例如算法和学习率。 这也是张量数据占位符的存储位置,以及在训练和测试期间运行的操作的位置。 创建一列火车类延伸培训/ BaseTrainer 此类存储批处理和培训步骤的逻辑
【文件预览】:
tfShell-master
----Models()
--------gan_model.py(4KB)
--------__init__.py(0B)
--------base_model.py(3KB)
----Graphs()
--------mlp.py(2KB)
--------gan.py(3KB)
--------__init__.py(0B)
----ann_example.py(2KB)
----setup.py(371B)
----.gitignore(52B)
----Training()
--------__init__.py(0B)
--------base_trainer.py(4KB)
--------gan_trainer.py(2KB)
----README.md(2KB)
----gan_example.py(444B)
----Testing()
--------__init__.py(0B)
----Utils()
--------__init__.py(0B)
--------image_loader.py(1KB)