caa69fe696723343cc1780caf356e22eacf3c2af,sklearn/sklearn-template/template/trainer/model.py,,get_estimator,#,28

Before Change

def get_estimator(flags):
  // TODO: Allow pre-processing to be configurable through flags
  classifier = ensemble.RandomForestClassifier(**flags)

  numeric_transformer = pipeline.Pipeline([
      ("imputer", impute.SimpleImputer(strategy="median")),
      ("scaler", preprocessing.StandardScaler()),

After Change

def get_estimator(flags):
  // TODO: Allow pre-processing to be configurable through flags
  classifier = ensemble.RandomForestClassifier(
      n_estimators=flags.n_estimators,
      max_depth=flags.max_depth,
      min_samples_leaf=flags.min_samples_leaf,
      criterion=flags.criterion,
  )

  numeric_transformer = pipeline.Pipeline([
      ("imputer", impute.SimpleImputer(strategy="median")),
      ("scaler", preprocessing.StandardScaler()),

Instances