Guest User

Untitled

a guest
Dec 19th, 2018
107
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.43 KB | None | 0 0
  1. X_train, X_test = train_test_split(Normalized_df, test_size=0.2, random_state=RANDOM_SEED)
  2.  
  3. X_train = X_train[X_train.Class == 0] #only normal transactions
  4. X_train = X_train.drop(['Class'], axis=1) #droping class (output) column
  5.  
  6.  
  7. Y_test = X_test['Class'] #save the class column for the test set
  8. X_test = X_test.drop(['Class'], axis=1) #drop the class column
  9.  
  10. X_train = X_train.values #transform to ndarray
  11. X_test = X_test.values
  12.  
  13. X_train.shape
Add Comment
Please, Sign In to add comment