Skip to content

Commit

Permalink
Pulling the pytorch stuff out of simple linear models as didn't add a…
Browse files Browse the repository at this point in the history
…nything and made things complex
  • Loading branch information
jameschapman19 committed Aug 7, 2023
1 parent 6e5d85a commit ed8cdc0
Showing 1 changed file with 2 additions and 14 deletions.
16 changes: 2 additions & 14 deletions cca_zoo/sequential.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
Check if each effect is significant, and if so, remove it from the data and repeat.
"""
from abc import ABCMeta
from typing import Iterable

from sklearn.base import MetaEstimatorMixin
from cca_zoo._base import BaseModel
Expand Down Expand Up @@ -57,20 +58,7 @@ def __init__(
self.p_threshold = p_threshold
self.corr_threshold = corr_threshold

def fit(self, views):
"""
Fits the model to the views.
Parameters
----------
views : list of array-likes of shape (n_samples, n_features)
A list of views, where each view is an array-like matrix of shape (n_samples, n_features).
Returns
-------
self : SequentialModel
The SequentialModel instance.
"""
def fit(self, views: Iterable[np.ndarray], y=None, **kwargs):
# Validate the input data and parameters
self._validate_data(views)
self._check_params()
Expand Down

0 comments on commit ed8cdc0

Please sign in to comment.