gradient_free_optimizers._results_manager

Complexity

Total Complexity

5

Size/Duplication

Total Lines	21
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	ResultsManager.__init__()	0	2	1
A	ResultsManager.best()	0	2	2
A	ResultsManager.dataframe()	0	3	1
A	ResultsManager.add()	0	5	1

import pandas as pd
from collections.abc import Sequence


class ResultsManager:
    def __init__(self):
        self._results_l = []

    def add(self, result, params) -> None:
        score_d = {"score": result.score}
        results_dict = {**score_d, **result.metrics}

        self._results_l.append({**results_dict, **params})

    @property
    def dataframe(self) -> pd.DataFrame:
        return pd.DataFrame(self._results_l)

    def best(self):
        return max(self._results, key=lambda r: r.score, default=None)