test_data_x = data.loc[:, (data.columns != 'Label') & (data.columns != 'Target gene')].iloc[test_index, :].astype(np.float32)
    test_data_y = data.loc[:, data.columns == 'Label'].iloc[test_index, :].astype(np.int32)
train_data_x = data.loc[:, (data.columns != 'Label') & (data.columns != 'Targetgene')].iloc[train_index,:].astype(np.float32)
    train_data_y = data.loc[:, data.columns == 'Label'].iloc[train_index, :].astype(np.int32)