使用SK-Learn处理不平衡数据

author =“maxwell flitton”,organization ='example',

file_name ='neural.sav')

这使您可以再次使用模型,而不会忘记输入顺序或指标等详细信息。要加载模型,您只需使用pickle:

import pickle

loaded_model = pickle.load(open("neural.sav", 'rb'))

这会加载字典。如果您有新数据,可以调用“scaler” key 访问保存的缩放器,并转换新的输入数据:

input_data = loaded_model['scaler'].transform([[1, 28, 0, 1, 30]])

现在您已经处理了新数据,您可以使用以下方法进行计算:

loaded_model['model'].predict(input_data)[0][0]

deploy-ml包装器只需几行代码就可以重新采样并保存记录良好的SK-Learn模型。