tensorbored 2.21.0rc1769983804__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (271) hide show
  1. tensorbored/__init__.py +112 -0
  2. tensorbored/_vendor/__init__.py +0 -0
  3. tensorbored/_vendor/bleach/__init__.py +125 -0
  4. tensorbored/_vendor/bleach/_vendor/__init__.py +0 -0
  5. tensorbored/_vendor/bleach/_vendor/html5lib/__init__.py +35 -0
  6. tensorbored/_vendor/bleach/_vendor/html5lib/_ihatexml.py +289 -0
  7. tensorbored/_vendor/bleach/_vendor/html5lib/_inputstream.py +918 -0
  8. tensorbored/_vendor/bleach/_vendor/html5lib/_tokenizer.py +1735 -0
  9. tensorbored/_vendor/bleach/_vendor/html5lib/_trie/__init__.py +5 -0
  10. tensorbored/_vendor/bleach/_vendor/html5lib/_trie/_base.py +40 -0
  11. tensorbored/_vendor/bleach/_vendor/html5lib/_trie/py.py +67 -0
  12. tensorbored/_vendor/bleach/_vendor/html5lib/_utils.py +159 -0
  13. tensorbored/_vendor/bleach/_vendor/html5lib/constants.py +2946 -0
  14. tensorbored/_vendor/bleach/_vendor/html5lib/filters/__init__.py +0 -0
  15. tensorbored/_vendor/bleach/_vendor/html5lib/filters/alphabeticalattributes.py +29 -0
  16. tensorbored/_vendor/bleach/_vendor/html5lib/filters/base.py +12 -0
  17. tensorbored/_vendor/bleach/_vendor/html5lib/filters/inject_meta_charset.py +73 -0
  18. tensorbored/_vendor/bleach/_vendor/html5lib/filters/lint.py +93 -0
  19. tensorbored/_vendor/bleach/_vendor/html5lib/filters/optionaltags.py +207 -0
  20. tensorbored/_vendor/bleach/_vendor/html5lib/filters/sanitizer.py +916 -0
  21. tensorbored/_vendor/bleach/_vendor/html5lib/filters/whitespace.py +38 -0
  22. tensorbored/_vendor/bleach/_vendor/html5lib/html5parser.py +2795 -0
  23. tensorbored/_vendor/bleach/_vendor/html5lib/serializer.py +409 -0
  24. tensorbored/_vendor/bleach/_vendor/html5lib/treeadapters/__init__.py +30 -0
  25. tensorbored/_vendor/bleach/_vendor/html5lib/treeadapters/genshi.py +54 -0
  26. tensorbored/_vendor/bleach/_vendor/html5lib/treeadapters/sax.py +50 -0
  27. tensorbored/_vendor/bleach/_vendor/html5lib/treebuilders/__init__.py +88 -0
  28. tensorbored/_vendor/bleach/_vendor/html5lib/treebuilders/base.py +417 -0
  29. tensorbored/_vendor/bleach/_vendor/html5lib/treebuilders/dom.py +239 -0
  30. tensorbored/_vendor/bleach/_vendor/html5lib/treebuilders/etree.py +343 -0
  31. tensorbored/_vendor/bleach/_vendor/html5lib/treebuilders/etree_lxml.py +392 -0
  32. tensorbored/_vendor/bleach/_vendor/html5lib/treewalkers/__init__.py +154 -0
  33. tensorbored/_vendor/bleach/_vendor/html5lib/treewalkers/base.py +252 -0
  34. tensorbored/_vendor/bleach/_vendor/html5lib/treewalkers/dom.py +43 -0
  35. tensorbored/_vendor/bleach/_vendor/html5lib/treewalkers/etree.py +131 -0
  36. tensorbored/_vendor/bleach/_vendor/html5lib/treewalkers/etree_lxml.py +215 -0
  37. tensorbored/_vendor/bleach/_vendor/html5lib/treewalkers/genshi.py +69 -0
  38. tensorbored/_vendor/bleach/_vendor/parse.py +1078 -0
  39. tensorbored/_vendor/bleach/callbacks.py +32 -0
  40. tensorbored/_vendor/bleach/html5lib_shim.py +757 -0
  41. tensorbored/_vendor/bleach/linkifier.py +633 -0
  42. tensorbored/_vendor/bleach/parse_shim.py +1 -0
  43. tensorbored/_vendor/bleach/sanitizer.py +638 -0
  44. tensorbored/_vendor/bleach/six_shim.py +19 -0
  45. tensorbored/_vendor/webencodings/__init__.py +342 -0
  46. tensorbored/_vendor/webencodings/labels.py +231 -0
  47. tensorbored/_vendor/webencodings/mklabels.py +59 -0
  48. tensorbored/_vendor/webencodings/x_user_defined.py +325 -0
  49. tensorbored/assets.py +36 -0
  50. tensorbored/auth.py +102 -0
  51. tensorbored/backend/__init__.py +0 -0
  52. tensorbored/backend/application.py +604 -0
  53. tensorbored/backend/auth_context_middleware.py +38 -0
  54. tensorbored/backend/client_feature_flags.py +113 -0
  55. tensorbored/backend/empty_path_redirect.py +46 -0
  56. tensorbored/backend/event_processing/__init__.py +0 -0
  57. tensorbored/backend/event_processing/data_ingester.py +276 -0
  58. tensorbored/backend/event_processing/data_provider.py +535 -0
  59. tensorbored/backend/event_processing/directory_loader.py +142 -0
  60. tensorbored/backend/event_processing/directory_watcher.py +272 -0
  61. tensorbored/backend/event_processing/event_accumulator.py +950 -0
  62. tensorbored/backend/event_processing/event_file_inspector.py +463 -0
  63. tensorbored/backend/event_processing/event_file_loader.py +292 -0
  64. tensorbored/backend/event_processing/event_multiplexer.py +521 -0
  65. tensorbored/backend/event_processing/event_util.py +68 -0
  66. tensorbored/backend/event_processing/io_wrapper.py +223 -0
  67. tensorbored/backend/event_processing/plugin_asset_util.py +104 -0
  68. tensorbored/backend/event_processing/plugin_event_accumulator.py +721 -0
  69. tensorbored/backend/event_processing/plugin_event_multiplexer.py +522 -0
  70. tensorbored/backend/event_processing/reservoir.py +266 -0
  71. tensorbored/backend/event_processing/tag_types.py +29 -0
  72. tensorbored/backend/experiment_id.py +71 -0
  73. tensorbored/backend/experimental_plugin.py +51 -0
  74. tensorbored/backend/http_util.py +263 -0
  75. tensorbored/backend/json_util.py +70 -0
  76. tensorbored/backend/path_prefix.py +67 -0
  77. tensorbored/backend/process_graph.py +74 -0
  78. tensorbored/backend/security_validator.py +202 -0
  79. tensorbored/compat/__init__.py +69 -0
  80. tensorbored/compat/proto/__init__.py +0 -0
  81. tensorbored/compat/proto/allocation_description_pb2.py +35 -0
  82. tensorbored/compat/proto/api_def_pb2.py +82 -0
  83. tensorbored/compat/proto/attr_value_pb2.py +80 -0
  84. tensorbored/compat/proto/cluster_pb2.py +58 -0
  85. tensorbored/compat/proto/config_pb2.py +271 -0
  86. tensorbored/compat/proto/coordination_config_pb2.py +45 -0
  87. tensorbored/compat/proto/cost_graph_pb2.py +87 -0
  88. tensorbored/compat/proto/cpp_shape_inference_pb2.py +70 -0
  89. tensorbored/compat/proto/debug_pb2.py +65 -0
  90. tensorbored/compat/proto/event_pb2.py +149 -0
  91. tensorbored/compat/proto/full_type_pb2.py +74 -0
  92. tensorbored/compat/proto/function_pb2.py +157 -0
  93. tensorbored/compat/proto/graph_debug_info_pb2.py +111 -0
  94. tensorbored/compat/proto/graph_pb2.py +41 -0
  95. tensorbored/compat/proto/histogram_pb2.py +39 -0
  96. tensorbored/compat/proto/meta_graph_pb2.py +254 -0
  97. tensorbored/compat/proto/node_def_pb2.py +61 -0
  98. tensorbored/compat/proto/op_def_pb2.py +81 -0
  99. tensorbored/compat/proto/resource_handle_pb2.py +48 -0
  100. tensorbored/compat/proto/rewriter_config_pb2.py +93 -0
  101. tensorbored/compat/proto/rpc_options_pb2.py +35 -0
  102. tensorbored/compat/proto/saved_object_graph_pb2.py +193 -0
  103. tensorbored/compat/proto/saver_pb2.py +38 -0
  104. tensorbored/compat/proto/step_stats_pb2.py +116 -0
  105. tensorbored/compat/proto/struct_pb2.py +144 -0
  106. tensorbored/compat/proto/summary_pb2.py +111 -0
  107. tensorbored/compat/proto/tensor_description_pb2.py +38 -0
  108. tensorbored/compat/proto/tensor_pb2.py +68 -0
  109. tensorbored/compat/proto/tensor_shape_pb2.py +46 -0
  110. tensorbored/compat/proto/tfprof_log_pb2.py +307 -0
  111. tensorbored/compat/proto/trackable_object_graph_pb2.py +90 -0
  112. tensorbored/compat/proto/types_pb2.py +105 -0
  113. tensorbored/compat/proto/variable_pb2.py +62 -0
  114. tensorbored/compat/proto/verifier_config_pb2.py +38 -0
  115. tensorbored/compat/proto/versions_pb2.py +35 -0
  116. tensorbored/compat/tensorflow_stub/__init__.py +38 -0
  117. tensorbored/compat/tensorflow_stub/app.py +124 -0
  118. tensorbored/compat/tensorflow_stub/compat/__init__.py +131 -0
  119. tensorbored/compat/tensorflow_stub/compat/v1/__init__.py +20 -0
  120. tensorbored/compat/tensorflow_stub/dtypes.py +692 -0
  121. tensorbored/compat/tensorflow_stub/error_codes.py +169 -0
  122. tensorbored/compat/tensorflow_stub/errors.py +507 -0
  123. tensorbored/compat/tensorflow_stub/flags.py +124 -0
  124. tensorbored/compat/tensorflow_stub/io/__init__.py +17 -0
  125. tensorbored/compat/tensorflow_stub/io/gfile.py +1011 -0
  126. tensorbored/compat/tensorflow_stub/pywrap_tensorflow.py +285 -0
  127. tensorbored/compat/tensorflow_stub/tensor_shape.py +1035 -0
  128. tensorbored/context.py +129 -0
  129. tensorbored/data/__init__.py +0 -0
  130. tensorbored/data/grpc_provider.py +365 -0
  131. tensorbored/data/ingester.py +46 -0
  132. tensorbored/data/proto/__init__.py +0 -0
  133. tensorbored/data/proto/data_provider_pb2.py +517 -0
  134. tensorbored/data/proto/data_provider_pb2_grpc.py +374 -0
  135. tensorbored/data/provider.py +1365 -0
  136. tensorbored/data/server_ingester.py +301 -0
  137. tensorbored/data_compat.py +159 -0
  138. tensorbored/dataclass_compat.py +224 -0
  139. tensorbored/default.py +124 -0
  140. tensorbored/errors.py +130 -0
  141. tensorbored/lazy.py +99 -0
  142. tensorbored/main.py +48 -0
  143. tensorbored/main_lib.py +62 -0
  144. tensorbored/manager.py +487 -0
  145. tensorbored/notebook.py +441 -0
  146. tensorbored/plugin_util.py +266 -0
  147. tensorbored/plugins/__init__.py +0 -0
  148. tensorbored/plugins/audio/__init__.py +0 -0
  149. tensorbored/plugins/audio/audio_plugin.py +229 -0
  150. tensorbored/plugins/audio/metadata.py +69 -0
  151. tensorbored/plugins/audio/plugin_data_pb2.py +37 -0
  152. tensorbored/plugins/audio/summary.py +230 -0
  153. tensorbored/plugins/audio/summary_v2.py +124 -0
  154. tensorbored/plugins/base_plugin.py +367 -0
  155. tensorbored/plugins/core/__init__.py +0 -0
  156. tensorbored/plugins/core/core_plugin.py +981 -0
  157. tensorbored/plugins/custom_scalar/__init__.py +0 -0
  158. tensorbored/plugins/custom_scalar/custom_scalars_plugin.py +320 -0
  159. tensorbored/plugins/custom_scalar/layout_pb2.py +85 -0
  160. tensorbored/plugins/custom_scalar/metadata.py +35 -0
  161. tensorbored/plugins/custom_scalar/summary.py +79 -0
  162. tensorbored/plugins/debugger_v2/__init__.py +0 -0
  163. tensorbored/plugins/debugger_v2/debug_data_multiplexer.py +631 -0
  164. tensorbored/plugins/debugger_v2/debug_data_provider.py +634 -0
  165. tensorbored/plugins/debugger_v2/debugger_v2_plugin.py +504 -0
  166. tensorbored/plugins/distribution/__init__.py +0 -0
  167. tensorbored/plugins/distribution/compressor.py +158 -0
  168. tensorbored/plugins/distribution/distributions_plugin.py +116 -0
  169. tensorbored/plugins/distribution/metadata.py +19 -0
  170. tensorbored/plugins/graph/__init__.py +0 -0
  171. tensorbored/plugins/graph/graph_util.py +129 -0
  172. tensorbored/plugins/graph/graphs_plugin.py +336 -0
  173. tensorbored/plugins/graph/keras_util.py +328 -0
  174. tensorbored/plugins/graph/metadata.py +42 -0
  175. tensorbored/plugins/histogram/__init__.py +0 -0
  176. tensorbored/plugins/histogram/histograms_plugin.py +144 -0
  177. tensorbored/plugins/histogram/metadata.py +63 -0
  178. tensorbored/plugins/histogram/plugin_data_pb2.py +34 -0
  179. tensorbored/plugins/histogram/summary.py +234 -0
  180. tensorbored/plugins/histogram/summary_v2.py +292 -0
  181. tensorbored/plugins/hparams/__init__.py +14 -0
  182. tensorbored/plugins/hparams/_keras.py +93 -0
  183. tensorbored/plugins/hparams/api.py +130 -0
  184. tensorbored/plugins/hparams/api_pb2.py +208 -0
  185. tensorbored/plugins/hparams/backend_context.py +606 -0
  186. tensorbored/plugins/hparams/download_data.py +158 -0
  187. tensorbored/plugins/hparams/error.py +26 -0
  188. tensorbored/plugins/hparams/get_experiment.py +71 -0
  189. tensorbored/plugins/hparams/hparams_plugin.py +206 -0
  190. tensorbored/plugins/hparams/hparams_util_pb2.py +69 -0
  191. tensorbored/plugins/hparams/json_format_compat.py +38 -0
  192. tensorbored/plugins/hparams/list_metric_evals.py +57 -0
  193. tensorbored/plugins/hparams/list_session_groups.py +1040 -0
  194. tensorbored/plugins/hparams/metadata.py +125 -0
  195. tensorbored/plugins/hparams/metrics.py +41 -0
  196. tensorbored/plugins/hparams/plugin_data_pb2.py +69 -0
  197. tensorbored/plugins/hparams/summary.py +205 -0
  198. tensorbored/plugins/hparams/summary_v2.py +597 -0
  199. tensorbored/plugins/image/__init__.py +0 -0
  200. tensorbored/plugins/image/images_plugin.py +232 -0
  201. tensorbored/plugins/image/metadata.py +65 -0
  202. tensorbored/plugins/image/plugin_data_pb2.py +34 -0
  203. tensorbored/plugins/image/summary.py +159 -0
  204. tensorbored/plugins/image/summary_v2.py +130 -0
  205. tensorbored/plugins/mesh/__init__.py +14 -0
  206. tensorbored/plugins/mesh/mesh_plugin.py +292 -0
  207. tensorbored/plugins/mesh/metadata.py +152 -0
  208. tensorbored/plugins/mesh/plugin_data_pb2.py +37 -0
  209. tensorbored/plugins/mesh/summary.py +251 -0
  210. tensorbored/plugins/mesh/summary_v2.py +214 -0
  211. tensorbored/plugins/metrics/__init__.py +0 -0
  212. tensorbored/plugins/metrics/metadata.py +17 -0
  213. tensorbored/plugins/metrics/metrics_plugin.py +623 -0
  214. tensorbored/plugins/pr_curve/__init__.py +0 -0
  215. tensorbored/plugins/pr_curve/metadata.py +75 -0
  216. tensorbored/plugins/pr_curve/plugin_data_pb2.py +34 -0
  217. tensorbored/plugins/pr_curve/pr_curves_plugin.py +241 -0
  218. tensorbored/plugins/pr_curve/summary.py +574 -0
  219. tensorbored/plugins/profile_redirect/__init__.py +0 -0
  220. tensorbored/plugins/profile_redirect/profile_redirect_plugin.py +49 -0
  221. tensorbored/plugins/projector/__init__.py +67 -0
  222. tensorbored/plugins/projector/metadata.py +26 -0
  223. tensorbored/plugins/projector/projector_config_pb2.py +54 -0
  224. tensorbored/plugins/projector/projector_plugin.py +795 -0
  225. tensorbored/plugins/projector/tf_projector_plugin/index.js +32 -0
  226. tensorbored/plugins/projector/tf_projector_plugin/projector_binary.html +524 -0
  227. tensorbored/plugins/projector/tf_projector_plugin/projector_binary.js +15536 -0
  228. tensorbored/plugins/scalar/__init__.py +0 -0
  229. tensorbored/plugins/scalar/metadata.py +60 -0
  230. tensorbored/plugins/scalar/plugin_data_pb2.py +34 -0
  231. tensorbored/plugins/scalar/scalars_plugin.py +181 -0
  232. tensorbored/plugins/scalar/summary.py +109 -0
  233. tensorbored/plugins/scalar/summary_v2.py +124 -0
  234. tensorbored/plugins/text/__init__.py +0 -0
  235. tensorbored/plugins/text/metadata.py +62 -0
  236. tensorbored/plugins/text/plugin_data_pb2.py +34 -0
  237. tensorbored/plugins/text/summary.py +114 -0
  238. tensorbored/plugins/text/summary_v2.py +124 -0
  239. tensorbored/plugins/text/text_plugin.py +288 -0
  240. tensorbored/plugins/wit_redirect/__init__.py +0 -0
  241. tensorbored/plugins/wit_redirect/wit_redirect_plugin.py +49 -0
  242. tensorbored/program.py +910 -0
  243. tensorbored/summary/__init__.py +35 -0
  244. tensorbored/summary/_output.py +124 -0
  245. tensorbored/summary/_tf/__init__.py +14 -0
  246. tensorbored/summary/_tf/summary/__init__.py +178 -0
  247. tensorbored/summary/_writer.py +105 -0
  248. tensorbored/summary/v1.py +51 -0
  249. tensorbored/summary/v2.py +25 -0
  250. tensorbored/summary/writer/__init__.py +13 -0
  251. tensorbored/summary/writer/event_file_writer.py +291 -0
  252. tensorbored/summary/writer/record_writer.py +50 -0
  253. tensorbored/util/__init__.py +0 -0
  254. tensorbored/util/encoder.py +116 -0
  255. tensorbored/util/grpc_util.py +311 -0
  256. tensorbored/util/img_mime_type_detector.py +40 -0
  257. tensorbored/util/io_util.py +20 -0
  258. tensorbored/util/lazy_tensor_creator.py +110 -0
  259. tensorbored/util/op_evaluator.py +104 -0
  260. tensorbored/util/platform_util.py +20 -0
  261. tensorbored/util/tb_logging.py +24 -0
  262. tensorbored/util/tensor_util.py +617 -0
  263. tensorbored/util/timing.py +122 -0
  264. tensorbored/version.py +21 -0
  265. tensorbored/webfiles.zip +0 -0
  266. tensorbored-2.21.0rc1769983804.dist-info/METADATA +49 -0
  267. tensorbored-2.21.0rc1769983804.dist-info/RECORD +271 -0
  268. tensorbored-2.21.0rc1769983804.dist-info/WHEEL +5 -0
  269. tensorbored-2.21.0rc1769983804.dist-info/entry_points.txt +6 -0
  270. tensorbored-2.21.0rc1769983804.dist-info/licenses/LICENSE +739 -0
  271. tensorbored-2.21.0rc1769983804.dist-info/top_level.txt +1 -0
