pytorch中数据集的划分方法及eError: take(): argument 'index' (position 1) must be Tensor, not numpy.ndarray错误原因
在使用pytorch框架时,难免需要对数据集进行训练集和验证集的划分,一般使用sklearn.model_selection中的train_test_split方法该方法使用如下:fromsklearn.model_selectionimporttrain_test_splitimportnumpyasnpimporttorchimporttorch.autogradimportVariablef