当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python clx.analytics.asset_classification.AssetClassification.save_model用法及代码示例


用法:

save_model(fname)

保存训练好的模型

参数：

save_to_path：(str) - 保存模型的目录路径

例子：

>>> from clx.analytics.asset_classification import AssetClassification
>>> ac = AssetClassification()
>>> cat_cols = ["1", "2", "3", "4", "5", "6", "7", "8", "9"]
>>> cont_cols = ["10"]
>>> ac.train_model(X_train, cat_cols, cont_cols, "label", batch_size, epochs, lr=0.01, wd=0.0)
>>> ac.save_model("ac.mdl")

相关用法


注：本文由纯净天空筛选整理自rapids.ai大神的英文原创作品 clx.analytics.asset_classification.AssetClassification.save_model。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。