def train()

in clearbox/training.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

  def train(
      self,
      model_builder: cbx_models.Model,
      features: list[F.Node],
      df: pd.DataFrame | None = None,
      metrics: list[cbx_metrics.BaseMetric] | None = None,
      target_col: str | None = None,
      query_col: str | None = None,
      rank_col: str | None = None,
      stratify_by_col: str | None = None,
      test_df: pd.DataFrame | None = None,
      print_progress: bool = False,
      num_parallel_workers: int | t.Literal['auto'] = 1,