diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-07-17 14:51:12 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-17 14:55:23 -0700 |
| commit | 1754a2c0c425006324eac85645beb8599366a4f5 (patch) | |
| tree | efa4af51a0ab68d99566f6c76871591ebc3c5840 /tensorflow/contrib/decision_trees | |
| parent | 7bf4e6cbaae9ca930aa17d058c94aa11119fc0c3 (diff) | |
TensorForest v4 kernel tests for proto-based model. Also some stuff to include decision_trees in pip package properly.
PiperOrigin-RevId: 162273792
Diffstat (limited to 'tensorflow/contrib/decision_trees')
| -rw-r--r-- | tensorflow/contrib/decision_trees/BUILD | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/tensorflow/contrib/decision_trees/BUILD b/tensorflow/contrib/decision_trees/BUILD new file mode 100644 index 0000000000..6be69a6f94 --- /dev/null +++ b/tensorflow/contrib/decision_trees/BUILD @@ -0,0 +1,37 @@ +# TensorFlow shared library for training decision trees. +package(default_visibility = [ + "//visibility:public", +]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +filegroup( + name = "all_files", + srcs = glob( + ["*"], + ), + visibility = ["//tensorflow:__subpackages__"], +) + +py_library( + name = "init_py", + srcs = [ + "__init__.py", + ], + srcs_version = "PY2AND3", + deps = [ + "//tensorflow/contrib/decision_trees/proto:generic_tree_model_extensions_py", + "//tensorflow/contrib/decision_trees/proto:generic_tree_model_py", + ], +) + +# Pip + +py_library( + name = "decision_trees_pip", + deps = [ + ":init_py", + ], +) |