Commit f57d5587 authored by wegamekinglc's avatar wegamekinglc

added svr model to api

parent 1fe4765e
...@@ -43,6 +43,7 @@ from alphamind.model import RandomForestClassifier ...@@ -43,6 +43,7 @@ from alphamind.model import RandomForestClassifier
from alphamind.model import XGBRegressor from alphamind.model import XGBRegressor
from alphamind.model import XGBClassifier from alphamind.model import XGBClassifier
from alphamind.model import XGBTrainer from alphamind.model import XGBTrainer
from alphamind.model import NvSVRModel
from alphamind.model import load_model from alphamind.model import load_model
from alphamind.model.data_preparing import fetch_data_package from alphamind.model.data_preparing import fetch_data_package
from alphamind.model.data_preparing import fetch_train_phase from alphamind.model.data_preparing import fetch_train_phase
...@@ -104,6 +105,7 @@ __all__ = [ ...@@ -104,6 +105,7 @@ __all__ = [
'XGBRegressor', 'XGBRegressor',
'XGBClassifier', 'XGBClassifier',
'XGBTrainer', 'XGBTrainer',
'NvSVRModel',
'load_model', 'load_model',
'NaiveExecutor', 'NaiveExecutor',
'ThresholdExecutor', 'ThresholdExecutor',
......
...@@ -16,6 +16,8 @@ from alphamind.model.treemodel import XGBRegressor ...@@ -16,6 +16,8 @@ from alphamind.model.treemodel import XGBRegressor
from alphamind.model.treemodel import XGBClassifier from alphamind.model.treemodel import XGBClassifier
from alphamind.model.treemodel import XGBTrainer from alphamind.model.treemodel import XGBTrainer
from alphamind.model.svm import NvSVRModel
from alphamind.model.loader import load_model from alphamind.model.loader import load_model
...@@ -28,4 +30,5 @@ __all__ = ['LinearRegression', ...@@ -28,4 +30,5 @@ __all__ = ['LinearRegression',
'XGBRegressor', 'XGBRegressor',
'XGBClassifier', 'XGBClassifier',
'XGBTrainer', 'XGBTrainer',
'NvSVRModel',
'load_model'] 'load_model']
\ No newline at end of file
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