40 lines
783 B
Python
40 lines
783 B
Python
# Description:
|
|
# Examples of tf.learn usage
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
py_binary(
|
|
name = "iris_custom_decay_dnn",
|
|
srcs = ["iris_custom_decay_dnn.py"],
|
|
python_version = "PY2",
|
|
srcs_version = "PY2AND3",
|
|
deps = ["//tensorflow:tensorflow_py"],
|
|
)
|
|
|
|
py_binary(
|
|
name = "iris_custom_model",
|
|
srcs = ["iris_custom_model.py"],
|
|
python_version = "PY2",
|
|
srcs_version = "PY2AND3",
|
|
deps = ["//tensorflow:tensorflow_py"],
|
|
)
|
|
|
|
sh_test(
|
|
name = "examples_test",
|
|
size = "large",
|
|
srcs = ["examples_test.sh"],
|
|
data = [
|
|
":iris_custom_decay_dnn",
|
|
":iris_custom_model",
|
|
],
|
|
tags = [
|
|
"manual",
|
|
"notap",
|
|
],
|
|
)
|