Junaid-EEE11 commited on
Commit
68af965
·
1 Parent(s): 02b1dab

Update models/fraud_detection_model.py

Browse files
Files changed (1) hide show
  1. models/fraud_detection_model.py +10 -0
models/fraud_detection_model.py CHANGED
@@ -8,3 +8,13 @@ def predict_fraud(model, data):
8
  # Placeholder for model prediction logic
9
  # Replace with actual prediction logic
10
  return model.predict_classes(data)
 
 
 
 
 
 
 
 
 
 
 
8
  # Placeholder for model prediction logic
9
  # Replace with actual prediction logic
10
  return model.predict_classes(data)
11
+ from tensorflow.keras.models import Sequential
12
+ from tensorflow.keras.layers import Dense, Dropout
13
+
14
+ def build_model(input_dim):
15
+ model = Sequential()
16
+ model.add(Dense(64, input_dim=input_dim, activation='relu'))
17
+ model.add(Dropout(0.5))
18
+ model.add(Dense(1, activation='sigmoid'))
19
+ model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
20
+ return model