Skip to content

Commit

Permalink
Removed list from lite (#2510)
Browse files Browse the repository at this point in the history
  • Loading branch information
kavya-r30 authored Jan 25, 2025
1 parent 6ee0597 commit 06cde8b
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 30 deletions.
20 changes: 8 additions & 12 deletions aeon/classification/deep_learning/_lite_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,12 +35,10 @@ class LITETimeClassifier(BaseClassifier):
if set to `True` then LITEMV is used. LITEMV is the
same architecture as LITE but specifically designed
to better handle multivariate time series.
n_filters : int or list of int32, default = 32
The number of filters used in one lite layer, if not a list, the same
number of filters is used in all lite layers.
kernel_size : int or list of int, default = 40
The head kernel size used for each lite layer, if not a list, the same
is used in all lite module.
n_filters : int, default = 32
The number of filters used in one lite layer.
kernel_size : int, default = 40
The head kernel size used for each lite layer.
strides : int or list of int, default = 1
The strides of kernels in convolution layers for each lite layer,
if not a list, the same is used in all lite layers.
Expand Down Expand Up @@ -340,12 +338,10 @@ class IndividualLITEClassifier(BaseDeepClassifier):
if set to `True` then LITEMV is used. LITEMV is the
same architecture as LITE but specifically designed
to better handle multivariate time series.
n_filters : int or list of int32, default = 32
The number of filters used in one lite layer, if not a list, the same
number of filters is used in all lite layers.
kernel_size : int or list of int, default = 40
The head kernel size used for each lite layer, if not a list, the same
is used in all lite layers.
n_filters : int, default = 32
The number of filters used in one lite layer.
kernel_size : int, default = 40
The head kernel size used for each lite layer.
strides : int or list of int, default = 1
The strides of kernels in convolution layers for each lite layer,
if not a list, the same is used in all lite layers.
Expand Down
10 changes: 4 additions & 6 deletions aeon/networks/_lite.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,12 +22,10 @@ class LITENetwork(BaseDeepLearningNetwork):
if set to `True` then LITEMV is used. LITEMV is the
same architecture as LITE but specifically designed
to better handle multivariate time series.
n_filters : int or list of int32, default = 32
The number of filters used in one lite layer, if not a list, the same
number of filters is used in all lite layers.
kernel_size : int or list of int, default = 40
The head kernel size used for each lite layer, if not a list, the same
is used in all lite layers.
n_filters : int, default = 32
The number of filters used in one lite layer.
kernel_size : int , default = 40
The head kernel size used for each lite layer,.
strides : int or list of int, default = 1
The strides of kernels in convolution layers for each lite layer,
if not a list, the same is used in all lite layers.
Expand Down
20 changes: 8 additions & 12 deletions aeon/regression/deep_learning/_lite_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,12 +34,10 @@ class LITETimeRegressor(BaseRegressor):
if set to `True` then LITEMV is used. LITEMV is the
same architecture as LITE but specifically designed
to better handle multivariate time series.
n_filters : int or list of int32, default = 32
The number of filters used in one lite layer, if not a list, the same
number of filters is used in all lite layers.
kernel_size : int or list of int, default = 40
The head kernel size used for each lite layer, if not a list, the same
is used in all lite module.
n_filters : int, default = 32
The number of filters used in one lite layer.
kernel_size : int, default = 40
The head kernel size used for each lite layer.
strides : int or list of int, default = 1
The strides of kernels in convolution layers for each lite layer,
if not a list, the same is used in all lite layers.
Expand Down Expand Up @@ -319,12 +317,10 @@ class IndividualLITERegressor(BaseDeepRegressor):
if set to `True` then LITEMV is used. LITEMV is the
same architecture as LITE but specifically designed
to better handle multivariate time series.
n_filters : int or list of int32, default = 32
The number of filters used in one lite layer, if not a list, the same
number of filters is used in all lite layers.
kernel_size : int or list of int, default = 40
The head kernel size used for each lite layer, if not a list, the same
is used in all lite layers.
n_filters : int, default = 32
The number of filters used in one lite layer.
kernel_size : int, default = 40
The head kernel size used for each lite layer.
strides : int or list of int, default = 1
The strides of kernels in convolution layers for each lite layer,
if not a list, the same is used in all lite layers.
Expand Down

0 comments on commit 06cde8b

Please sign in to comment.