(df: pd.DataFrame)
| 630 | |
| 631 | |
| 632 | def get_classification_hyperparams(df: pd.DataFrame) -> tuple[int, object]: |
| 633 | n_classes = df.completion.nunique() |
| 634 | pos_class = None |
| 635 | if n_classes == 2: |
| 636 | pos_class = df.completion.value_counts().index[0] |
| 637 | return n_classes, pos_class |
| 638 | |
| 639 | |
| 640 | def write_out_file(df: pd.DataFrame, fname: str, any_remediations: bool, auto_accept: bool) -> None: |