python保存类对象到文件_python 保存对象文件

之前用 R 语言一直感觉 .Rdata 格式的文件很好用，可以把每次执行的中间文件保存便于下次调用，刚熟悉 Python 还没接触这块知识，所以有时候做项目不太顺手，索性上网搜了下，整理如下：

模型存档

#############

# joblib 库

from sklearn.linear_model import LogisticRegression

from sklearn.externals import joblib

# 模型保存

lr_model = LogisticRegression()

joblib.dump(lr_model, 'xx.model')

# 模型载入

lr_model = joblib.load('xx.model')

#############

# pickle 库

# 模型保存

import pickle

with open('lr_model.pickle', 'wb') as fp:

pickle.dump(lr_model, fp)

# 模型载入

with open('lr_model.pickle', 'rb') as fp:

lr_model = pickle.load(fp)

命令行参数解析

import argparse

# 定义参数解析

def build_arg_parser():

parser = argparse.ArgumentParser(description='Trains the