提交 0d736a45 编辑于 作者: Zachary Charles's avatar Zachary Charles 提交者: tensorflow-copybara
浏览文件

Add FedProx to the OSS API.

PiperOrigin-RevId: 413722250
上级 ebbefcff
......@@ -18,7 +18,10 @@ py_library(
srcs = ["__init__.py"],
srcs_version = "PY3",
visibility = ["//tensorflow_federated/python/learning:__pkg__"],
deps = [":client_scheduled_federated_averaging"],
deps = [
":client_scheduled_federated_averaging",
":fed_prox",
],
)
py_library(
......
......@@ -14,3 +14,4 @@
"""Libraries providing implementations of federated learning algorithms."""
from tensorflow_federated.python.learning.algorithms.client_scheduled_federated_averaging import build_client_scheduled_federated_averaging_process
from tensorflow_federated.python.learning.algorithms.fed_prox import build_example_weighted_fed_prox_process
Supports Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册