Feature extraction and model training

app.py

@@ -72,7 +72,7 @@ generated_code = f"""
 import pandas as pd
 from sklearn.model_selection import train_test_split
 from sklearn.preprocessing import StandardScaler
-from {model.__module__} import {model.__class__.__name__}
+from {model_mapping[model].__class__.__module__} import {model_mapping[model].__class__.__name__}
 
 # Load Data
 df = pd.read_csv('{dataset_path}')
@@ -88,7 +88,7 @@ X_train = scaler.fit_transform(X_train)
 X_test = scaler.transform(X_test)
 
 # Train Model 
-model = {model.__class__.__name__}()
+model = {model_mapping[model].__class__.__name__}()
 model.fit(X_train, y_train)
 
 # Predict