name refactors

Author: SamoraHunter

ml_grid/pipeline/model_class_list.py

@@ -20,29 +20,27 @@
 from ml_grid.model_classes.h2o_deeplearning_classifier_class import (
     H2O_DeepLearning_class,
 )
-from ml_grid.model_classes.h2o_drf_classifier_class import H2ODRFClass as H2O_DRF_class
+from ml_grid.model_classes.h2o_drf_classifier_class import H2ODRFClass as H2O_DRF_class # noqa
 from ml_grid.model_classes.h2o_gam_classifier_class import H2OGAMClass as H2O_GAM_class
 from ml_grid.model_classes.h2o_gbm_classifier_class import (
-    H2O_GBM_class,  # No change needed here
+    H2O_GBM_class, # No change needed here
 )  # Correctly named
 from ml_grid.model_classes.h2o_glm_classifier_class import H2O_GLM_class
 from ml_grid.model_classes.h2o_naive_bayes_classifier_class import (
     H2O_NaiveBayes_class,
 )
-from ml_grid.model_classes.h2o_rulefit_classifier_class import (
-    H2ORuleFitClass as H2O_RuleFit_class,
-)
+from ml_grid.model_classes.h2o_rulefit_classifier_class import H2ORuleFitClass as H2O_RuleFit_class # noqa
 from ml_grid.model_classes.h2o_stackedensemble_classifier_class import (
     H2O_StackedEnsemble_class,
 )
 from ml_grid.model_classes.h2o_xgboost_classifier_class import H2O_XGBoost_class
 from ml_grid.model_classes.keras_classifier_class import KerasClassifierClass
 from ml_grid.model_classes.knn_classifier_class import KNeighborsClassifierClass
 from ml_grid.model_classes.knn_gpu_classifier_class import KNNGpuWrapperClass
-from ml_grid.model_classes.light_gbm_class import LightGBMClassifierWrapper
+from ml_grid.model_classes.light_gbm_class import LightGBMClassifierWrapper # noqa
 from ml_grid.model_classes.logistic_regression_class import LogisticRegressionClass
 from ml_grid.model_classes.mlp_classifier_class import MLPClassifierClass
-from ml_grid.model_classes.NeuralNetworkClassifier_class import (
+from ml_grid.model_classes.NeuralNetworkClassifier_class import ( # noqa
     NeuralNetworkClassifier_class,
 )
 from ml_grid.model_classes.quadratic_discriminant_class import (
@@ -52,8 +50,8 @@
     RandomForestClassifierClass,
 )
 from ml_grid.model_classes.svc_class import SVCClass
-from ml_grid.model_classes.tabtransformer_classifier_class import TabTransformerClass
-from ml_grid.model_classes.xgb_classifier_class import XGBClassifierClass
+from ml_grid.model_classes.tabtransformer_classifier_class import TabTransformerClass # noqa
+from ml_grid.model_classes.xgb_classifier_class import XGBClassifierClass # noqa
 from ml_grid.pipeline.data import pipe
 
 

ml_grid/pipeline/model_class_list_ts.py

@@ -4,8 +4,8 @@
 
 from ml_grid.model_classes_time_series.ArsenalClassifier_module import Arsenal_class
 from ml_grid.model_classes_time_series.CNNClassifier_module import CNNClassifier_class
-from ml_grid.model_classes_time_series.ContractableBOSSClassifier_module import (
-    ContractableBOSS_class,
+from ml_grid.model_classes_time_series.elasticEnsembleClassifier_module import (
+    ElasticEnsemble_class,
 )
 from ml_grid.model_classes_time_series.EncoderClassifier_module import (
     EncoderClassifier_class,
@@ -14,9 +14,6 @@
 from ml_grid.model_classes_time_series.FreshPRINCEClassifier_module import (
     FreshPRINCEClassifier_class,
 )
-from ml_grid.model_classes_time_series.HIVECOTEV1Classifier_module import (
-    HIVECOTEV1_class,
-)
 from ml_grid.model_classes_time_series.HIVECOTEV2Classifier_module import (
     HIVECOTEV2_class,
 )
@@ -33,7 +30,6 @@
     KNeighborsTimeSeriesClassifier_class,
 )
 
-
 # from ml_grid.model_classes import LogisticRegression_class
 from ml_grid.model_classes_time_series.MLPClassifier_module import MLPClassifier_class
 from ml_grid.model_classes_time_series.MUSEClassifier_module import MUSE_class
@@ -43,28 +39,25 @@
 from ml_grid.model_classes_time_series.ResNetClassifier_module import (
     ResNetClassifier_class,
 )
+from ml_grid.model_classes_time_series.rocketClassifier_module import (
+    RocketClassifier_class,
+)
+from ml_grid.model_classes_time_series.shapeDTWClassifier_module import ShapeDTW_class
 from ml_grid.model_classes_time_series.SignatureClassifier_module import (
     SignatureClassifier_class,
 )
 from ml_grid.model_classes_time_series.SummaryClassifier_module import (
     SummaryClassifier_class,
 )
-from ml_grid.model_classes_time_series.TSFreshClassifier_module import (
-    TSFreshClassifier_class,
-)
 from ml_grid.model_classes_time_series.TapNetClassifier_module import (
     TapNetClassifier_class,
 )
 from ml_grid.model_classes_time_series.TemporalDictionaryEnsembleClassifier_module import (
     TemporalDictionaryEnsemble_class,
 )
-from ml_grid.model_classes_time_series.elasticEnsembleClassifier_module import (
-    ElasticEnsemble_class,
-)
-from ml_grid.model_classes_time_series.rocketClassifier_module import (
-    RocketClassifier_class,
+from ml_grid.model_classes_time_series.TSFreshClassifier_module import (
+    TSFreshClassifier_class,
 )
-from ml_grid.model_classes_time_series.shapeDTWClassifier_module import ShapeDTW_class
 from ml_grid.pipeline.data import pipe