def test_evaluate_model(): dd = DGADetector() dd.init_model() accuracy = dd.evaluate_model(test_partitioned_dfs, test_dataset_len) assert isinstance(accuracy, (int, float))
def test_evaluate_model(): if torch.cuda.is_available(): dd = DGADetector() dd.init_model() accuracy = dd.evaluate_model(dataset) assert isinstance(accuracy, (int, float))