Pandas中的iloc函数; 查看.pt文件内容

1 Pandas中的iloc函数

前面的冒号就是取行数,后面的冒号是取列数:

import pandas as pd
df_train=pd.read_json('./Twibot-20/train.json')
df_test=pd.read_json('./Twibot-20/test.json')
df_train=df_train.iloc[:,[0,1,2,3,5]]
df_test=df_test.iloc[:,[0,1,2,3,5]]

loc表示location的意思;iloc中的loc意思相同,前面的i表示integer,所以它只接受整数作为参数

2 查看.pt文件内容

在Pytorch中,一个约定俗成的方法是使用.pt扩展的文件格式来保存张量,使用的方法为torch.save() -- 保存张量;

在Pytorch中,使用torch.load()方法加载torch.save()方法保存的文件 -- 加载张量

使用循环查看torch保存的.pt文件网络参数 -- 查看张量文件

import torch  # 命令行是逐行立即执行的
content = torch.load('t.pth',map_location=torch.device('cpu') )
print(content.keys())   # keys()
# 之后有其他需求比如要看 key 为 model 的内容有啥
print(content['model'])

# 或者
for parameter in lz.parameters():
    print(parameter)

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值