diff --git a/aimodel/src/subcommands/rainfall_stats.py b/aimodel/src/subcommands/rainfall_stats.py index ffa5fe5..647422e 100755 --- a/aimodel/src/subcommands/rainfall_stats.py +++ b/aimodel/src/subcommands/rainfall_stats.py @@ -61,7 +61,7 @@ def run(args): batch_stddev = tf.math.reduce_std(rainfall_flat) batch_max = tf.math.reduce_max(rainfall_flat) - print("BATCH", "mean", batch_mean, "stddev", batch_stddev, "max", batch_max) + print("BATCH mean\t", batch_mean.numpy().list()[0], "stddev\t", batch_stddev.numpy().list()[0], "max\t", batch_max.numpy().list()[0]) calc_mean.append(batch_mean) calc_stddev.append(batch_stddev)