Skip to content

Commit

Permalink
Update TensorForestEstimator to use non-deprecated input method durin…
Browse files Browse the repository at this point in the history
…g export.

Change: 141109887
  • Loading branch information
tensorflower-gardener committed Dec 6, 2016
1 parent a487ee1 commit 9423429
Showing 1 changed file with 4 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -317,6 +317,7 @@ def export(self,
export_dir,
input_fn,
signature_fn=None,
input_feature_key=None,
default_batch_size=1):
"""See BaseEstimator.export."""
# Reset model function with basic device assigner.
Expand All @@ -330,7 +331,9 @@ def export(self,
weights_name=self.weights_name)
result = self._estimator.export(
export_dir=export_dir,
use_deprecated_input_fn=True,
input_fn=input_fn,
input_feature_key=input_feature_key,
use_deprecated_input_fn=False,
signature_fn=(signature_fn or
(export.regression_signature_fn
if self.params.regression else
Expand Down

0 comments on commit 9423429

Please sign in to comment.