is the below code correct? and then how can I load the data ?
x_train = train_data[train_data.targets == 1]
y_train = train_data.targets[train_data.targets==1]

x_test = test_data[test_data.targets != 1]
y_test = test_data.targets[test_data.targets!=1]