當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。