crypten_vfl_demo

使用crypten做纵向联邦学习demo。详细介绍参考知乎专栏

数据下载

使用 UCI ADULT 数据集，下载数据集后，解压得到训练数据adult.data 和测试数据 adult.test，将adult.data改名为adult.train.csv，将adult.test改名为adult.test.csv，放置到项目目录下

运行data_process.py，进行数据预处理

python data_process.py

运行train_single.py，进行单机训练

python train_single.py

假设有3个参与方，分别为A,B,C

打开三个终端，分别输入命令：

RENDEZVOUS=file:///tmp/vfl WORLD_SIZE=3 RANK=0 python train_multi.py

RENDEZVOUS=file:///tmp/vfl WORLD_SIZE=3 RANK=1 python train_multi.py

RENDEZVOUS=file:///tmp/vfl WORLD_SIZE=3 RANK=2 python train_multi.py

即可开始单机纵向联邦学习

假设有3个参与方，分别为A,B,C

A的IP为192.168.1.100，B的IP为192.168.1.101，C的IP为192.168.1.102，A,B,C可以互相访问将项目文件分别复制到3台机器下，将dataset/a下的数据复制到A中，dataset/b下的数据复制到B中，dataset/c下的文件复制到C中

在A,B,C三台机器的终端中，分别输入命令

RENDEZVOUS=tcp://192.168.1.100:2345 WORLD_SIZE=3 RANK=0 python train_multi.py

RENDEZVOUS=tcp://192.168.1.100:2345 WORLD_SIZE=3 RANK=1 python train_multi.py

RENDEZVOUS=tcp://192.168.1.100:2345 WORLD_SIZE=3 RANK=2 python train_multi.py

即可开始单机纵向联邦学习

单机 vs 纵向联邦学习