Spaces:

TwanAPI
/

trsstapi

Paused

trsstapi / src /security /anomalyEngine.js

Create src/security/anomalyEngine.js

22f5471 verified 11 months ago

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;
}

	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;
	}