Smoketrees模型动物园套餐
forest-utils的Python项目详细描述
森林实用工具-一个实用工具为烟雾树模型动物园
这个软件包将有助于实现被人们亲切地称为“烟树森林”的烟树模型动物园的所有设施
特点
- [十] 从动物园里取下HDF5(.h5)模型并装载keras
- [十] 从动物园下拉SavedModel(.pb)模型并加载keras
- [十] 从动物园下拉检查点(.ckpt)模型并加载keras
- [十] 从动物园取下spacy模型并将其加载到nlp管道中
- [十] 根据huggingface模型架构,下拉标记器和模型并重用它们
- [十] 下拉数据集以便在您自己的项目中使用它们
从PyPI
安装包的步骤pip install forest-utils
从源代码
安装包的步骤- 通过以下命令克隆repo ^{pr2}$ 在
- 更改包的目录
cd forest-utils
在 - 执行以下命令
pip install -e .
在
示例用法
- 基于张量流的负荷模型
fromforest_utilsimportexport_kerasmodel=export_keras.ModelFromH5().model
- 加载数据集
fromforest_utilsimportdatasetstweets=datasets.Dataset().get_emo_tweets()
下拉模型后,将其用于预测和其他评估功能。在
有关更多使用示例,请参见SmokeTrees ModelZoo
标签:
- 项目
标签: