Skip to content

Commit

Permalink
[DOC] LITE Time classifier metrics (#2464)
Browse files Browse the repository at this point in the history
State that Keras Metrics class objects (and functions) can be used. Also link to
the Keras documentation.
  • Loading branch information
dschrempf authored Dec 18, 2024
1 parent 5c7bdad commit c8efd14
Showing 1 changed file with 14 additions and 10 deletions.
24 changes: 14 additions & 10 deletions aeon/classification/deep_learning/_lite_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,11 +96,13 @@ class LITETimeClassifier(BaseClassifier):
The keras optimizer used for training.
loss : str, default = "categorical_crossentropy"
The name of the keras training loss.
metrics : str or list[str], default="accuracy"
The evaluation metrics to use during training. If
a single string metric is provided, it will be
used as the only metric. If a list of metrics are
provided, all will be used for evaluation.
metrics : str or list[str|function|keras.metrics.Metric], default="accuracy"
the evaluation metrics to use during training. each of this can be a
string, function or a keras.metrics.metric instance (for details, see
https://keras.io/api/metrics/).
if a single string metric is provided, it will be used as the only
metric. if a list of metrics are provided, all will be used for
evaluation.
References
----------
Expand Down Expand Up @@ -398,11 +400,13 @@ class IndividualLITEClassifier(BaseDeepClassifier):
The keras optimizer used for training.
loss : str, default = "categorical_crossentropy"
The name of the keras training loss.
metrics : str or list[str], default="accuracy"
The evaluation metrics to use during training. If
a single string metric is provided, it will be
used as the only metric. If a list of metrics are
provided, all will be used for evaluation.
metrics : str or list[str|function|keras.metrics.Metric], default="accuracy"
the evaluation metrics to use during training. each of this can be a
string, function or a keras.metrics.metric instance (for details, see
https://keras.io/api/metrics/).
if a single string metric is provided, it will be used as the only
metric. if a list of metrics are provided, all will be used for
evaluation.
References
----------
Expand Down

0 comments on commit c8efd14

Please sign in to comment.