@@ -0,0 +1,130 @@
1
+ # Copyright 2019 The TensorFlow Authors. All Rights Reserved.
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ==============================================================================
15
+ """Public APIs for the HParams plugin.
16
+
17
+ This module supports a spectrum of use cases, depending on how much
18
+ structure you want. In the simplest case, you can simply collect your
19
+ hparams into a dict, and use a Keras callback to record them:
20
+
21
+ >>> from tensorbored.plugins.hparams import api as hp
22
+ >>> hparams = {
23
+ ... "optimizer": "adam",
24
+ ... "fc_dropout": 0.2,
25
+ ... "neurons": 128,
26
+ ... # ...
27
+ ... }
28
+ >>>
29
+ >>> model = model_fn(hparams)
30
+ >>> callbacks = [
31
+ >>> tf.keras.callbacks.TensorBoard(logdir),
32
+ >>> hp.KerasCallback(logdir, hparams),
33
+ >>> ]
34
+ >>> model.fit(..., callbacks=callbacks)
35
+
36
+ The Keras callback requires that TensorFlow eager execution be enabled.
37
+
38
+ If not using Keras, use the `hparams` function to write the values
39
+ directly:
40
+
41
+ >>> # In eager mode:
42
+ >>> with tf.create_file_writer(logdir).as_default():
43
+ ... hp.hparams(hparams)
44
+ >>>
45
+ >>> # In legacy graph mode:
46
+ >>> with tf.compat.v2.create_file_writer(logdir).as_default() as w:
47
+ ... sess.run(w.init())
48
+ ... sess.run(hp.hparams(hparams))
49
+ ... sess.run(w.flush())
50
+
51
+ To control how hyperparameters and metrics appear in the TensorBoard UI,
52
+ you can define `HParam` and `Metric` objects, and write an experiment
53
+ summary to the top-level log directory:
54
+
55
+ >>> HP_OPTIMIZER = hp.HParam("optimizer")
56
+ >>> HP_FC_DROPOUT = hp.HParam(
57
+ ... "fc_dropout",
58
+ ... display_name="f.c. dropout",
59
+ ... description="Dropout rate for fully connected subnet.",
60
+ ... )
61
+ >>> HP_NEURONS = hp.HParam("neurons", description="Neurons per dense layer")
62
+ >>>
63
+ >>> with tf.summary.create_file_writer(base_logdir).as_default():
64
+ ... hp.hparams_config(
65
+ ... hparams=[
66
+ ... HP_OPTIMIZER,
67
+ ... HP_FC_DROPOUT,
68
+ ... HP_NEURONS,
69
+ ... ],
70
+ ... metrics=[
71
+ ... hp.Metric("xent", group="validation", display_name="cross-entropy"),
72
+ ... hp.Metric("f1", group="validation", display_name="F₁ score"),
73
+ ... hp.Metric("loss", group="train", display_name="training loss"),
74
+ ... ],
75
+ ... )
76
+
77
+ You can continue to pass a string-keyed dict to the Keras callback or
78
+ the `hparams` function, or you can use `HParam` objects as the keys. The
79
+ latter approach enables better static analysis: your favorite Python
80
+ linter can tell you if you misspell a hyperparameter name, your IDE can
81
+ help you find all the places where a hyperparameter is used, etc:
82
+
83
+ >>> hparams = {
84
+ ... HP_OPTIMIZER: "adam",
85
+ ... HP_FC_DROPOUT: 0.2,
86
+ ... HP_NEURONS: 128,
87
+ ... # ...
88
+ ... }
89
+ >>>
90
+ >>> model = model_fn(hparams)
91
+ >>> callbacks = [
92
+ >>> tf.keras.callbacks.TensorBoard(logdir),
93
+ >>> hp.KerasCallback(logdir, hparams),
94
+ >>> ]
95
+
96
+ Finally, you can choose to annotate your hparam definitions with domain
97
+ information:
98
+
99
+ >>> HP_OPTIMIZER = hp.HParam("optimizer", hp.Discrete(["adam", "sgd"]))
100
+ >>> HP_FC_DROPOUT = hp.HParam("fc_dropout", hp.RealInterval(0.1, 0.4))
101
+ >>> HP_NEURONS = hp.HParam("neurons", hp.IntInterval(64, 256))
102
+
103
+ The TensorBoard HParams plugin does not provide tuners, but you can
104
+ integrate these domains into your preferred tuning framework if you so
105
+ desire. The domains will also be reflected in the TensorBoard UI.
106
+
107
+ See the `Experiment`, `HParam`, `Metric`, and `KerasCallback` classes
108
+ for API specifications. Consult the `hparams_demo.py` script in the
109
+ TensorBoard repository for an end-to-end MNIST example.
110
+ """
111
+
112
+ from tensorbored.plugins.hparams import _keras
113
+ from tensorbored.plugins.hparams import summary_v2
114
+
115
+ Discrete = summary_v2.Discrete
116
+ Domain = summary_v2.Domain
117
+ HParam = summary_v2.HParam
118
+ IntInterval = summary_v2.IntInterval
119
+ Metric = summary_v2.Metric
120
+ RealInterval = summary_v2.RealInterval
121
+ hparams = summary_v2.hparams
122
+ hparams_pb = summary_v2.hparams_pb
123
+ hparams_config = summary_v2.hparams_config
124
+ hparams_config_pb = summary_v2.hparams_config_pb
125
+
126
+ KerasCallback = _keras.Callback
127
+
128
+
129
+ del _keras
130
+ del summary_v2
@@ -0,0 +1,208 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: tensorbored/plugins/hparams/api.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf.internal import enum_type_wrapper
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pool as _descriptor_pool
8
+ from google.protobuf import message as _message
9
+ from google.protobuf import reflection as _reflection
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ # @@protoc_insertion_point(imports)
12
+
13
+ _sym_db = _symbol_database.Default()
14
+
15
+
16
+ from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
17
+
18
+
19
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%tensorbored/plugins/hparams/api.proto\x12\x13tensorbored.hparams\x1a\x1cgoogle/protobuf/struct.proto\"\xc6\x01\n\nExperiment\x12\x0c\n\x04name\x18\x06 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x01 \x01(\t\x12\x0c\n\x04user\x18\x02 \x01(\t\x12\x19\n\x11time_created_secs\x18\x03 \x01(\x01\x12\x35\n\x0chparam_infos\x18\x04 \x03(\x0b\x32\x1f.tensorbored.hparams.HParamInfo\x12\x35\n\x0cmetric_infos\x18\x05 \x03(\x0b\x32\x1f.tensorbored.hparams.MetricInfo\"\xfe\x01\n\nHParamInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12+\n\x04type\x18\x04 \x01(\x0e\x32\x1d.tensorbored.hparams.DataType\x12\x35\n\x0f\x64omain_discrete\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.ListValueH\x00\x12\x38\n\x0f\x64omain_interval\x18\x06 \x01(\x0b\x32\x1d.tensorbored.hparams.IntervalH\x00\x12\x0f\n\x07\x64iffers\x18\x07 \x01(\x08\x42\x08\n\x06\x64omain\"0\n\x08Interval\x12\x11\n\tmin_value\x18\x01 \x01(\x01\x12\x11\n\tmax_value\x18\x02 \x01(\x01\"(\n\nMetricName\x12\r\n\x05group\x18\x01 \x01(\t\x12\x0b\n\x03tag\x18\x02 \x01(\t\"\x9e\x01\n\nMetricInfo\x12-\n\x04name\x18\x01 \x01(\x0b\x32\x1f.tensorbored.hparams.MetricName\x12\x14\n\x0c\x64isplay_name\x18\x03 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x36\n\x0c\x64\x61taset_type\x18\x05 \x01(\x0e\x32 .tensorbored.hparams.DatasetType\"\xa3\x02\n\x0cSessionGroup\x12\x0c\n\x04name\x18\x01 \x01(\t\x12?\n\x07hparams\x18\x02 \x03(\x0b\x32..tensorbored.hparams.SessionGroup.HparamsEntry\x12\x37\n\rmetric_values\x18\x03 \x03(\x0b\x32 .tensorbored.hparams.MetricValue\x12.\n\x08sessions\x18\x04 \x03(\x0b\x32\x1c.tensorbored.hparams.Session\x12\x13\n\x0bmonitor_url\x18\x05 \x01(\t\x1a\x46\n\x0cHparamsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.Value:\x02\x38\x01\"z\n\x0bMetricValue\x12-\n\x04name\x18\x01 \x01(\x0b\x32\x1f.tensorbored.hparams.MetricName\x12\r\n\x05value\x18\x02 \x01(\x01\x12\x15\n\rtraining_step\x18\x03 \x01(\x05\x12\x16\n\x0ewall_time_secs\x18\x04 \x01(\x01\"\xd5\x01\n\x07Session\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x17\n\x0fstart_time_secs\x18\x02 \x01(\x01\x12\x15\n\rend_time_secs\x18\x03 \x01(\x01\x12+\n\x06status\x18\x04 \x01(\x0e\x32\x1b.tensorbored.hparams.Status\x12\x11\n\tmodel_uri\x18\x05 \x01(\t\x12\x37\n\rmetric_values\x18\x06 \x03(\x0b\x32 .tensorbored.hparams.MetricValue\x12\x13\n\x0bmonitor_url\x18\x07 \x01(\t\"\x8f\x01\n\x14GetExperimentRequest\x12\x17\n\x0f\x65xperiment_name\x18\x01 \x01(\t\x12\x1c\n\x0finclude_metrics\x18\x02 \x01(\x08H\x00\x88\x01\x01\x12\x1a\n\rhparams_limit\x18\x03 \x01(\x05H\x01\x88\x01\x01\x42\x12\n\x10_include_metricsB\x10\n\x0e_hparams_limit\"\xf6\x02\n\x18ListSessionGroupsRequest\x12\x17\n\x0f\x65xperiment_name\x18\x06 \x01(\t\x12\x35\n\x10\x61llowed_statuses\x18\x07 \x03(\x0e\x32\x1b.tensorbored.hparams.Status\x12\x32\n\ncol_params\x18\x01 \x03(\x0b\x32\x1e.tensorbored.hparams.ColParams\x12>\n\x10\x61ggregation_type\x18\x02 \x01(\x0e\x32$.tensorbored.hparams.AggregationType\x12;\n\x12\x61ggregation_metric\x18\x03 \x01(\x0b\x32\x1f.tensorbored.hparams.MetricName\x12\x13\n\x0bstart_index\x18\x04 \x01(\x05\x12\x12\n\nslice_size\x18\x05 \x01(\x05\x12\x1c\n\x0finclude_metrics\x18\x08 \x01(\x08H\x00\x88\x01\x01\x42\x12\n\x10_include_metrics\"\x8f\x03\n\tColParams\x12\x31\n\x06metric\x18\x01 \x01(\x0b\x32\x1f.tensorbored.hparams.MetricNameH\x00\x12\x10\n\x06hparam\x18\x02 \x01(\tH\x00\x12-\n\x05order\x18\x03 \x01(\x0e\x32\x1e.tensorbored.hparams.SortOrder\x12\x1c\n\x14missing_values_first\x18\x04 \x01(\x08\x12\x17\n\rfilter_regexp\x18\x05 \x01(\tH\x01\x12\x38\n\x0f\x66ilter_interval\x18\x06 \x01(\x0b\x32\x1d.tensorbored.hparams.IntervalH\x01\x12\x35\n\x0f\x66ilter_discrete\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.ListValueH\x01\x12\x1e\n\x16\x65xclude_missing_values\x18\x08 \x01(\x08\x12\x1e\n\x11include_in_result\x18\t \x01(\x08H\x02\x88\x01\x01\x42\x06\n\x04nameB\x08\n\x06\x66ilterB\x14\n\x12_include_in_result\"j\n\x19ListSessionGroupsResponse\x12\x39\n\x0esession_groups\x18\x01 \x03(\x0b\x32!.tensorbored.hparams.SessionGroup\x12\x12\n\ntotal_size\x18\x03 \x01(\x05\"}\n\x16ListMetricEvalsRequest\x12\x17\n\x0f\x65xperiment_name\x18\x03 \x01(\t\x12\x14\n\x0csession_name\x18\x01 \x01(\t\x12\x34\n\x0bmetric_name\x18\x02 \x01(\x0b\x32\x1f.tensorbored.hparams.MetricName*`\n\x08\x44\x61taType\x12\x13\n\x0f\x44\x41TA_TYPE_UNSET\x10\x00\x12\x14\n\x10\x44\x41TA_TYPE_STRING\x10\x01\x12\x12\n\x0e\x44\x41TA_TYPE_BOOL\x10\x02\x12\x15\n\x11\x44\x41TA_TYPE_FLOAT64\x10\x03*P\n\x0b\x44\x61tasetType\x12\x13\n\x0f\x44\x41TASET_UNKNOWN\x10\x00\x12\x14\n\x10\x44\x41TASET_TRAINING\x10\x01\x12\x16\n\x12\x44\x41TASET_VALIDATION\x10\x02*X\n\x06Status\x12\x12\n\x0eSTATUS_UNKNOWN\x10\x00\x12\x12\n\x0eSTATUS_SUCCESS\x10\x01\x12\x12\n\x0eSTATUS_FAILURE\x10\x02\x12\x12\n\x0eSTATUS_RUNNING\x10\x03*A\n\tSortOrder\x12\x15\n\x11ORDER_UNSPECIFIED\x10\x00\x12\r\n\tORDER_ASC\x10\x01\x12\x0e\n\nORDER_DESC\x10\x02*\x7f\n\x0f\x41ggregationType\x12\x15\n\x11\x41GGREGATION_UNSET\x10\x00\x12\x13\n\x0f\x41GGREGATION_AVG\x10\x01\x12\x16\n\x12\x41GGREGATION_MEDIAN\x10\x02\x12\x13\n\x0f\x41GGREGATION_MIN\x10\x03\x12\x13\n\x0f\x41GGREGATION_MAX\x10\x04\x62\x06proto3')
20
+
21
+ _DATATYPE = DESCRIPTOR.enum_types_by_name['DataType']
22
+ DataType = enum_type_wrapper.EnumTypeWrapper(_DATATYPE)
23
+ _DATASETTYPE = DESCRIPTOR.enum_types_by_name['DatasetType']
24
+ DatasetType = enum_type_wrapper.EnumTypeWrapper(_DATASETTYPE)
25
+ _STATUS = DESCRIPTOR.enum_types_by_name['Status']
26
+ Status = enum_type_wrapper.EnumTypeWrapper(_STATUS)
27
+ _SORTORDER = DESCRIPTOR.enum_types_by_name['SortOrder']
28
+ SortOrder = enum_type_wrapper.EnumTypeWrapper(_SORTORDER)
29
+ _AGGREGATIONTYPE = DESCRIPTOR.enum_types_by_name['AggregationType']
30
+ AggregationType = enum_type_wrapper.EnumTypeWrapper(_AGGREGATIONTYPE)
31
+ DATA_TYPE_UNSET = 0
32
+ DATA_TYPE_STRING = 1
33
+ DATA_TYPE_BOOL = 2
34
+ DATA_TYPE_FLOAT64 = 3
35
+ DATASET_UNKNOWN = 0
36
+ DATASET_TRAINING = 1
37
+ DATASET_VALIDATION = 2
38
+ STATUS_UNKNOWN = 0
39
+ STATUS_SUCCESS = 1
40
+ STATUS_FAILURE = 2
41
+ STATUS_RUNNING = 3
42
+ ORDER_UNSPECIFIED = 0
43
+ ORDER_ASC = 1
44
+ ORDER_DESC = 2
45
+ AGGREGATION_UNSET = 0
46
+ AGGREGATION_AVG = 1
47
+ AGGREGATION_MEDIAN = 2
48
+ AGGREGATION_MIN = 3
49
+ AGGREGATION_MAX = 4
50
+
51
+
52
+ _EXPERIMENT = DESCRIPTOR.message_types_by_name['Experiment']
53
+ _HPARAMINFO = DESCRIPTOR.message_types_by_name['HParamInfo']
54
+ _INTERVAL = DESCRIPTOR.message_types_by_name['Interval']
55
+ _METRICNAME = DESCRIPTOR.message_types_by_name['MetricName']
56
+ _METRICINFO = DESCRIPTOR.message_types_by_name['MetricInfo']
57
+ _SESSIONGROUP = DESCRIPTOR.message_types_by_name['SessionGroup']
58
+ _SESSIONGROUP_HPARAMSENTRY = _SESSIONGROUP.nested_types_by_name['HparamsEntry']
59
+ _METRICVALUE = DESCRIPTOR.message_types_by_name['MetricValue']
60
+ _SESSION = DESCRIPTOR.message_types_by_name['Session']
61
+ _GETEXPERIMENTREQUEST = DESCRIPTOR.message_types_by_name['GetExperimentRequest']
62
+ _LISTSESSIONGROUPSREQUEST = DESCRIPTOR.message_types_by_name['ListSessionGroupsRequest']
63
+ _COLPARAMS = DESCRIPTOR.message_types_by_name['ColParams']
64
+ _LISTSESSIONGROUPSRESPONSE = DESCRIPTOR.message_types_by_name['ListSessionGroupsResponse']
65
+ _LISTMETRICEVALSREQUEST = DESCRIPTOR.message_types_by_name['ListMetricEvalsRequest']
66
+ Experiment = _reflection.GeneratedProtocolMessageType('Experiment', (_message.Message,), {
67
+ 'DESCRIPTOR' : _EXPERIMENT,
68
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
69
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.Experiment)
70
+ })
71
+ _sym_db.RegisterMessage(Experiment)
72
+
73
+ HParamInfo = _reflection.GeneratedProtocolMessageType('HParamInfo', (_message.Message,), {
74
+ 'DESCRIPTOR' : _HPARAMINFO,
75
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
76
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.HParamInfo)
77
+ })
78
+ _sym_db.RegisterMessage(HParamInfo)
79
+
80
+ Interval = _reflection.GeneratedProtocolMessageType('Interval', (_message.Message,), {
81
+ 'DESCRIPTOR' : _INTERVAL,
82
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
83
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.Interval)
84
+ })
85
+ _sym_db.RegisterMessage(Interval)
86
+
87
+ MetricName = _reflection.GeneratedProtocolMessageType('MetricName', (_message.Message,), {
88
+ 'DESCRIPTOR' : _METRICNAME,
89
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
90
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.MetricName)
91
+ })
92
+ _sym_db.RegisterMessage(MetricName)
93
+
94
+ MetricInfo = _reflection.GeneratedProtocolMessageType('MetricInfo', (_message.Message,), {
95
+ 'DESCRIPTOR' : _METRICINFO,
96
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
97
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.MetricInfo)
98
+ })
99
+ _sym_db.RegisterMessage(MetricInfo)
100
+
101
+ SessionGroup = _reflection.GeneratedProtocolMessageType('SessionGroup', (_message.Message,), {
102
+
103
+ 'HparamsEntry' : _reflection.GeneratedProtocolMessageType('HparamsEntry', (_message.Message,), {
104
+ 'DESCRIPTOR' : _SESSIONGROUP_HPARAMSENTRY,
105
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
106
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.SessionGroup.HparamsEntry)
107
+ })
108
+ ,
109
+ 'DESCRIPTOR' : _SESSIONGROUP,
110
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
111
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.SessionGroup)
112
+ })
113
+ _sym_db.RegisterMessage(SessionGroup)
114
+ _sym_db.RegisterMessage(SessionGroup.HparamsEntry)
115
+
116
+ MetricValue = _reflection.GeneratedProtocolMessageType('MetricValue', (_message.Message,), {
117
+ 'DESCRIPTOR' : _METRICVALUE,
118
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
119
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.MetricValue)
120
+ })
121
+ _sym_db.RegisterMessage(MetricValue)
122
+
123
+ Session = _reflection.GeneratedProtocolMessageType('Session', (_message.Message,), {
124
+ 'DESCRIPTOR' : _SESSION,
125
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
126
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.Session)
127
+ })
128
+ _sym_db.RegisterMessage(Session)
129
+
130
+ GetExperimentRequest = _reflection.GeneratedProtocolMessageType('GetExperimentRequest', (_message.Message,), {
131
+ 'DESCRIPTOR' : _GETEXPERIMENTREQUEST,
132
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
133
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.GetExperimentRequest)
134
+ })
135
+ _sym_db.RegisterMessage(GetExperimentRequest)
136
+
137
+ ListSessionGroupsRequest = _reflection.GeneratedProtocolMessageType('ListSessionGroupsRequest', (_message.Message,), {
138
+ 'DESCRIPTOR' : _LISTSESSIONGROUPSREQUEST,
139
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
140
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.ListSessionGroupsRequest)
141
+ })
142
+ _sym_db.RegisterMessage(ListSessionGroupsRequest)
143
+
144
+ ColParams = _reflection.GeneratedProtocolMessageType('ColParams', (_message.Message,), {
145
+ 'DESCRIPTOR' : _COLPARAMS,
146
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
147
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.ColParams)
148
+ })
149
+ _sym_db.RegisterMessage(ColParams)
150
+
151
+ ListSessionGroupsResponse = _reflection.GeneratedProtocolMessageType('ListSessionGroupsResponse', (_message.Message,), {
152
+ 'DESCRIPTOR' : _LISTSESSIONGROUPSRESPONSE,
153
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
154
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.ListSessionGroupsResponse)
155
+ })
156
+ _sym_db.RegisterMessage(ListSessionGroupsResponse)
157
+
158
+ ListMetricEvalsRequest = _reflection.GeneratedProtocolMessageType('ListMetricEvalsRequest', (_message.Message,), {
159
+ 'DESCRIPTOR' : _LISTMETRICEVALSREQUEST,
160
+ '__module__' : 'tensorbored.plugins.hparams.api_pb2'
161
+ # @@protoc_insertion_point(class_scope:tensorbored.hparams.ListMetricEvalsRequest)
162
+ })
163
+ _sym_db.RegisterMessage(ListMetricEvalsRequest)
164
+
165
+ if _descriptor._USE_C_DESCRIPTORS == False:
166
+
167
+ DESCRIPTOR._options = None
168
+ _SESSIONGROUP_HPARAMSENTRY._options = None
169
+ _SESSIONGROUP_HPARAMSENTRY._serialized_options = b'8\001'
170
+ _DATATYPE._serialized_start=2597
171
+ _DATATYPE._serialized_end=2693
172
+ _DATASETTYPE._serialized_start=2695
173
+ _DATASETTYPE._serialized_end=2775
174
+ _STATUS._serialized_start=2777
175
+ _STATUS._serialized_end=2865
176
+ _SORTORDER._serialized_start=2867
177
+ _SORTORDER._serialized_end=2932
178
+ _AGGREGATIONTYPE._serialized_start=2934
179
+ _AGGREGATIONTYPE._serialized_end=3061
180
+ _EXPERIMENT._serialized_start=93
181
+ _EXPERIMENT._serialized_end=291
182
+ _HPARAMINFO._serialized_start=294
183
+ _HPARAMINFO._serialized_end=548
184
+ _INTERVAL._serialized_start=550
185
+ _INTERVAL._serialized_end=598
186
+ _METRICNAME._serialized_start=600
187
+ _METRICNAME._serialized_end=640
188
+ _METRICINFO._serialized_start=643
189
+ _METRICINFO._serialized_end=801
190
+ _SESSIONGROUP._serialized_start=804
191
+ _SESSIONGROUP._serialized_end=1095
192
+ _SESSIONGROUP_HPARAMSENTRY._serialized_start=1025
193
+ _SESSIONGROUP_HPARAMSENTRY._serialized_end=1095
194
+ _METRICVALUE._serialized_start=1097
195
+ _METRICVALUE._serialized_end=1219
196
+ _SESSION._serialized_start=1222
197
+ _SESSION._serialized_end=1435
198
+ _GETEXPERIMENTREQUEST._serialized_start=1438
199
+ _GETEXPERIMENTREQUEST._serialized_end=1581
200
+ _LISTSESSIONGROUPSREQUEST._serialized_start=1584
201
+ _LISTSESSIONGROUPSREQUEST._serialized_end=1958
202
+ _COLPARAMS._serialized_start=1961
203
+ _COLPARAMS._serialized_end=2360
204
+ _LISTSESSIONGROUPSRESPONSE._serialized_start=2362
205
+ _LISTSESSIONGROUPSRESPONSE._serialized_end=2468
206
+ _LISTMETRICEVALSREQUEST._serialized_start=2470
207
+ _LISTMETRICEVALSREQUEST._serialized_end=2595
208
+ # @@protoc_insertion_point(module_scope)