Overview
| Name | Dataset | Description | Scene | Download | Remark |
|---|---|---|---|---|---|
| agnews_classification | AGNEWS | See here | Horizontal FL | Click Here | - |
| imdb_classification | IMDB | See here | Horizontal FL | Click Here | |
| sst2_classification | SST2 | See here | Horizontal FL | Click Here |
Details
agnews_classification
model
| Model Name | Non-Fed Performance | NumPara | Implementation |
|---|---|---|---|
| TextClassificationModel | - | - |
supported partitioner
| Name | IsDefault | Comments |
|---|---|---|
| IIDPartitioner | yes | |
| DiversityPartitioner | Partitioning according to label diversity | |
| DirichletPartitioner | Partitioning according to dir. distribution of labels |
imdb_classification
-
model
| Model Name | Non-Fed Performance | NumPara | Implementation |
|---|---|---|---|
| TextClassificationModel | - | - |
supported partitioner
| Name | IsDefault | Comments |
|---|---|---|
| IIDPartitioner | yes | |
| DiversityPartitioner | Partitioning according to label diversity | |
| DirichletPartitioner | Partitioning according to dir. distribution of labels |
sst2_classification
-
model
| Model Name | Non-Fed Performance | NumPara | Implementation |
|---|---|---|---|
| TextClassificationModel | - | - |
supported partitioner
| Name | IsDefault | Comments |
|---|---|---|
| IIDPartitioner | yes | |
| DiversityPartitioner | Partitioning according to label diversity | |
| DirichletPartitioner | Partitioning according to dir. distribution of labels |