Esempi in Python per _bt_model_fn

Esempio n. 1

 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)

Esempio n. 2

 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)