aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/random_forest.py')
-rw-r--r--tensorflow/contrib/learn/python/learn/estimators/random_forest.py9
1 files changed, 1 insertions, 8 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/random_forest.py b/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
index 4aa090ee4d..ec70453163 100644
--- a/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
+++ b/tensorflow/contrib/learn/python/learn/estimators/random_forest.py
@@ -69,8 +69,7 @@ class TensorForestEstimator(estimator.BaseEstimator):
def __init__(self, params, device_assigner=None, model_dir=None,
graph_builder_class=tensor_forest.RandomForestGraphs,
master='', accuracy_metric=None,
- tf_random_seed=None, continue_training=False, verbose=1,
- max_to_keep=5, save_checkpoint_secs=300):
+ tf_random_seed=None, config=None):
self.params = params.fill()
self.accuracy_metric = (accuracy_metric or
('r2' if self.params.regression else 'accuracy'))
@@ -81,12 +80,6 @@ class TensorForestEstimator(estimator.BaseEstimator):
self.training_args = {}
self.construction_args = {}
- config = run_config.RunConfig(
- master=master,
- tf_random_seed=(tf_random_seed or int((time.time() * 1000) % 1000)),
- save_checkpoints_secs=save_checkpoint_secs,
- keep_checkpoint_max=max_to_keep)
-
super(TensorForestEstimator, self).__init__(model_dir=model_dir,
config=config)