mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-06 04:20:57 +08:00
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT License.
|
|
try:
|
|
from .catboost_model import CatBoostModel
|
|
except ModuleNotFoundError:
|
|
CatBoostModel = None
|
|
print("ModuleNotFoundError. CatBoostModel are skipped. (optional: maybe installing CatBoostModel can fix it.)")
|
|
try:
|
|
from .double_ensemble import DEnsembleModel
|
|
from .gbdt import LGBModel
|
|
except ModuleNotFoundError:
|
|
DEnsembleModel, LGBModel = None, None
|
|
print(
|
|
"ModuleNotFoundError. DEnsembleModel and LGBModel are skipped. (optional: maybe installing lightgbm can fix it.)"
|
|
)
|
|
try:
|
|
from .xgboost import XGBModel
|
|
except ModuleNotFoundError:
|
|
XGBModel = None
|
|
print("ModuleNotFoundError. XGBModel is skipped(optional: maybe installing xgboost can fix it).")
|
|
try:
|
|
from .linear import LinearModel
|
|
except ModuleNotFoundError:
|
|
LinearModel = None
|
|
print("ModuleNotFoundError. LinearModel is skipped(optional: maybe installing scipy and sklearn can fix it).")
|
|
# import pytorch models
|
|
try:
|
|
from .pytorch_alstm import ALSTM
|
|
from .pytorch_gats import GATs
|
|
from .pytorch_gru import GRU
|
|
from .pytorch_lstm import LSTM
|
|
from .pytorch_nn import DNNModelPytorch
|
|
from .pytorch_tabnet import TabnetModel
|
|
from .pytorch_sfm import SFM_Model
|
|
from .pytorch_tcn import TCN
|
|
from .pytorch_add import ADD
|
|
|
|
pytorch_classes = (ALSTM, GATs, GRU, LSTM, DNNModelPytorch, TabnetModel, SFM_Model, TCN, ADD)
|
|
except ModuleNotFoundError:
|
|
pytorch_classes = ()
|
|
print("ModuleNotFoundError. PyTorch models are skipped (optional: maybe installing pytorch can fix it).")
|
|
|
|
all_model_classes = (CatBoostModel, DEnsembleModel, LGBModel, XGBModel, LinearModel) + pytorch_classes
|