如何阅读Google Colab上传的腌制文件

时间:2018-03-19 00:13:45

标签: pickle google-colaboratory

我已经使用

在google colab上上传了pickled文件

from google.colab import files uploaded = files.upload()

说我的pickle文件名是Train.p,我如何使用典型函数使用它,我已经尝试了下面的代码,但它不起作用。

with open(io.StringIO(uploaded['train.p']), 'rb') as file:
train = pickle.load(file)

2 个答案:

答案 0 :(得分:1)

我实际上稍微调整了上面的答案代码,这对我有用:

import pickle
train = pickle.load(io.BytesIO(uploaded[‘train.p’]))

答案 1 :(得分:0)

试试这个

import io
train = pickle.load(io.BytesIO(uploaded[‘train.p’]))