fixed bug

alphamind/api.py

@@ -22,6 +22,9 @@ from alphamind.data.standardize import standardize
 from alphamind.data.neutralize import neutralize
 from alphamind.data.engines.sqlengine import factor_tables
 
+from alphamind.model.linearmodel import LinearRegression
+from alphamind.model.linearmodel import ConstLinearModel
+from alphamind.model.loader import load_model
 from alphamind.model.data_preparing import fetch_data_package
 
 from alphamind.utilities import alpha_logger
@@ -44,5 +47,8 @@ __all__ = [
     'neutralize',
     'factor_tables',
     'fetch_data_package',
+    'LinearRegression',
+    'ConstLinearModel',
+    'load_model',
     'alpha_logger'
 ]
\ No newline at end of file

alphamind/model/linearmodel.py

@@ -69,7 +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_)
+        model_desc['coef'] = self.impl.coef_[0].tolist()
         return model_desc
 
     def score(self) -> float: