From 152f00fb624c135f93a849035ede5bbded72bf7d Mon Sep 17 00:00:00 2001
From: Le Roux Erwan <erwan.le-roux@irstea.fr>
Date: Tue, 5 Mar 2019 15:53:16 +0100
Subject: [PATCH] [EXTREME ESTIMATOR] Add is_convergence_successful field to
ResultFromFit object
---
extreme_estimator/extreme_models/result_from_fit.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/extreme_estimator/extreme_models/result_from_fit.py b/extreme_estimator/extreme_models/result_from_fit.py
index b8a8d7d1..f2a41d50 100644
--- a/extreme_estimator/extreme_models/result_from_fit.py
+++ b/extreme_estimator/extreme_models/result_from_fit.py
@@ -21,9 +21,13 @@ class ResultFromFit(object):
return self.name_to_value.keys()
@property
- def convergence(self):
+ def convergence(self) -> str:
convergence_value = self.name_to_value[self.CONVERGENCE_NAME]
- return convergence_value
+ return convergence_value[0]
+
+ @property
+ def is_convergence_successful(self) -> bool:
+ return self.convergence == "successful"
@property
def fitted_values(self) -> Dict[str, float]:
--
GitLab