reload the __str__ and __repr__ to print the model summary

zjuwss · Mar 17, 2024 · 748a5df · 748a5df
1 parent 9df75e6
commit 748a5df
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/gnnwr/models.py b/src/gnnwr/models.py
@@ -795,6 +795,14 @@ def getCoefs(self):
         result_data.set_index('id', inplace=True)
         result_data = result_data.join(data)
         return result_data
+    def __str__(self) -> str:
+        print("Model Name: ", self._modelName)
+        print("Model Structure: ", self._model)
+        return ""
+    def __repr__(self) -> str:
+        print("Model Name: ", self._modelName)
+        print("Model Structure: ", self._model)
+        return ""
 
 
 class GTNNWR(GNNWR):