service.py 文件源码

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),
        }