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