def train_model(self, request: Dict) -> Dict:
ws_id = request['workspace_id']
try:
pages = request['pages']
pages = self._fetch_pages_html(pages)
result = train_model(
pages, model_cls=self.model_cls,
progress_callback=partial(self.progress_callback, ws_id=ws_id),
**self.model_kwargs)
except Exception as e:
logging.error('Failed to train a model', exc_info=e)
result = ModelMeta(
model=None,
meta=Meta(advice=[AdviceItem(
ERROR,
'Unknown error while training a model: {}'.format(e))]))
return {
'workspace_id': ws_id,
'quality': json.dumps(attr.asdict(result.meta)),
'model': (encode_object(result.model) if result.model is not None
else None),
}
评论列表
文章目录