ertk.train.ExperimentResult

class ertk.train.ExperimentResult(scores_dict: ~typing.Dict[str, ~numpy.ndarray] = <factory>, scores_df: ~pandas.core.frame.DataFrame = <factory>, predictions: ~numpy.ndarray | None = None)

Bases: object

Class to hold results of an experiment.

__init__(scores_dict: ~typing.Dict[str, ~numpy.ndarray] = <factory>, scores_df: ~pandas.core.frame.DataFrame = <factory>, predictions: ~numpy.ndarray | None = None) None

Methods

__init__([scores_dict, scores_df, predictions])

Inherited Methods

Attributes

predictions

Array of predictions.

scores_dict

Dictionary of scores.

scores_df

Scores dataframe, with a column per key in scores_dict and row per evaluation.

predictions: ndarray | None = None

Array of predictions. If not given, this is None.

scores_df: DataFrame

Scores dataframe, with a column per key in scores_dict and row per evaluation.

scores_dict: Dict[str, ndarray]

Dictionary of scores. Each key is a string and all values are either scalar or arrays.