def create_and_save_model(self):
model_params = self.job_args['model']['params']
params = base64.b64decode(model_params)
list_params = literal_eval(params)
dataset_format = self.job_args.get('dataset_format')
if dataset_format == 'libsvm':
self.model = self.controller.create_model_libsvm(self.data,
list_params)
elif dataset_format == 'text':
self.model = self.controller.create_model_text(self.data,
list_params)
else:
self.model = self.controller.create_model(self.data, list_params)
if self.model:
self.model.save(self.context, self.modelpath)
评论列表
文章目录