(파이썬 2.7) pickle 파일 저장 및 불러오기

[ 제 네이버 블로그 글을 티스토리로 이전한 글입니다. 해당 글은 제 네이버 블로그에 2018.04.05에 작성되었습니다 ]

 

파이썬 환경에서 처리한 내용들을 저장하고 싶을 때가 종종 있습니다.
txt 파일, mat 파일, p 파일 등으로 저장을 하는 편인데, 그중에서 pickle을 이용해서 .p로 저장하고 불러오는 것이 가장 편했습니다.

.p로 저장할 때는 dictionary를 이용해서 한번에 파일들을 넣어주었습니다.

예제 코드

파일 저장 시

import pickle x_acc = ~~ x_f1 = ~~ x_loss =  ~~ x_set = {'acc': x_acc,               'f1': x_f1,               'loss': x_loss} saving_name = '/root/save/test.p' pikcle.dump(x_set, open(saving_name, "wb"))

파일 불러올 때

import cPickle saving_name = '/root/save/test.p' x = cPikcle.load(saving_name, 'rb')) acc = x['acc'] f1 = x['f1'] loss = x['loss']