From 5bf9b854b466525fe0f1aae60422c903715ef61d Mon Sep 17 00:00:00 2001 From: Harmanaya Sharma Date: Wed, 23 Oct 2024 00:20:29 +0530 Subject: [PATCH] Resolved mypy checks --- machine_learning/ridge_regression.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/machine_learning/ridge_regression.py b/machine_learning/ridge_regression.py index 1206d41b5..ad34600b3 100644 --- a/machine_learning/ridge_regression.py +++ b/machine_learning/ridge_regression.py @@ -1,5 +1,6 @@ import numpy as np import pandas as pd +from typing import Optional, Tuple class RidgeRegression: @@ -15,7 +16,7 @@ class RidgeRegression: self.alpha = alpha self.lambda_ = lambda_ self.iterations = iterations - self.theta = None + self.theta: Optional[np.ndarray] = None # Initialize as None, later will be ndarray def feature_scaling( self, features: np.ndarray @@ -92,6 +93,9 @@ class RidgeRegression: >>> predictions.shape == target.shape True """ + if self.theta is None: + raise ValueError("Model is not trained yet. Call the `fit` method first.") + features_scaled, _, _ = self.feature_scaling( features ) # Scale features using training data @@ -114,6 +118,9 @@ class RidgeRegression: >>> isinstance(cost, float) True """ + if self.theta is None: + raise ValueError("Model is not trained yet. Call the `fit` method first.") + features_scaled, _, _ = self.feature_scaling( features ) # Scale features using training data