xgb_joblib.py

from sklearn.externals import joblib
def save_model(model, name):
    '''
    saves trained model
    '''

    joblib.dump(model, name)


def load_model(name):
    '''
    loads pretrained model
    '''

    return joblib.load(name)

model, pred, pred_proba = create_test_model(lg,
                              x_tr, y_tr, x_te, y_te)
save_model(model, "xgb_model_c_{}.pkl".format(i))