zhxchd / vfedcce

A vertical federated learning algorithm for classfication problems with gradient-based optimization.

vFedCCE

A vertical federated learning algorithm for classfication problems with gradient-based optimization.

Data set structure

This algorithm is going to train a shared model on the common IDs (aligned by some private entity resolution technique) between vertically partitioned datasets, as shown in the following figure.

The Proposed vFedCCE Algorithm

The algorithm itself can be described by the following sequence diagram.

Details of the algorithm can be found in the report.

Details of the implementation (with TensorFlow and Keras) can be found under the directory src.

vfedcce's Issues

TypeError: Input 'y' of 'Sub' Op has type float32 that does not match type int64 of argument 'x'.

model.fit(trainX, trainY, epochs=50, verbose=0, batch_size=batch_size), i get the error: TypeError: Input 'y' of 'Sub' Op has type float32 that does not match type int64 of argument 'x'.

How do I fix it?