[d9566e]: / tests / test_create_sybilnet.py

Download this file

19 lines (13 with data), 371 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
import argparse
import datetime
import os
from sybil import Serie, Sybil
def test_create_sybilnet():
from sybil.models.sybil import SybilNet
fake_args = argparse.Namespace(
dropout=0.1,
max_followup=5,
)
sybil_net = SybilNet(fake_args)
assert sybil_net.hidden_dim == 512
assert sybil_net.prob_of_failure_layer is not None