Commit 0d736a45 authored by Zachary Charles's avatar Zachary Charles Committed by tensorflow-copybara
Browse files

Add FedProx to the OSS API.

PiperOrigin-RevId: 413722250
parent 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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment