docker run -v $(pwd):/data air_quality_prediction:1.0 --train_data_folder /data/data/ --pred_data_path /data/data.csv --variable u10 v10 t2m sp tp AQI PM10 NO2 CO O3 SO2 --air_quality PM2.5 --geo_coord x y --output_best_arg /data/best_arg.json