From 33391eaf16145fb74af17c4800a17a6f71f92c85 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 26 Oct 2022 17:21:19 +0100 Subject: [PATCH] train_predict/jsonl: don't argmax I'm interested inthe raw values --- aimodel/src/subcommands/train_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_predict.py b/aimodel/src/subcommands/train_predict.py index c96ce15..3624cf8 100644 --- a/aimodel/src/subcommands/train_predict.py +++ b/aimodel/src/subcommands/train_predict.py @@ -152,7 +152,7 @@ def do_jsonl(args, ai, dataset, model_params): i_batch = 0 for water_predict in water_predict_batch: # [ width, height, softmax_probabilities ] → [ batch, width, height ] - water_predict = tf.math.argmax(water_predict, axis=-1) + # water_predict = tf.math.argmax(water_predict, axis=-1) # [ width, height ] water_actual = tf.squeeze(water_actual_batch[i_batch])