mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-09 09:07:49 +00:00
Mark Aron Szulyovszky
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from run_pipeline import run_training, setup_config
|
|
from config.types import RawConfig
|
|
from data_loader.collections import data_collections
|
|
from reporting.reporting import report_results
|
|
from config.presets import get_default_config
|
|
|
|
|
|
def run_multi_asset_pipeline(
|
|
project_name: str, with_wandb: bool, sweep: bool, raw_config: RawConfig
|
|
):
|
|
collection = data_collections["fivemin_crypto"]
|
|
for asset in collection:
|
|
print(f"# Predicting asset: {asset[1]}\n")
|
|
raw_config.target_asset = asset[1]
|
|
wandb, config = setup_config(project_name, with_wandb, sweep, raw_config)
|
|
pipeline_outcome = run_training(config)
|
|
report_results(
|
|
pipeline_outcome.directional_training.training.stats,
|
|
pipeline_outcome.get_output_stats(),
|
|
pipeline_outcome.get_output_weights(),
|
|
config,
|
|
wandb,
|
|
sweep,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_multi_asset_pipeline(
|
|
project_name="price-prediction",
|
|
with_wandb=False,
|
|
sweep=False,
|
|
raw_config=get_default_config(),
|
|
)
|