From 9bf7229b1e1f35cb23b5e844adcf268c4228724b Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 21:14:45 +0100 Subject: [PATCH] Added TODOs to train module Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index fabcb0a..9db111b 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -30,6 +30,10 @@ Functions: prepare_training_data(...): prepares the mnist training data train(...): trains the AAE models +Todos: + - fix early stopping + - fix losses reaching exactly zero + """ import functools @@ -621,7 +625,7 @@ def _normalize(feature: tf.Tensor, label: tf.Tensor) -> Tuple[tf.Tensor, tf.Tens if __name__ == "__main__": tf.enable_eager_execution() - inlier_classes = [0] + inlier_classes = [3] iteration = 1 train_dataset, _ = prepare_training_data(test_fold_id=0, inlier_classes=inlier_classes, total_classes=10)