diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index 5ba6640..231e6fd 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -250,7 +250,6 @@ if PATH_CHECKPOINT is None:
 			mean_iou(),
 			sensitivity(), # How many true positives were accurately predicted
 			specificity # How many true negatives were accurately predicted?
-			# TODO: Add IoU, F1, Precision, Recall,  here. 
 		],
 		steps_per_execution=STEPS_PER_EXECUTION,
 		jit_compile=JIT_COMPILE