Skip to content

Commit ed28732

Browse files
kevinzb56baraline
andauthored
Documentation improvement of certain BaseClasses (aeon-toolkit#2516)
Co-authored-by: Antoine Guillaume <antoine.guillaume45@gmail.com>
1 parent 0dfe3ca commit ed28732

File tree

2 files changed

+38
-1
lines changed

2 files changed

+38
-1
lines changed

aeon/base/_base_collection.py

+21-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,24 @@
1-
"""Base class for estimators that fit collections of time series."""
1+
"""
2+
Base class for estimators that fit collections of time series.
3+
4+
class name: BaseCollectionEstimator
5+
6+
Defining methods:
7+
preprocessing - _preprocess_collection(self, X, store_metadata=True)
8+
input checking - _check_X(self, X)
9+
input conversion - _convert_X(self, X)
10+
shape checking - _check_shape(self, X)
11+
12+
Inherited inspection methods:
13+
hyper-parameter inspection - get_params()
14+
fitted parameter inspection - get_fitted_params()
15+
16+
State:
17+
fitted model/strategy - by convention, any attributes ending in "_"
18+
fitted state flag - is_fitted (property)
19+
fitted state inspection - check_is_fitted()
20+
21+
"""
222

323
from abc import abstractmethod
424

aeon/classification/deep_learning/base.py

+17
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,23 @@
11
"""
22
Abstract base class for the Keras neural network classifiers.
33
4+
class name: BaseDeepClassifier
5+
6+
Defining methods:
7+
fitting - fit(self, X, y)
8+
predicting - predict(self, X)
9+
- predict_proba(self, X)
10+
model building - build_model(self, input_shape, n_classes) (abstract method)
11+
12+
Inherited inspection methods:
13+
hyper-parameter inspection - get_params()
14+
fitted parameter inspection - get_fitted_params()
15+
16+
State:
17+
fitted model/strategy - by convention, any attributes ending in "_"
18+
fitted state flag - is_fitted (property)
19+
fitted state inspection - check_is_fitted()
20+
421
The reason for this class between BaseClassifier and deep_learning classifiers is
522
because we can generalise tags, _predict and _predict_proba
623
"""

0 commit comments

Comments
 (0)