pytorch
xiao___qiao
计科在读硕士,在自己的方向上,写一些自己的理解
展开
-
处理数据
# 做成集合 去重set()# 排序sorted()# 矩阵转置input.t()原创 2021-03-05 17:47:23 · 54 阅读 · 0 评论 -
pyroch rnn
# hello -> ohlolimport torchfrom torch.nn import Moduleclass model_rnn(Module): def __init__(self, input_size, hidden_size, batch_size, num_layers): super(model_rnn,self).__init__() # 多继承时,防止父类多次调用 self.num_layers = num_layers原创 2021-03-05 12:41:45 · 143 阅读 · 0 评论 -
pytorch 安装
创建虚拟环境conda create -n pytorch python=3.6激活环境conda activate pytorch退出环境conda deactivate查看工具包pip list查看版本python –version更新conda update python安装conda install python=3.8jupyter的安装conda install nb_condadir() 打开help() 说明说dataset?? 解释...原创 2021-03-04 22:46:30 · 152 阅读 · 1 评论 -
tensorboard 使用
from torch.utils.tensorboard import SummaryWriterwriter =SummaryWriter("logs")for i in range(100): writer.add_scalar("y=2x", 2*i, i)writer.close()# 命令行输入 tensorboard --logdir=logs --port=6007原创 2021-03-04 22:40:38 · 110 阅读 · 1 评论 -
plt、nn 使用
import pandas as pdimport numpy as npimport matplotlib.pyplot as pltdata = pd.read_csv('csv_data/test.csv')data.head(3) # 查看前3行data.info() # 查看类型、是否空plt.scatter(data.num,data.debt) # 散点图plt.xlabel('num')plt.ylabel('debt')from torch import nni原创 2021-03-04 22:39:34 · 231 阅读 · 1 评论 -
dataset 加载 image、label
from torch.utils.data import Datasetimport osfrom PIL import Imageclass MyData(Dataset): def __init__(self, root_dir, image_dir, label_dir): self.root_dir = root_dir self.image_dir = image_dir self.label_dir = label_dir原创 2021-03-04 22:36:52 · 695 阅读 · 1 评论 -
查询函数用法
原创 2021-03-04 22:35:02 · 195 阅读 · 1 评论 -
tensor 计算
import torcha1 = torch.rand(2,3) # 随机2x3的tensora2 = torch.randn(3, 4)a3 = torch.zeros(1, 4)a4 = torch.ones((1, 2), dtype=torch.int64)print(a2.size())print(a2.shape)print(a2.size(0)) # 0维度a5 = torch.tensor([6, 2], dtype=torch.float32)a5.type() #原创 2021-03-04 22:32:55 · 165 阅读 · 1 评论 -
dataset、dataloader 加载 txt
数据格式from torch.utils.data import Dataset,DataLoaderimport numpy as npimport torch重载Datasetclass mydata(Dataset): def __init__(self, path): self.xy = np.loadtxt(path,delimiter=',',dtype=np.float32) self.x_data = torch.from_numpy(s原创 2021-03-04 22:25:03 · 710 阅读 · 1 评论