= count func

Author: amirdianov

configs/linear_regression_cfg.py

@@ -5,5 +5,4 @@
 cfg = EasyDict()
 cfg.dataframe_path = os.path.basename("linear_regression_dataset.csv")
 # TODO list of basis functions
-
-cfg.base_functions = [lambda x, degree=i: x**degree for i in range(1000)]
+cfg.base_functions = [lambda x, degree=i: x**degree for i in range(8)]

execute.py

@@ -4,7 +4,7 @@
 from utils.visualisation import Visualisation
 
 
-def experiment(lin_reg_cfg, visualise_prediction=False):
+def experiment(lin_reg_cfg, visualise_prediction=True):
     lin_reg_model = LinearRegression(lin_reg_cfg.base_functions)
     linreg_dataset = LinRegDataset()(lin_reg_cfg.dataframe_path)