diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index 740de294e4b447f57ed469b199d74422d6f14462..d3e7a5fe646b64c95cf81d8ac800065956a3da0b 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -29,7 +29,7 @@ class DAN:
     The class initializes useful parameters: the device and the temperature scalara parameter.
     """
 
-    def __init__(self, device, temperature):
+    def __init__(self, device, temperature=1.0):
         """
         Constructor of the DAN class.
         :param device: The device to use.