| /external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
| linear.py | 87 model_dir=None,
99 model_dir: Directory to save model parameters, graph and etc. This can
118 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| dnn.py | 90 model_dir=None,
107 model_dir: Directory to save model parameters, graph and etc. This can
134 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| dnn_linear_combined.py | 97 model_dir=None,
112 model_dir: Directory to save model parameters, graph and etc. This can
164 model_fn=_model_fn, model_dir=model_dir, config=config)
|
| dnn_linear_combined_test.py | 45 model_dir=None,
56 model_dir=model_dir,
95 model_dir=None,
104 model_dir=model_dir,
161 model_dir=self._model_dir)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| trainer_hooks_test.py | 40 trainer_hooks.FeatureImportanceSummarySaver(model_dir=None)
44 model_dir = tempfile.mkdtemp()
45 hook = trainer_hooks.FeatureImportanceSummarySaver(model_dir)
54 model_dir = tempfile.mkdtemp()
55 hook = trainer_hooks.FeatureImportanceSummarySaver(model_dir)
71 self.assertTrue(os.path.exists(os.path.join(model_dir, "featA")))
72 self.assertTrue(os.path.exists(os.path.join(model_dir, "featB")))
|
| estimator.py | 38 model_dir=None,
56 model_dir: Directory for model exports, etc.
102 model_dir=model_dir,
118 model_dir=None,
139 model_dir: Directory for model exports, etc.
169 model_dir=model_dir,
187 model_dir=None,
204 model_dir: Directory for model exports, etc
[all...] |
| dnn_tree_combined_estimator_test.py | 84 model_dir = tempfile.mkdtemp()
94 model_dir=model_dir,
|
| /external/tensorflow/tensorflow/examples/learn/ |
| random_forest_mnist.py | 37 def build_estimator(model_dir):
48 params, graph_builder_class=graph_builder_class, model_dir=model_dir)
53 model_dir = tempfile.mkdtemp() if not FLAGS.model_dir else FLAGS.model_dir
54 print('model directory = %s' % model_dir)
56 est = build_estimator(model_dir)
95 '--model_dir',
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| run_config_test.py | 41 tf_random_seed=RANDOM_SEED, model_dir=TEST_DIR)
229 self.assertIsNone(empty_config.model_dir)
231 config = run_config_lib.RunConfig(model_dir=TEST_DIR)
232 self.assertEqual(TEST_DIR, config.model_dir)
235 tf_config = {"model_dir": TEST_DIR}
238 self.assertEqual(TEST_DIR, run_config.model_dir)
241 tf_config = {"model_dir": TEST_DIR}
243 run_config = run_config_lib.RunConfig(model_dir=TEST_DIR)
244 self.assertEqual(TEST_DIR, run_config.model_dir)
247 tf_config = {"model_dir": TEST_DIR
[all...] |
| debug.py | 156 model_dir=None,
165 model_dir: Directory to save model parameters, graph and etc. This can
196 model_dir=model_dir,
272 model_dir=None,
280 model_dir: Directory to save model parameters, graph and etc. This can
307 model_dir=model_dir,
|
| composable_model.py | 185 def get_weights(self, model_dir):
189 model_dir: Directory where model parameters, graph and etc. are saved.
194 all_variables = [name for name, _ in list_variables(model_dir)]
201 values[name] = load_variable(model_dir, name)
206 def get_bias(self, model_dir):
210 model_dir: Directory where model parameters, graph and etc. are saved.
215 return load_variable(model_dir, name=(self._scope + "/bias_weight"))
303 def get_weights(self, model_dir):
307 model_dir: Directory where model parameters, graph and etc. are saved.
314 model_dir, name=(self._scope + "/hiddenlayer_%d/weights" % i)
[all...] |
| run_config.py | 231 model_dir=None,
272 model_dir: directory where model parameters, graph etc are saved. If
273 `None`, will use `model_dir` property in `TF_CONFIG` environment
307 self._model_dir = _get_model_dir(model_dir)
345 def model_dir(self): member in class:RunConfig
442 def _get_model_dir(model_dir):
443 """Returns `model_dir` based user provided `model_dir` or `TF_CONFIG`."""
446 os.environ.get('TF_CONFIG') or '{}').get('model_dir', None)
448 if model_dir is not None and model_dir_in_tf_config != model_dir
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| learn_runner_test.py | 41 "Must specify a model directory `model_dir` in `run_config`.")
63 def __init__(self, default=None, config=None, model_dir=None):
66 internal_model_dir = model_dir or config.model_dir
73 def model_dir(self): member in class:TestExperiment.__init__.Estimator
106 return TestExperiment(model_dir=output_dir)
112 return TestExperiment(config=run_config, model_dir=output_dir)
210 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
218 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
227 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR
[all...] |
| monitors_test.py | 324 model_dir = 'model/dir'
325 estimator.model_dir = model_dir
336 mock_latest_checkpoint.assert_called_with(model_dir)
344 model_dir = 'model/dir'
345 estimator.model_dir = model_dir
347 mock_latest_checkpoint.return_value = '%s/ckpt' % model_dir
363 model_dir = 'model/dir'
364 estimator.model_dir = model_di
[all...] |
| evaluable.py | 30 def model_dir(self): member in class:Evaluable
101 latest checkpoint in `model_dir` is used.
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| contrib_estimator_predictor_test.py | 39 model_dir = tempfile.mkdtemp()
41 core=False, model_dir=model_dir)
|
| core_estimator_predictor_test.py | 39 model_dir = tempfile.mkdtemp()
41 core=True, model_dir=model_dir)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| estimators.py | 44 def __init__(self, model, state_manager=None, optimizer=None, model_dir=None,
54 model_dir: See `Estimator`.
70 model_dir=model_dir,
171 optimizer=None, model_dir=None, config=None):
204 model_dir: See `Estimator`.
242 model_dir=model_dir,
249 def __init__(self, model, state_manager=None, optimizer=None, model_dir=None,
262 model_dir=model_dir
[all...] |
| estimators_test.py | 49 model_dir = tempfile.mkdtemp(dir=self.get_temp_dir())
50 first_estimator = estimator_fn(model_dir)
70 second_estimator = estimator_fn(model_dir)
130 def _estimator_fn(model_dir):
133 num_features=1, model_dir=model_dir, config=_SeedRunConfig(),
141 def _estimator_fn(model_dir):
143 num_features=1, periodicities=10, model_dir=model_dir, dtype=dtype,
|
| /external/tensorflow/tensorflow/python/tools/ |
| import_pb_to_tensorboard.py | 33 def import_to_tensorboard(model_dir, log_dir):
37 model_dir: The location of the protobuf (`pb`) model to visualize
46 with gfile.FastGFile(model_dir, "rb") as f:
58 import_to_tensorboard(FLAGS.model_dir, FLAGS.log_dir)
64 "--model_dir",
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| sdca_estimator.py | 210 model_dir=None,
228 model_dir: Directory to save model parameters, graph etc. This can also be
274 model_dir=model_dir,
334 model_dir=None,
351 model_dir: Directory to save model parameters, graph etc. This can also be
372 model_dir=model_dir,
461 model_dir=None,
479 model_dir: Directory to save model parameters, graph etc. This can also b
[all...] |
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| dnn_testing_utils.py | 92 def create_checkpoint(weights_and_biases, global_step, model_dir):
98 model_dir: Directory into which checkpoint is saved.
124 saver.Saver().save(sess, os.path.join(model_dir, 'model.ckpt'))
754 model_dir=self._ckpt_and_vocab_dir,
767 warm_start_from=dnn_classifier.model_dir)
786 model_dir=self._ckpt_and_vocab_dir,
797 warm_start_from=dnn_regressor.model_dir)
816 model_dir=self._ckpt_and_vocab_dir,
[all...] |
| /external/tensorflow/tensorflow/tools/dist_test/scripts/ |
| dist_census_widendeep_test.sh | 78 MODEL_DIR="/shared/census_widendeep_model"
80 rm -rf ${MODEL_DIR} || \
81 die "Failed to remove existing model directory: ${MODEL_DIR}"
106 --model_dir="${MODEL_DIR}" \
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| gmm.py | 77 model_dir=None,
87 model_dir: the directory to save the model results and log files.
104 model_fn=self._model_builder(), model_dir=model_dir, config=config)
132 self.model_dir, gmm_ops.GmmAlgorithm.CLUSTERS_WEIGHT)
137 self.model_dir, gmm_ops.GmmAlgorithm.CLUSTERS_VARIABLE)
143 self.model_dir, gmm_ops.GmmAlgorithm.CLUSTERS_COVS_VARIABLE)
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| kernel_estimators.py | 157 model_dir=None,
183 model_dir=model_dir,
219 model_dir=...,
227 model_dir=...,
259 model_dir=None,
271 model_dir: Directory to save model parameters, graph etc. This can also be
302 model_dir=model_dir,
|
