Spaces:
Sleeping
Sleeping
rmm
commited on
Commit
·
ba36d57
1
Parent(s):
43cae2c
feat: added member var for top predictions in the observation class
Browse files
src/input/input_observation.py
CHANGED
|
@@ -56,6 +56,16 @@ class InputObservation:
|
|
| 56 |
self.date_option = date_option
|
| 57 |
self.time_option = time_option
|
| 58 |
self.uploaded_filename = uploaded_filename
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
|
| 60 |
def __str__(self):
|
| 61 |
return f"Observation: {self.image}, {self.latitude}, {self.longitude}, {self.author_email}, {self.date}, {self.time}, {self.date_option}, {self.time_option}, {self.uploaded_filename}"
|
|
|
|
| 56 |
self.date_option = date_option
|
| 57 |
self.time_option = time_option
|
| 58 |
self.uploaded_filename = uploaded_filename
|
| 59 |
+
self._top_predictions = []
|
| 60 |
+
|
| 61 |
+
def set_top_predictions(self, top_predictions:list):
|
| 62 |
+
self._top_predictions = top_predictions
|
| 63 |
+
|
| 64 |
+
# add a method to get the top predictions (property?)
|
| 65 |
+
@property
|
| 66 |
+
def top_predictions(self):
|
| 67 |
+
return self._top_predictions
|
| 68 |
+
|
| 69 |
|
| 70 |
def __str__(self):
|
| 71 |
return f"Observation: {self.image}, {self.latitude}, {self.longitude}, {self.author_email}, {self.date}, {self.time}, {self.date_option}, {self.time_option}, {self.uploaded_filename}"
|