diff --git a/qlib/contrib/model/pytorch_alstm.py b/qlib/contrib/model/pytorch_alstm.py index b4fbbd504..a149272da 100644 --- a/qlib/contrib/model/pytorch_alstm.py +++ b/qlib/contrib/model/pytorch_alstm.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_alstm_ts.py b/qlib/contrib/model/pytorch_alstm_ts.py index 87ebb489b..c38727b9e 100644 --- a/qlib/contrib/model/pytorch_alstm_ts.py +++ b/qlib/contrib/model/pytorch_alstm_ts.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_gats.py b/qlib/contrib/model/pytorch_gats.py index 9a077b736..53afb5404 100644 --- a/qlib/contrib/model/pytorch_gats.py +++ b/qlib/contrib/model/pytorch_gats.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn import torch.optim as optim diff --git a/qlib/contrib/model/pytorch_gats_ts.py b/qlib/contrib/model/pytorch_gats_ts.py index 35fd95ab5..f02bf1e47 100644 --- a/qlib/contrib/model/pytorch_gats_ts.py +++ b/qlib/contrib/model/pytorch_gats_ts.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn import torch.optim as optim diff --git a/qlib/contrib/model/pytorch_gru.py b/qlib/contrib/model/pytorch_gru.py index 8b59e0ee4..5eba33595 100755 --- a/qlib/contrib/model/pytorch_gru.py +++ b/qlib/contrib/model/pytorch_gru.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_gru_ts.py b/qlib/contrib/model/pytorch_gru_ts.py index 514b44c8e..2839b35e4 100755 --- a/qlib/contrib/model/pytorch_gru_ts.py +++ b/qlib/contrib/model/pytorch_gru_ts.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_lstm.py b/qlib/contrib/model/pytorch_lstm.py index b4920921e..636ef6e3a 100755 --- a/qlib/contrib/model/pytorch_lstm.py +++ b/qlib/contrib/model/pytorch_lstm.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_lstm_ts.py b/qlib/contrib/model/pytorch_lstm_ts.py index 9d513dff8..c978e84c7 100755 --- a/qlib/contrib/model/pytorch_lstm_ts.py +++ b/qlib/contrib/model/pytorch_lstm_ts.py @@ -9,14 +9,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py index 326af92db..caf34b330 100644 --- a/qlib/contrib/model/pytorch_nn.py +++ b/qlib/contrib/model/pytorch_nn.py @@ -19,7 +19,7 @@ from ...model.base import Model from ...data.dataset import DatasetH from ...data.dataset.handler import DataHandlerLP from ...utils import unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger from ...workflow import R diff --git a/qlib/contrib/model/pytorch_sfm.py b/qlib/contrib/model/pytorch_sfm.py index 0bb2b4704..db3e8bb12 100644 --- a/qlib/contrib/model/pytorch_sfm.py +++ b/qlib/contrib/model/pytorch_sfm.py @@ -8,14 +8,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn diff --git a/qlib/contrib/model/pytorch_tabnet.py b/qlib/contrib/model/pytorch_tabnet.py index a33ba0f78..450e6f5d1 100644 --- a/qlib/contrib/model/pytorch_tabnet.py +++ b/qlib/contrib/model/pytorch_tabnet.py @@ -7,14 +7,13 @@ import os import numpy as np import pandas as pd import copy -from sklearn.metrics import roc_auc_score, mean_squared_error from ...utils import ( unpack_archive_with_buffer, save_multiple_parts_file, get_or_create_path, drop_nan_by_y_index, ) -from ...log import get_module_logger, TimeInspector +from ...log import get_module_logger import torch import torch.nn as nn