delta-mpc / crypten_vfl_demo

vertical federated learning demo with crypten
MIT License
42 stars 6 forks source link

data_process报错: ValueError: '<=50K.' is not in list #4

Open smudag opened 2 years ago

smudag commented 2 years ago
Traceback (most recent call last):
  File "data_process.py", line 134, in <module>
    arr = convert_csv_to_arr("adult.test.csv", mean_std)
  File "data_process.py", line 102, in convert_csv_to_arr
    arr = df_to_arr(df)
  File "data_process.py", line 56, in df_to_arr
    val = attr_classes[col].index(raw_val)
ValueError: '<=50K.' is not in list
image
mh739025250 commented 2 years ago

我刚刚试了下,没有问题

你的adult.test.csv长什么样?可以发来看看么

wuzh3nq1 commented 2 years ago

需要把test中的50K.替换为50K