backend-junPreP Update(Update whitelist model), front-urlpredictor.jsx UI Update
This commit is contained in:
@@ -44,7 +44,7 @@ def predict_url(url: str) -> dict:
|
||||
input_data = preprocessed[features_cols]
|
||||
|
||||
# ✅ 전처리된 데이터 확인
|
||||
print("Preprocessed input:", input_data)
|
||||
#print("Preprocessed input:", input_data)
|
||||
|
||||
# 평균 확률 계산
|
||||
probs = [float(model.predict_proba(input_data)[0, 1]) for model in models_load]
|
||||
@@ -61,8 +61,8 @@ def predict_url(url: str) -> dict:
|
||||
# 예: malicious_probability가 np.float32 타입일 경우
|
||||
return {
|
||||
"url": url,
|
||||
"malicious_probability": mean_pred, # ⬅️ numpy -> float
|
||||
"is_malicious": bool(is_malicious), # ⬅️ numpy -> bool
|
||||
"malicious_probability": mean_pred,
|
||||
"is_malicious": is_malicious,
|
||||
"threshold": float(BEST_THRESHOLD) # ⬅️ numpy -> float
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user