如何在 Python 中定义 split_train_test？

Question

我正在编写 python 代码以使用 GAN 生成网络流量，但我收到此错误 split_train_test 未定义。我从 sklearn.model_selection 导入了 train_test_split，但它似乎不起作用。我做错了什么？这是错误信息；

NameError                                 Traceback (most recent call last)
<ipython-input-153-a2836ba27bc4> in <module>
      9     cross_validation_flg = False
     10     benign_file = '../data/attack_normal_data/benign_data.csv'
---> 11     benign_model, benign_test_loader = run_main(benign_file, num_features=41)
     12     # Save the model checkpoint
     13     torch.save(benign_model.state_dict(), 'benign_model_epoches%d.ckpt' % num_epochs)

<ipython-input-147-e59bfccfe2c7> in run_main(input_file, num_features)
      5     dataset = TrafficDataset(input_file, transform=None, normalization_flg=True)
      6 
----> 7     train_sampler, test_sampler = split_train_test(dataset, split_percent=0.7, shuffle=True)
      8     cntr = Counter(dataset.y)
      9     print('dataset: ', len(dataset), ' y:', sorted(cntr.items()))

NameError: name 'split_train_test' is not defined

Answer 1

应该是train_test_split

参考docs

如何在 Python 中定义 split_train_test？

How do I define the split_train_test in Python?

python-3.8