Advertisement
Guest User

Untitled

a guest
Dec 10th, 2019
112
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.70 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.ensemble import RandomForestClassifier
  3. from sklearn.model_selection import train_test_split
  4. from sklearn.preprocessing import LabelEncoder
  5.  
  6. def fraud_pred(t_type, amount):
  7. data = pd.read_csv('dataset.csv')
  8.  
  9. lab_encoder = LabelEncoder().fit(data["type"])
  10. data["type"] = lab_encoder.transform(data["type"])
  11.  
  12. X, y = data[["type", "amount"]], data["isFraud"]
  13. algorithm = RandomForestClassifier(n_estimators=100, max_depth=3)
  14. algorithm.fit(X, y)
  15. prediction = algorithm.predict([[lab_encoder.transform([t_type])[0], amount]])[0]
  16. print(prediction)
  17. return "Fraud" if prediction == 1 else "Not fraud"
  18.  
  19. fraud_pred("TRANSFER", 50000000.00)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement