142 lines
5.1 KiB
Python
142 lines
5.1 KiB
Python
import pandas as pd
|
|
from datetime import datetime, timedelta
|
|
import numpy as np
|
|
from xgboost import XGBClassifier
|
|
from sklearn.ensemble import StackingClassifier, RandomForestClassifier
|
|
from sklearn.neighbors import KNeighborsClassifier
|
|
from sklearn.naive_bayes import GaussianNB
|
|
from sklearn.svm import SVC
|
|
from sklearn.linear_model import LogisticRegression
|
|
from sklearn.metrics import precision_score, recall_score, f1_score, roc_auc_score, accuracy_score
|
|
from sklearn.preprocessing import MinMaxScaler, StandardScaler
|
|
from sklearn.decomposition import PCA
|
|
import lightgbm as lgb
|
|
|
|
from tqdm import tqdm
|
|
from collections import defaultdict
|
|
import asyncio
|
|
import aiohttp
|
|
import aiofiles
|
|
import pickle
|
|
import time
|
|
import os
|
|
|
|
|
|
class ScorePredictor:
|
|
def __init__(self):
|
|
self.scaler = StandardScaler()
|
|
self.pca = PCA(n_components=0.95)
|
|
|
|
# Define base models
|
|
self.xgb_model = XGBClassifier(
|
|
n_estimators=100,
|
|
max_depth=10,
|
|
learning_rate=0.001,
|
|
random_state=42,
|
|
n_jobs=10,
|
|
tree_method='gpu_hist',
|
|
)
|
|
'''
|
|
self.lgb_model = lgb.LGBMClassifier(
|
|
n_estimators=100,
|
|
learning_rate=0.001,
|
|
max_depth=10,
|
|
n_jobs=10
|
|
)
|
|
'''
|
|
self.rf_model = RandomForestClassifier(
|
|
n_estimators=100,
|
|
max_depth=10,
|
|
random_state=42,
|
|
n_jobs=10
|
|
)
|
|
|
|
self.svc_model = SVC(probability=True, kernel='rbf')
|
|
self.knn_model = KNeighborsClassifier(n_neighbors=5)
|
|
self.nb_model = GaussianNB()
|
|
|
|
|
|
# Stacking ensemble (XGBoost + LightGBM) with Logistic Regression as meta-learner
|
|
self.model = StackingClassifier(
|
|
estimators=[
|
|
('xgb', self.xgb_model),
|
|
#('lgb', self.lgb_model),
|
|
('rf', self.rf_model),
|
|
('svc', self.svc_model),
|
|
('knn', self.knn_model),
|
|
('nb', self.nb_model)
|
|
],
|
|
final_estimator=LogisticRegression(),
|
|
n_jobs=10
|
|
)
|
|
|
|
self.warm_start_model_path = 'ml_models/weights/ai-score/stacking_weights.pkl'
|
|
|
|
def preprocess_train_data(self, X):
|
|
X = np.where(np.isinf(X), np.nan, X)
|
|
X = np.nan_to_num(X)
|
|
X = self.scaler.fit_transform(X)
|
|
return self.pca.fit_transform(X)
|
|
|
|
def preprocess_test_data(self, X):
|
|
X = np.where(np.isinf(X), np.nan, X)
|
|
X = np.nan_to_num(X)
|
|
X = self.scaler.transform(X)
|
|
return self.pca.transform(X)
|
|
|
|
def warm_start_training(self, X_train, y_train):
|
|
X_train = self.preprocess_train_data(X_train)
|
|
if os.path.exists(self.warm_start_model_path):
|
|
with open(self.warm_start_model_path, 'rb') as f:
|
|
self.model = pickle.load(f)
|
|
self.model.fit(X_train, y_train)
|
|
pickle.dump(self.model, open(self.warm_start_model_path, 'wb'))
|
|
print("Warm start model saved.")
|
|
|
|
def fine_tune_model(self, X_train, y_train):
|
|
X_train = self.preprocess_train_data(X_train)
|
|
with open(self.warm_start_model_path, 'rb') as f:
|
|
self.model = pickle.load(f)
|
|
|
|
self.model.fit(X_train, y_train)
|
|
print("Model fine-tuned")
|
|
|
|
def evaluate_model(self, X_test, y_test):
|
|
X_test = self.preprocess_test_data(X_test)
|
|
test_predictions = self.model.predict_proba(X_test)
|
|
class_1_probabilities = test_predictions[:, 1]
|
|
binary_predictions = (class_1_probabilities >= 0.5).astype(int)
|
|
|
|
# Calculate and print metrics
|
|
test_precision = precision_score(y_test, binary_predictions)
|
|
test_accuracy = accuracy_score(y_test, binary_predictions)
|
|
test_f1_score = f1_score(y_test, binary_predictions)
|
|
test_recall_score = recall_score(y_test, binary_predictions)
|
|
test_roc_auc_score = roc_auc_score(y_test, binary_predictions)
|
|
|
|
print(f"Test Precision: {round(test_precision * 100)}%")
|
|
print(f"Test Accuracy: {round(test_accuracy * 100)}%")
|
|
print(f"F1 Score: {round(test_f1_score * 100)}%")
|
|
print(f"Recall: {round(test_recall_score * 100)}%")
|
|
print(f"ROC AUC: {round(test_roc_auc_score * 100)}%")
|
|
|
|
last_prediction_prob = class_1_probabilities[-1]
|
|
print(f"Last prediction probability: {last_prediction_prob}")
|
|
|
|
thresholds = [0.8, 0.75, 0.7, 0.6, 0.5, 0.45, 0.4, 0.35, 0.3, 0]
|
|
scores = [10, 9, 8, 7, 6, 5, 4, 3, 2, 1]
|
|
|
|
score = None
|
|
for threshold, value in zip(thresholds, scores):
|
|
if last_prediction_prob >= threshold:
|
|
score = value
|
|
break
|
|
|
|
return {
|
|
'accuracy': round(test_accuracy * 100),
|
|
'precision': round(test_precision * 100),
|
|
'f1_score': round(test_f1_score * 100),
|
|
'recall_score': round(test_recall_score * 100),
|
|
'roc_auc_score': round(test_roc_auc_score * 100),
|
|
'score': score
|
|
} |