Add minNbOfParticipants field to trainingInformation

epfml · Aug 20, 2024 · c3e7082 · c3e7082
1 parent 92b51a6
commit c3e7082
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/discojs/src/dataset/data/preprocessing/text_preprocessing.spec.ts b/discojs/src/dataset/data/preprocessing/text_preprocessing.spec.ts
@@ -21,6 +21,7 @@ describe('text preprocessing', function () {
         validationSplit: 0,
         batchSize: 8,
         scheme: 'local',
+        minNbOfParticipants: 1,
         dataType: 'text',
         tokenizer: 'Xenova/gpt2',
         tensorBackend: 'gpt'

diff --git a/docs/examples/custom_task.ts b/docs/examples/custom_task.ts
@@ -30,6 +30,7 @@ const customTask: TaskProvider = {
           'Output'
         ],
         scheme: 'federated',
+        minNbOfParticipants: 2,
         tensorBackend: 'tfjs',
         noiseScale: undefined,
         clippingRadius: undefined

diff --git a/server/tests/client/federated.spec.ts b/server/tests/client/federated.spec.ts
@@ -48,6 +48,7 @@ describe("federated client", () => {
           validationSplit: 0,
           batchSize: 1,
           scheme: "federated",
+          minNbOfParticipants: 2,
           dataType: "tabular",
           tensorBackend: 'tfjs'
         },