8184b9fdf51e3f75835fe1f2d56c294d16686241,examples/federated_learning_with_encryption.py,,,#,162

Before Change

    clients = []
    clients.append(Client("Alice", server.pubkey))
    clients.append(Client("Bob", server.pubkey))
    clients.append(Client("Carol", server.pubkey))

    // Each client trains a linear regressor on its own data
    for (i, c) in enumerate(clients):

After Change

    //     print(mean_square_error(y_pred, y_test))

    // The federated learning with gradient from the google paper
    n_iter = 5
    for i in range(n_iter):

        // Compute gradients, encrypt and aggregate
        encrypt_aggr = clients[0].encrypted_gradient(sum_to=None)
        encrypt_aggr = clients[1].encrypted_gradient(sum_to=encrypt_aggr)
        encrypt_aggr = clients[2].encrypted_gradient(sum_to=encrypt_aggr)

        // Send aggregate to server, which decrypts
        aggr = server.decrypt_aggregate(encrypt_aggr, n_clients)

        // Take gradient steps
        clients[0].gradient_step(aggr)
        clients[1].gradient_step(aggr)
        clients[2].gradient_step(aggr)

    for (i, c) in enumerate(clients):
        y_pred = c.predict(c.X)
        print(mean_square_error(y_pred, c.y))

Instances