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

update for lasso model

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