comparison ludwig_experiment.py @ 4:e3c63770083d draft default tip

planemo upload for repository https://github.com/goeckslab/Galaxy-Ludwig.git commit 4dc221b2fa9717552787f0985ad3fc3df4460158
author goeckslab
date Sat, 21 Jun 2025 15:04:30 +0000
parents d5d69c29d537
children
comparison
equal deleted inserted replaced
3:56c4f4c0c6df 4:e3c63770083d
2 import logging 2 import logging
3 import os 3 import os
4 import pickle 4 import pickle
5 import sys 5 import sys
6 6
7 import pandas as pd
7 from ludwig.experiment import cli 8 from ludwig.experiment import cli
8 from ludwig.globals import ( 9 from ludwig.globals import (
9 DESCRIPTION_FILE_NAME, 10 DESCRIPTION_FILE_NAME,
10 PREDICTIONS_PARQUET_FILE_NAME, 11 PREDICTIONS_PARQUET_FILE_NAME,
11 TEST_STATISTICS_FILE_NAME, 12 TEST_STATISTICS_FILE_NAME,
12 TRAIN_SET_METADATA_FILE_NAME 13 TRAIN_SET_METADATA_FILE_NAME
13 ) 14 )
14 from ludwig.utils.data_utils import get_split_path 15 from ludwig.utils.data_utils import get_split_path
15 from ludwig.visualize import get_visualizations_registry 16 from ludwig.visualize import get_visualizations_registry
16
17 from model_unpickler import SafeUnpickler 17 from model_unpickler import SafeUnpickler
18
19 import pandas as pd
20
21 from utils import ( 18 from utils import (
22 encode_image_to_base64, 19 encode_image_to_base64,
23 get_html_closing, 20 get_html_closing,
24 get_html_template 21 get_html_template
25 ) 22 )
33 30
34 # visualization 31 # visualization
35 output_directory = None 32 output_directory = None
36 for ix, arg in enumerate(sys.argv): 33 for ix, arg in enumerate(sys.argv):
37 if arg == "--output_directory": 34 if arg == "--output_directory":
38 output_directory = sys.argv[ix+1] 35 output_directory = sys.argv[ix + 1]
39 break 36 break
40 37
41 viz_output_directory = os.path.join(output_directory, "visualizations") 38 viz_output_directory = os.path.join(output_directory, "visualizations")
42 39
43 40