quant_test / scripts /build_gp_dataset.py
lucky-loster's picture
Upload folder using huggingface_hub
590a501 verified
Raw
History Blame Contribute Delete
663 Bytes
#!/usr/bin/env python3
"""Build qlib DatasetH / DataHandler artifacts from GP mining outputs."""
import argparse
import sys
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
sys.path.insert(0, str(ROOT))
from data_pipeline.gp_dataset_builder import export_gp_artifacts
def main():
parser = argparse.ArgumentParser(description="Export GP features to qlib handler/dataset pickles")
parser.add_argument("--run-id", type=str, default=None, help="GP mining run id (default: RUN_ID env)")
args = parser.parse_args()
export_gp_artifacts(run_id=args.run_id)
if __name__ == "__main__":
main()