来自B站视频,API查阅,TORCH.NN文章来源地址https://www.toymoban.com/news/detail-533628.html
class dataset(Dataset):
def __init__(self):
df=pd.read_csv()
feat=df.iloc[].values
label=df.iloc[].values
self.x=torch.from_numpy(feat)
self.y=torch.from_numpy(label)
def __len(self):
return len(self.y)
def __getitem__(self,index):
return self.x[index],self.y[index]
- CE weight 用于类别不均衡情况,ignore_index 用于padding 部分,reduction 是对 batch 的操作
文章来源:https://www.toymoban.com/news/detail-533628.html
到了这里,关于pytorch笔记:数据导入实例和目标函数的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!