Fixed axis variable usage (#37)

DiogenesAnalytics · DiogenesAnalytics · commit 18794e88f6fc · 2024-01-21T17:30:19.000-05:00
diff --git a/src/autoencoder/training.py b/src/autoencoder/training.py
@@ -28,7 +28,7 @@ def anomaly_diff(
         """Calculates mean training/anomalous data reconstruction error difference."""
         # calculate the dynamic mean reconstruction error on training data
         train_reconstruction_errors = tf.reduce_mean(
-            tf.square(y_true - y_pred), axis=(1, 2, 3)
+            tf.square(y_true - y_pred), axis=axis
         )
         dynamic_threshold = tf.reduce_mean(train_reconstruction_errors)
 

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def anomaly_diff(`
`28`	`28`	`"""Calculates mean training/anomalous data reconstruction error difference."""`
`29`	`29`	`# calculate the dynamic mean reconstruction error on training data`
`30`	`30`	`train_reconstruction_errors = tf.reduce_mean(`
`31`		`- tf.square(y_true - y_pred), axis=(1, 2, 3)`
	`31`	`+ tf.square(y_true - y_pred), axis=axis`
`32`	`32`	`)`
`33`	`33`	`dynamic_threshold = tf.reduce_mean(train_reconstruction_errors)`
`34`	`34`