Commit 152f00fb authored by Le Roux Erwan's avatar Le Roux Erwan
Browse files

[EXTREME ESTIMATOR] Add is_convergence_successful field to ResultFromFit object

parent 8d081557
No related merge requests found
Showing with 6 additions and 2 deletions
+6 -2
......@@ -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]:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment