Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Sign in
Toggle navigation
A
alpha-mind
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Dr.李
alpha-mind
Commits
bd2a2d3d
Commit
bd2a2d3d
authored
Dec 07, 2017
by
Dr.李
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update for lasso model
parent
5addc3aa
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
1 deletion
+4
-1
linearmodel.py
alphamind/model/linearmodel.py
+1
-1
loader.py
alphamind/model/loader.py
+3
-0
No files found.
alphamind/model/linearmodel.py
View file @
bd2a2d3d
...
...
@@ -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'
]
...
...
alphamind/model/loader.py
View file @
bd2a2d3d
...
...
@@ -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
))
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment