search

aigc-apps / AMFormer

The AMFormer algorithm, accepted at AAAI-2024, for deep tabular learning
GNU General Public License v3.0
25 stars 7 forks source link

valid_X为未定义的变量： #13

Open Cgetier520990 opened 2 months ago

Cgetier520990 commented 2 months ago

在data_hcder.py中，在 init 方法中，如果 dataset == 'valid' 时，代码中引用了 valid_X 和 valid_Y，但这些变量没有定义。

Ch3ngY1 commented 2 months ago

可以修改以下内容 if dataset in ['train', 'test']: train_X, test_X, train_Y, test_Y = train_test_split(data, labels, test_size=0.2, random_state=args.seed) if args.val_split: trainX, , trainY, = train_test_split(train_X, train_Y, test_size=0.185, random_state=args.seed) 来规定你的train和val数据的划分