Commit 762a245e authored by Joaquin Torres's avatar Joaquin Torres

Completed comments

parent 1e69d01c
# Hyperparameter Tuning
# Author: Joaquín Torres Bravo
""" """
Finding optimal hyperparameters through RandomSearchCV for each group (1. pre - 2. post) Finding optimal hyperparameters through RandomSearchCV for each group (1. pre - 2. post)
and method: and pipeline:
1. Original training dataset 1. Original training dataset
2. Original training dataset - Cost sensitive 2. Original training dataset with cost-sensitive learning (penalizing misclassification of minority class)
3. Oversampling 3. Oversampling
4. Undersampling 4. Undersampling
""" """
...@@ -11,16 +13,18 @@ ...@@ -11,16 +13,18 @@
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
import pandas as pd import pandas as pd
import numpy as np import numpy as np
# Models
from xgboost import XGBClassifier from xgboost import XGBClassifier
from sklearn.model_selection import StratifiedKFold
from sklearn.ensemble import RandomForestClassifier, BaggingClassifier, AdaBoostClassifier from sklearn.ensemble import RandomForestClassifier, BaggingClassifier, AdaBoostClassifier
from sklearn.neural_network import MLPClassifier from sklearn.neural_network import MLPClassifier
from sklearn.svm import SVC from sklearn.svm import SVC
from sklearn.linear_model import LogisticRegression from sklearn.linear_model import LogisticRegression
from sklearn.tree import DecisionTreeClassifier from sklearn.tree import DecisionTreeClassifier
from scipy.stats import randint, uniform # CV
from sklearn.model_selection import StratifiedKFold
from sklearn.model_selection import RandomizedSearchCV from sklearn.model_selection import RandomizedSearchCV
# Misc
from scipy.stats import randint, uniform
# -------------------------------------------------------------------------------------------------------- # --------------------------------------------------------------------------------------------------------
# Function to read training datasets # Function to read training datasets
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment