Taeef Najib
10/01/2022, 11:04 AM# Creating the Multiple Linear Regression model
def build_model(model, X_train: pd.DataFrame, y_train: pd.DataFrame):
# Use the following values: model = LinearRegression(), X_train = X_train, y_train = y_train
reg = model
reg.fit(X_train, y_train)
return reg
I’m using int
, float
, pd.DataFrame
, np.ndarray
etc. as my type hints.
My question is what type hint do I use for the argument model
and since it is returning the model, what type hint should I use for reg
?Ketan (kumare3)
Taeef Najib
10/01/2022, 8:00 PMfrom sklearn.base import RegressorMixin
def build_model(model: RegressorMixin, X_train: pd.DataFrame, y_train: pd.DataFrame) -> RegressorMixin:
# Use the following values: model = LinearRegression(), X_train = X_train, y_train = y_train
reg = model
reg.fit(X_train, y_train)
return reg
Since all regression models inherit from sklearn.base.RegressorMixin
Can we do this?Ketan (kumare3)
Taeef Najib
10/02/2022, 7:34 AMKetan (kumare3)
Taeef Najib
10/02/2022, 2:57 PM