Merge pull request #1588 from FedML-AI/raphael/fix-fediot

[FedIot] Fix Test Method Bug

iot/anomaly_detection_for_cybersecurity/config/fedml_config.yaml

@@ -32,7 +32,7 @@ train_args:
   weight_decay: 0.001
 
 validation_args:
-  frequency_of_the_test: 5
+  frequency_of_the_test: 1
 
 device_args:
   worker_num: 2

iot/anomaly_detection_for_cybersecurity/trainer/fed_detect_aggregator.py

@@ -4,6 +4,7 @@
 import numpy as np
 import pandas as pd
 import torch
+import fedml
 from torch import nn
 
 from fedml.core import ServerAggregator
@@ -20,7 +21,7 @@ def set_model_params(self, model_parameters):
     def test(self, test_data, device, args):
         pass
 
-    def test_on_the_server(
+    def test_all(
         self, train_data_local_dict, test_data_local_dict, device, args=None
     ) -> bool:
         model = self.model
@@ -67,6 +68,9 @@ def test_on_the_server(
         logging.info("The True positive number is {}".format(true_positive))
         logging.info("The False negative number is {}".format(false_negative))
 
+        logging.info(f"The type of auc is {type(accuracy)}")
+        fedml.mlops.log({"round_idx": args.round_idx, "accuracy": float(accuracy)})
+
         logging.info("The accuracy is {}".format(accuracy))
         logging.info("The precision is {}".format(precision))
         logging.info("The false positive rate is {}".format(false_positive_rate))