Completed comments

explainability/fit_final_models.py

+# Fitting Final Models
+# Author: Joaquín Torres Bravo
+"""
+    Script to fit and save chosen models to be later used in SHAP
+"""
+
 # Libraries
 # --------------------------------------------------------------------------------------------------------
 import pandas as pd
 import numpy as np
-import shap
-import ast
-import matplotlib.pyplot as plt
-import pickle
+import shap # Explainability
+import pickle # Loading/saving models
+# Models
 from xgboost import XGBClassifier
 from sklearn.ensemble import RandomForestClassifier, BaggingClassifier, AdaBoostClassifier
 from sklearn.neural_network import MLPClassifier
 from sklearn.svm import SVC
 from sklearn.linear_model import  LogisticRegression
 from sklearn.tree import DecisionTreeClassifier
+import ast # String to dictionary
 # --------------------------------------------------------------------------------------------------------
 
 # Reading training data