Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- X_train, X_test = train_test_split(Normalized_df, test_size=0.2, random_state=RANDOM_SEED)
- X_train = X_train[X_train.Class == 0] #only normal transactions
- X_train = X_train.drop(['Class'], axis=1) #droping class (output) column
- Y_test = X_test['Class'] #save the class column for the test set
- X_test = X_test.drop(['Class'], axis=1) #drop the class column
- X_train = X_train.values #transform to ndarray
- X_test = X_test.values
- X_train.shape
Add Comment
Please, Sign In to add comment