MahdiFattahi's picture
Upload model.py with huggingface_hub
cca251b verified
Raw
History Blame Contribute Delete
227 Bytes
import numpy as np
from sklearn.ensemble import IsolationForest
def train_anomaly_detector(features):
model = IsolationForest(n_estimators=100, contamination=0.05, random_state=42)
model.fit(features)
return model