/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,
|