def __init__(self,
read_queue: Queue,
write_queue: Queue,
use_gpu: bool = False):
MLModelWorker.__init__(self,
name='SentenceStructureModelWorker',
read_queue=read_queue,
write_queue=write_queue,
use_gpu=use_gpu)
def run(self):
self._model = AOLReactionModel(use_gpu=self._use_gpu)
MLModelWorker.run(self)
def __init__(self, read_queue: Queue, write_queue: Queue, use_gpu: bool = False):
MLModelWorker.__init__(self, name='AOLReactionModelWorker', read_queue=read_queue, write_queue=write_queue,
use_gpu=use_gpu)
def run(self):
self._model = StructureModel(use_gpu=self._use_gpu)
MLModelWorker.run(self)
def run(self):
self._model = StructureModel(use_gpu=self._use_gpu)
MLModelWorker.run(self)
def __init__(self, read_queue: Queue, write_queue: Queue, use_gpu: bool = False):
MLModelWorker.__init__(self, name='SentenceStructureModelWorker', read_queue=read_queue,
write_queue=write_queue,
use_gpu=use_gpu)