Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support new Quantile Regression Model #461

Open
brifordwylie opened this issue Jun 30, 2024 · 0 comments
Open

Support new Quantile Regression Model #461

brifordwylie opened this issue Jun 30, 2024 · 0 comments
Assignees
Labels
model SageWorks Model

Comments

@brifordwylie
Copy link
Member

  • sageworks/core/artifacts/model_core.py", line 616

Endpoint Core

        # Compute the standard performance metrics for this model
        model_type = self.model_type()
        if model_type == ModelType.REGRESSOR.value:
            prediction_df = self.residuals(target_column, prediction_df)
            metrics = self.regression_metrics(target_column, prediction_df)
        elif model_type == ModelType.CLASSIFIER.value:
            metrics = self.classification_metrics(target_column, prediction_df)
        else:
            # Unknown Model Type: Give log message and set metrics to empty dataframe
            self.log.warning(f"Unknown Model Type: {model_type}")
            metrics = pd.DataFrame()
@brifordwylie brifordwylie added the model SageWorks Model label Jun 30, 2024
@brifordwylie brifordwylie added this to the SageWorks 0.7.0 milestone Jun 30, 2024
@brifordwylie brifordwylie self-assigned this Jun 30, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
model SageWorks Model
Projects
Status: In Progress
Development

No branches or pull requests

1 participant