File size: 867 Bytes
72cd03d d3536e2 72cd03d d3536e2 72cd03d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
from flask import Flask, request, jsonify
from sklearn.tree import DecisionTreeClassifier
import numpy as np
app = Flask(__name__)
# Basit örnek veriler (Bu veriler yerel olarak tutulabilir)
# Örnek veri formatı: [phishing_response, ransomware_response, ddos_response, malware_response]
example_data = [
[0, 0, 1, 1],
[1, 1, 0, 0],
[1, 0, 1, 0],
[0, 1, 0, 1]
]
# Örnek sonuçlar: 1 = güvenli, 0 = güvensiz
example_labels = [1, 0, 1, 0]
# Model eğitimi
model = DecisionTreeClassifier()
model.fit(example_data, example_labels)
@app.route('/analyze', methods=['POST'])
def analyze():
data = request.json
response = [data['phishing'], data['ransomware'], data['ddos'], data['malware']]
prediction = model.predict([response])
return jsonify({'prediction': int(prediction[0])})
if __name__ == '__main__':
app.run(debug=True) |