当前位置:   article > 正文

数据存储格式-pkl_net.pkl

net.pkl

1.pkl文件


pkl文件是python里面保存文件的一种格式,如果直接打开会显示一堆序列化的东西(二进制文件)。
常用于保存神经网络训练的模型或者各种需要存储的数据。

保存神经网络训练模型举例(使用pytorch进行保存)
保存整个网络:torch.save(net, ‘net.pkl’)
保存网络的状态信息:torch.save(net.state_dict(), ‘net_params.pkl’)
提取神经网络的方法:
torch.load(‘net.pkl’)
存储数据举例

  1. import pandas as pd
  2. import numpy as np
  3. df = pd.DataFrame(np.arange(20).reshape(4,5))
  4. df.to_pickle('foo.pkl')
  5. data=pd.read_pickle('foo.pkl')
  6. print(data)

2.pickle库对文件进行写入,读取操作

  1. 写入pkl文件
  1. import pickle
  2. #wb是覆盖写,如果需要追加,则为‘ab'
  3. f = open('data.pkl','wb')
  4. #待写入数据
  5. datas = {'name':'Bruce','age':25,'high':175}
  6. #写入
  7. data= pickle.dump(datas,f,-1)
  8. #关闭文件
  9. f.close()
  1. 读取.pkl文件

 

  1. f = open('data.pkl','rb')
  2. #使用load的方法将数据从pkl文件中读取出来
  3. pickle.load(f)
  4. #关闭文件
  5. f.close()

3.pandas库对文件进行写入,读取操作

  1. 写入.pkl文件:

使用DataFrame的to_pickle就可以生成pickle文件,因此如果需要存储其他类型的数据将其转化为DataFrame即可存取,例如将dict类型数据保存在.pkl文件中

  1. import pandas as pd
  2. import numpy as np
  3. df = pd.DataFrame(np.arange(20).reshape(4,5))
  4. #使用DataFrame的to_pickle就可以生成pickle文件
  5. df.to_pickle('data.pkl')
  6. #通过字典创建数据
  7. dic={'A':[1,2,3],'B':[4,5,6],'C':[7,8,9]}
  8. df2 = pd.DataFrame(dic,index=["a","b","c"])
  9. df2.to_pickle('data2.pkl')
  10. #字典的键和值将转换为DataFrame的两列,其列名如选项列中所给
  11. stu = {
  12. 'xiaoming':2,
  13. 'xiaohong':3,
  14. 'xiaoqiang':7,}
  15. print(pd.DataFrame(list(stu.items()),columns=['name', 'age']))
  16. '''
  17. 输出结果为:
  18. name age
  19. 0 xiaoming 2
  20. 1 xiaohong 3
  21. 2 xiaoqiang 7
  22. '''

2.读取.pkl文件:

 

  1. data=pd.read_pickle('data.pkl')
  2. data2=pd.read_pickle('data2.pkl')
  3. print('data:\n',data)
  4. print('data2:\n',data2)

原文链接:https://blog.csdn.net/Ving_x/article/details/114488844

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Li_阴宅/article/detail/756107
推荐阅读
相关标签
  

闽ICP备14008679号