Как сериализовать модель машинного обучения в файл JSON - PullRequest
Новая
       12

Как сериализовать модель машинного обучения в файл JSON

0 голосов
/ 01 ноября 2019

Я построил модель машинного обучения, используя xgboost.XGBClassifier в python. Я сейчас использую pickle для сериализации модели, но хотел бы также иметь json-файл этой модели. Как я могу это сделать?

Итак, в основном у меня есть этот код: 

from xgboost import XGBClassifier
import pickle
import pandas as pd

# read data
df = pd.read_csv('data.csv', sep='\t')

# features and target : 
X = df.drop(columns=['target'],axis=1)
Y = df['target']

# split the data
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.3, random_state=0)

model = XGBClassifier().fit(X_train,Y_train)

pickle_out = open("pickle_file.pkl","wb")
pickle.dump(model, pickle_out)
pickle_out.close()
...