liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-09 14:00:55 +08:00
Code Activity

Update run_all_model and format

Browse Source
This commit is contained in:
Jactus
2020-11-26 13:49:51 +08:00
parent 5102566aad
commit 2a17062421
3 changed files with 2 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/benchmarks/TFT/README.md
View File

@@ -8,7 +8,7 @@
Users can follow the ``workflow_by_code_tft.py`` to run the benchmark.
### Notes
1. Please be **aware** that this script can only support `Python 3.5 - 3.8`, and `Cuda 10.0 or 10.1`.
2. Please remember to install `cudatoolkit==10.1` and `cudnn==7.6` on your machine.
1. Please be **aware** that this script can only support `Python 3.5 - 3.8`.
2. If the CUDA version on your machine is not 10.0, please remember to run the following commands `conda install anaconda cudatoolkit=10.0` and `conda install cudnn` on your machine.
3. The model must run in GPU, or an error will be raised.
4. New datasets should be registered in ``data_formatters``, for detail please visit the source.

15
examples/run_all_model.py
View File

@@ -162,20 +162,6 @@ class ExtendedEnvBuilder(venv.EnvBuilder):
self.install_script(context, "pip", url)
# function to check cuda version on the machine, this case is for the model TFT
def check_cuda(folders):
path = "/usr/local/cuda/version.txt" # TODO: FIX ME, this will not work on other os systems.
exclude_tft = True
if os.path.exists(path):
with open(path, "r") as f:
if "10.1" in str(f.read()) or "10.0" in str(f.read()):
exclude_tft = False
if exclude_tft and "TFT" in folders:
sys.stderr.write("Compatible CUDA version not found! Removing TFT from the workflow...\n")
del folders["TFT"]
return folders
# function to calculate the mean and std of a list in the results dictionary
def cal_mean_std(results) -> dict:
mean_std = dict()
@@ -205,7 +191,6 @@ def get_all_folders(models, exclude) -> dict:
if add:
path = Path("benchmarks") / f.name
folders[f.name] = str(path.resolve())
folders = check_cuda(folders)
return folders

1
qlib/contrib/model/pytorch_lstm.py
View File

@@ -38,7 +38,6 @@ class LSTM(Model):
the GPU ID(s) used for training
"""
def __init__(
self,
d_feat=6,