AI
蔚蓝的珊瑚海_xdcaxy2013
研究兴趣:J2EE,深度学习,NLP
展开
-
PyTorch获取对应类并设置相应参数
Pytorch获取类中对应字段值方法原创 2023-03-04 22:48:16 · 85 阅读 · 0 评论 -
torch.where使用总结
a = torch.randn(3, 5) print(a) b = torch.ones(3, 5) print(b) c = torch.where(a > 0, a, b) print(c)合并两个tensor,a>0的地方取a值,否则取b值原创 2021-12-28 14:48:56 · 968 阅读 · 0 评论 -
Python中dict结构根据时间字段排序方法研究
对于Python3来说,引入from functools import cmp_to_key方法。from functools import cmp_to_keydef cmp_datetime(a, b): a_datetime = datetime.datetime.strptime(a["time"], '%Y-%m-%d') b_datetime = datetime.datetime.strptime(b["time"], '%Y-%m-%d') if a_d原创 2021-11-12 14:41:10 · 786 阅读 · 0 评论 -
GPU上数据转换为一般的Python数据方法
cuda为True时候调用cuda().data.cpu().numpy().tolist()方法,CPU时候调用numpy().tolist()方法if args.cuda: pred_labels = preds.cuda().data.cpu().numpy().tolist()else: pred_labels = preds.numpy().tolist()原创 2021-09-09 12:07:31 · 352 阅读 · 0 评论 -
Python两个列表位置对齐方法研究
在NLP任务中,经常会处理序列问题,比如把一段英文字符输入某个工具,输出的列表中数字会有变化,但有时还需要和原始列表进行位置对齐,比如对齐词性信息,标签信息等。原创 2021-06-04 12:12:25 · 1795 阅读 · 4 评论 -
(常用)Python 数据保存与读取(Pickle,Json)
def save_pickle_data(save_data_dir, save_data, dataname): # 保存数据集 with open(save_data_dir + "/" + dataname + ".pickle", "wb") as handle: # pickle.dump(save_data, handle) pickle.dump(save_data, handle, protocol=pickle.HIGHEST_PROTOC.原创 2021-05-31 12:04:21 · 512 阅读 · 0 评论 -
Python保存与读取
保存:def save_obj(obj, name): with open(name + '.pkl', 'wb') as f: pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)读取:def load_obj(name): with open(name + '.pkl', 'rb') as f: tag2label = pickle.load(f)存为Jsondef predict_to_fi.原创 2020-07-29 21:39:55 · 708 阅读 · 0 评论