python pickle

save data

pickle_file="dataset.pickle"
try:
    f=open(pickle_file,"wb")
    save={"train_dataset":train_dataset,
          "train_labels":train_labels,
          "validate_dataset":validate_dataset,
          "validate_labels":validate_labels"test_dataset":test_dataset,
          "test_labels":test_labels}
    pickle.dump(save,f,pickle.HIGHEST_PROTOCOL)
    f.close()
except Exception as error:
    print("unable to save datat to", pickle_file,":",error)
    raise

load data

f=open(pickle_file,"rb")
dataset=pickle.load(f)
train_dataset=dataset["train_dataset"]
train_labels=dataset["train_labels"]
...

你可能感兴趣的:(python)