ImportError:无法从“数据集”导入名称“AdultDataset”

Question

请注意，我正在为分类任务处理成人数据集

我发现：from dataset import AdultDataset

正在给出以下错误：

ImportError: cannot import name 'AdultDataset' from 'dataset'

导入相关库

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.preprocessing import LabelEncoder
from sklearn.model_selection import train_test_split

import torch
import torch.nn as nn

import torch.optim as optim
from torch.autograd import Variable
from torch.utils.data import DataLoader


from dataset import AdultDataset

所以当我尝试创建一个三层前馈神经网络时 使用pytorch，将数据集条目作为输入并进行分类 如果个人收益大于或小于50K（即fnlwgt标签）

从下面的代码开始，我得到一个错误

train_dataset = AdultDataset(X_train, y_train)
test_dataset = AdultDataset(X_test, y_test)


---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-49-4ab31bdb6258> in <module>
      1 # Using norm_D01
      2 
----> 3 train_dataset = AdultDataset(X_train, y_train)
      4 test_dataset = AdultDataset(X_test, y_test)
      5 

NameError: name 'AdultDataset' is not defined

Answer 1

我相信你的问题可以这样解决：

from aif360.datasets import AdultDataset

也许你用的是一本旧指南