Commit 96264dfe authored by Dr.李's avatar Dr.李

model enhancement

parent f46e445d
......@@ -69,6 +69,7 @@ class LinearRegression(ModelBase):
model_desc['desc'] = encoded.decode('ascii')
model_desc['sklearn_version'] = sklearn_version
model_desc['trained_time'] = self.trained_time
model_desc['coef'] = list(self.impl.coef_)
return model_desc
def score(self) -> float:
......
......@@ -6,6 +6,7 @@ Created on 2017-9-4
"""
import abc
import arrow
import numpy as np
from alphamind.utilities import alpha_logger
......@@ -32,6 +33,7 @@ class ModelBase(metaclass=abc.ABCMeta):
model_desc = dict(model_name=self.__class__.__module__ + "." + self.__class__.__name__,
language='python',
saved_time=arrow.now().format("YYYY-MM-DD HH:mm:ss"),
features=list(self.features))
return model_desc
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment