feat(lab-4): make pipeline

2024-12-07 13:00:14 +04:00 · 2024-12-07 13:00:14 +04:00 · 5ab313468c
commit 5ab313468c
parent bd8c7a6d2b
2 changed files with 1091 additions and 0 deletions
--- a/notebooks/lab4.ipynb
+++ b/notebooks/lab4.ipynb
--- a/notebooks/transformers.py
+++ b/notebooks/transformers.py
@ -0,0 +1,17 @@
+import numpy as np
+from sklearn.base import BaseEstimator, TransformerMixin
+
+
+class CarsFeatures(BaseEstimator, TransformerMixin):
+    def __init__(self):
+        pass
+
+    def fit(self, X, y=None):
+        return self
+
+    def transform(self, X, y=None):
+        X["Age"] = 2020 - X["Prod. year"]
+        return X
+
+    def get_feature_names_out(self, features_in):
+        return np.append(features_in, ["Age"], axis=0)