X_train, X_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state = 2020, stratify=y) X_train_user = X_train[y_train == \'ji2hwh\'] X_train_a