# Import statement
from hottbox.algorithms.classification import LSSTM
LSSTM
(C=1, tol=0.001, max_iter=100, probability=False, verbose=False)[source]¶Least Squares Support Tensor Machine (LS-STM) for binary classification.
Penalty parameter of the error term.
Tolerance for stopping criterion.
Hard limit on iterations within solver.
Whether to enable probability estimates. This must be enabled prior
to calling fit
, and will slow down that method.
Enable verbose output.
Attributes
weights_ |
(list[np.ndarray]) List of weights for each mode of the training data. |
bias_ |
(np.float64) |
eta_history_ |
(np.ndarray) |
bias_history_ |
(np.ndarray) |
References
Cichocki, Andrzej, et al. “Tensor networks for dimensionality reduction and large-scale optimization: Part 2 applications and future perspectives.” Foundations and Trends in Machine Learning 9.6 (2017): 431-673.
Methods
|
Fit the LS-STM model according to the given data. |
|
Get parameters for this estimator. |
|
Predict the class labels for the provided data. |
|
Compute probabilities of possible outcomes for samples in the provided data. |
|
Returns the mean accuracy on the given test data and labels. |
|
Set the parameters of this estimator. |
fit
(self, X, y)[source]¶Fit the LS-STM model according to the given data.
List of training samples of the same order and size.
Target values relative to X for classification.
get_params
(self)[source]¶Get parameters for this estimator.
Dictionary of parameter names mapped to their values.
predict
(self, X)[source]¶Predict the class labels for the provided data.
List of test samples.
Class labels for samples in X.
predict_proba
(self, X)[source]¶Compute probabilities of possible outcomes for samples in the provided data.