trsstapi / src /security /anomalyEngine.js
Twan07's picture
Create src/security/anomalyEngine.js
22f5471 verified
raw
history blame
288 Bytes
import IsolationForest from "ml-isolation-forest";
let model = null;
export function trainModel(data) {
model = new IsolationForest({ contamination: 0.02 });
model.fit(data);
}
export function isAnomaly(vec) {
if (!model) return false;
return model.predict([vec])[0] === -1;
}