Data-Driven Fault Diagnosis for Rolling Bearing Based on DIT-FFT and XGBoost

Table 2

Training set and test set.

Damage diameter (inches)	Fault type	Training set		Test set		Label
		Group	Length	Group	Length
0.007	RE fault	50	102400	9	18432	0
	IR fault	50	102400	9	18432	1
	OR fault	50	102400	9	18432	2
0.014	RE fault	50	102400	9	18432	3
	IR fault	50	102400	9	18432	4
	OR fault	50	102400	9	18432	5
0.021	RE fault	50	102400	9	18432	6
	IR fault	50	102400	9	18432	7
	OR fault	50	102400	9	18432	8
0	Normal state	50	102400	186	380928	9