search

ZhaoZhibin / DL-based-Intelligent-Diagnosis-Benchmark

Source codes for the paper "Deep Learning Algorithms for Rotating Machinery Intelligent Diagnosis: An Open Source Benchmark Study"
MIT License
611 stars 167 forks source link

--processing_type O_A doesn't work #24

Open LeeJMJM opened 9 months ago

LeeJMJM commented 9 months ago

When using CWRU dataset, If this option is O_A, the train.py doesn't work. If this option is R_A or R_NA, it works.

File "c:/Users/jashm/OneDrive/桌面/DL-based-Intelligent-Diagnosis-Benchmark-master/DL-based-Intelligent-Diagnosis-Benchmark-master/train.py", line 65, in trainer.setup() File "c:\Users\jashm\OneDrive\桌面\DL-based-Intelligent-Diagnosis-Benchmark-master\DL-based-Intelligent-Diagnosis-Benchmark-master\utils\train_utils.py", line 55, in setup self.datasets['train'], self.datasets['val'] = Dataset(args.data_dir,args.normlizetype).data_preprare() File "c:\Users\jashm\OneDrive\桌面\DL-based-Intelligent-Diagnosis-Benchmark-master\DL-based-Intelligent-Diagnosis-Benchmark-master\CNN_Datasets\O_A\datasets\CWRUFFT.py", line 144, in data_preprare train_pd, val_pd = train_test_split_order(data_pd, test_size=0.2, num_classes= 10) File "c:\Users\jashm\OneDrive\桌面\DL-based-Intelligent-Diagnosis-Benchmark-master\DL-based-Intelligent-Diagnosis-Benchmark-master\CNN_Datasets\O_A\datasets\CWRUFFT.py", line 121, in train_test_split_order train_pd = pd.DataFrame(columns=('data', 'label')) File "C:\Users\jashm\anaconda3\envs\env1\lib\site-packages\pandas\core\frame.py", line 392, in init mgr = init_dict(data, index, columns, dtype=dtype) File "C:\Users\jashm\anaconda3\envs\env1\lib\site-packages\pandas\core\internals\construction.py", line 196, in init_dict nan_dtype) File "C:\Users\jashm\anaconda3\envs\env1\lib\site-packages\pandas\core\dtypes\cast.py", line 1175, in construct_1d_arraylike_from_scalar dtype = dtype.dtype AttributeError: type object 'object' has no attribute 'dtype'