def generate_module(): RANDOM_STATE = np.random.RandomState(1031412) fname = os.path.join(os.path.dirname(__file__), "..", "src", "metrics", "test.rs") module = Module(imports=["metrics::ranking::*"]) for num in range(5): name = "roc_auc_test_%s" % num module.add_test(_generate_test(name, RANDOM_STATE)) return fname, module
def generate_module(): RANDOM_STATE = np.random.RandomState(1031412) fname = os.path.join(os.path.dirname(__file__), '..', 'src', 'metrics', 'test.rs') module = Module(imports=['metrics::ranking::*']) for num in range(5): name = 'roc_auc_test_%s' % num module.add_test(_generate_test(name, RANDOM_STATE)) return fname, module
def onEnableChannel(self, channel): Module.onEnableChannel(self, channel) if channel == 'ups_downs': self.addStatusKey('download_payload_rate') self.addStatusKey('upload_payload_rate') elif channel == 'tracker_status': self.addStatusKey('state') self.addStatusKey('tracker_status') self.addStatusKey('download_payload_rate') self.addStatusKey('upload_payload_rate') elif channel == 'downloads': self.addStatusKey('eta') self.addStatusKey('total_done') self.addStatusKey('total_size') self.addStatusKey('download_payload_rate') self.addStatusKey('upload_payload_rate')
def __init__(self, cfg=dict()): Module.__init__(self, 'deluge') self.dc = None self.statusKeys = [] self.result = None self.interval = cfg.get('interval', 15) self.ups = 0 self.downs = 0 self.trackers = {} self.downloads = {} self.torrents = [] self.channels = { 'ups_downs': False, 'trackers': False, 'downloads': False } self.svc = None