diff --git a/verticapy/machine_learning/vertica/linear_model.py b/verticapy/machine_learning/vertica/linear_model.py index 3d9d6c91f..3700df290 100755 --- a/verticapy/machine_learning/vertica/linear_model.py +++ b/verticapy/machine_learning/vertica/linear_model.py @@ -257,7 +257,7 @@ def plot( """ -class ElasticNet(Regressor, LinearModel): +class ElasticNet(LinearModel, Regressor): """ Creates an ElasticNet object using the Vertica Linear Regression algorithm. The Elastic Net @@ -698,7 +698,7 @@ def __init__( } -class Lasso(Regressor, LinearModel): +class Lasso(LinearModel, Regressor): """ Creates a Lasso object using the Vertica Linear Regression algorithm. @@ -1140,7 +1140,7 @@ def __init__( } -class LinearRegression(Regressor, LinearModel): +class LinearRegression(LinearModel, Regressor): """ Creates a LinearRegression object using the Vertica Linear Regression algorithm. @@ -2031,7 +2031,7 @@ def __init__( } -class Ridge(Regressor, LinearModel): +class Ridge(LinearModel, Regressor): """ Creates a Ridge object using the Vertica Linear Regression algorithm. @@ -2501,7 +2501,7 @@ def __init__( """ -class LogisticRegression(BinaryClassifier, LinearModelClassifier): +class LogisticRegression(LinearModelClassifier, BinaryClassifier): """ Creates a LogisticRegression object using the Vertica Logistic Regression algorithm. diff --git a/verticapy/tests_new/machine_learning/vertica/test_base_model_methods.py b/verticapy/tests_new/machine_learning/vertica/test_base_model_methods.py index c5e475e31..a8a16e6db 100644 --- a/verticapy/tests_new/machine_learning/vertica/test_base_model_methods.py +++ b/verticapy/tests_new/machine_learning/vertica/test_base_model_methods.py @@ -805,7 +805,7 @@ def test_get_attributes(self, get_models, model_class): "features_importance_trees_", ] else: - vpy_model_attributes = [] + vpy_model_attributes = ["coef_", "intercept_", "features_importance_"] assert get_models.vpy.model.get_attributes() == vpy_model_attributes