wehub-resource-sync
53 lines
1.7 KiB
C
53 lines
1.7 KiB
C
/*
|
|
Tests our OutlierDetector
|
|
|
|
compile and run as (from dev/test directory)
|
|
gcc -O3 -I../../llmc -o test_outlier_detector test_outlier_detector.c -lm && ./test_outlier_detector
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include "../../llmc/outlier_detector.h"
|
|
|
|
int main(void) {
|
|
OutlierDetector detector;
|
|
init_detector(&detector);
|
|
|
|
srand(1337); // init rng
|
|
|
|
// generate OUTLIER_DETECTOR_WINDOW_SIZE * 2 random numbers between -1 and 1
|
|
for (int i = 0; i < OUTLIER_DETECTOR_WINDOW_SIZE * 2; i++) {
|
|
double val = (double)rand() / RAND_MAX * 2 - 1; // Random number between -1 and 1
|
|
double zscore = update_detector(&detector, val);
|
|
|
|
printf("Step %d: Value = %.4f, zscore = %.4f\n", i, val, zscore);
|
|
|
|
// check that the first OUTLIER_DETECTOR_WINDOW_SIZE values return nan
|
|
if (i < OUTLIER_DETECTOR_WINDOW_SIZE) {
|
|
if (!isnan(zscore)) {
|
|
printf("Error: Expected nan, got %.4f\n", zscore);
|
|
return EXIT_FAILURE;
|
|
}
|
|
} else {
|
|
// check that the zscore is within reasonable bounds
|
|
if (zscore < -3.0 || zscore > 3.0) {
|
|
printf("Error: Z-score %.4f is outside of expected range\n", zscore);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
}
|
|
|
|
// simulate an outlier
|
|
double outlier = 10.0; // <--- loss spike
|
|
double zscore = update_detector(&detector, outlier);
|
|
printf("Outlier Step: Value = %.4f, zscore = %.4f\n", outlier, zscore);
|
|
|
|
// check that the z-score here is large
|
|
if (zscore < 5.0) {
|
|
printf("Error: Z-score %.4f is not large enough for an outlier\n", zscore);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
printf("OK\n");
|
|
return EXIT_SUCCESS;
|
|
}
|