【AutoGluon_03】保存模型并调用模型

在训练好autogluon模型之后，可以将模型进行保存。之后当有新的数据需要使用autogluon进行预测的时候，就可以直接加载原来训练好的模型进行训练。

import pandas as pd
from sklearn.model_selection import train_test_split
from autogluon.tabular import TabularDataset, TabularPredictor
label='bugState'
predictor = TabularPredictor(label=label,path="bugStatemodel").fit(df4)

其中，参数path="bugStatemodel"就表示在当前路径下新建一个叫bugStatemodel的文件夹，里面存放着训练好的模型

当要调用模型时：

predictorbs = TabularPredictor.load("bugStatemodel")
bs_pred = predictorbs.predict(test_data)

也可以一次性调用多个训练好的模型：

from autogluon.tabular import TabularDataset, TabularPredictor
predictorCBT = TabularPredictor.load("CBTmodel")
predictorCBD = TabularPredictor.load("CBDmodel")
predictorRE = TabularPredictor.load("REmodel")
predictorXDL = TabularPredictor.load("XDLmodel")

test_data = TabularDataset(yc) # yc是新的数据，即要使用autogluon预测的数据
CBT_pred = predictorCBT.predict(test_data)
CBD_pred = predictorCBD.predict(test_data)
RE_pred = predictorRE.predict(test_data)
XDL_pred = predictorXDL.predict(test_data)