Skip to content

Model_building - univariate_selection #68

@JanBenisek

Description

@JanBenisek
  • Since AUC cannot be applied to continuous target, we need another metric. Since we want univariate selection RMSE could be used instead. Review this before continuing.
  • After, extend the function with model_type which can be [classification, regression]. Then, the function returns either RMSE or AUC. The order has to be correct (we want the highest AUC, but the lowest RMSE)
    def compute_univariate_preselection(target_enc_train_data: pd.DataFrame,
  • Document and write/modify unit tests

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions