diff --git a/Black Friday Sales Prediction Analysis - Regression/Black Friday Sales Prediction Analysis - Regression.ipynb b/Black Friday Sales Prediction Analysis - Regression/Black Friday Sales Prediction Analysis - Regression.ipynb index f6a8f42..ee4f606 100644 --- a/Black Friday Sales Prediction Analysis - Regression/Black Friday Sales Prediction Analysis - Regression.ipynb +++ b/Black Friday Sales Prediction Analysis - Regression/Black Friday Sales Prediction Analysis - Regression.ipynb @@ -44,6 +44,13 @@ "import numpy as np\n", "import seaborn as sns\n", "import matplotlib.pyplot as plt\n", + "from sklearn.model_selection import train_test_split, cross_val_score, GridSearchCV\n" + "from sklearn.metrics import mean_squared_error\n" + "from sklearn.preprocessing import LabelEncoder\n" + "from sklearn.ensemble import RandomForestRegressor\n" + "from xgboost import XGBRegressor\n" + "from sklearn.feature_selection import SelectKBest, f_regression\n" + "from scipy.stats import zscore\n" "import warnings\n", "%matplotlib inline\n", "warnings.filterwarnings('ignore')"