def _model_fn(features, labels, mode, config): return canned_boosted_trees._bt_model_fn(features, labels, mode, head, feature_columns, tree_hparams, n_batches_per_layer, config=config)
def _model_fn(features, labels, mode, config): return canned_boosted_trees._bt_model_fn( features, labels, mode, head, feature_columns, tree_hparams, n_batches_per_layer=1, config=config, closed_form_grad_and_hess_fn=closed_form, train_in_memory=True)