Commit bd2a2d3d authored by Dr.李's avatar Dr.李

update for lasso model

parent 5addc3aa
...@@ -122,7 +122,7 @@ class LassoRegression(ModelBase): ...@@ -122,7 +122,7 @@ class LassoRegression(ModelBase):
@classmethod @classmethod
def load(cls, model_desc: dict): def load(cls, model_desc: dict):
obj_layout = cls() obj_layout = cls(alpha=0.)
obj_layout.features = model_desc['features'] obj_layout.features = model_desc['features']
obj_layout.trained_time = model_desc['trained_time'] obj_layout.trained_time = model_desc['trained_time']
......
...@@ -8,6 +8,7 @@ Created on 2017-9-5 ...@@ -8,6 +8,7 @@ Created on 2017-9-5
from alphamind.model.modelbase import ModelBase from alphamind.model.modelbase import ModelBase
from alphamind.model.linearmodel import ConstLinearModel from alphamind.model.linearmodel import ConstLinearModel
from alphamind.model.linearmodel import LinearRegression from alphamind.model.linearmodel import LinearRegression
from alphamind.model.linearmodel import LassoRegression
def load_model(model_desc: dict) -> ModelBase: def load_model(model_desc: dict) -> ModelBase:
...@@ -19,5 +20,7 @@ def load_model(model_desc: dict) -> ModelBase: ...@@ -19,5 +20,7 @@ def load_model(model_desc: dict) -> ModelBase:
return ConstLinearModel.load(model_desc) return ConstLinearModel.load(model_desc)
elif 'LinearRegression' in model_name_parts: elif 'LinearRegression' in model_name_parts:
return LinearRegression.load(model_desc) return LinearRegression.load(model_desc)
elif 'LassoRegression' in model_name_parts:
return LassoRegression.load(model_desc)
else: else:
raise ValueError('{0} is not currently supported in model loader.'.format(model_name)) raise ValueError('{0} is not currently supported in model loader.'.format(model_name))
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