擅长:python、mysql、java
<p>只需转换<code>pandas dataframe -> numpy array -> pytorch tensor</code>。下面介绍一个这样的例子:</p>
<pre><code>import pandas as pd
import numpy as np
import torch
df = pd.read_csv('train.csv')
target = pd.DataFrame(df['target'])
del df['target']
train = data_utils.TensorDataset(torch.Tensor(np.array(df)), torch.Tensor(np.array(target)))
train_loader = data_utils.DataLoader(train, batch_size = 10, shuffle = True)
</code></pre>
<p>希望这将帮助您使用pytorch(与最新版本的pytorch兼容)创建自己的数据集。</p>