comparison ludwig_train.py @ 0:dceb8493730d draft default tip

planemo upload for repository https://github.com/goeckslab/Galaxy-Ludwig.git commit bdea9430787658783a51cc6c2ae951a01e455bb4
author goeckslab
date Tue, 07 Jan 2025 22:44:54 +0000
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:dceb8493730d
1 import logging
2 import pickle
3 import sys
4
5 from ludwig.train import cli
6
7 from ludwig_experiment import (
8 convert_parquet_to_csv,
9 generate_html_report,
10 make_visualizations
11 )
12
13 from model_unpickler import SafeUnpickler
14
15
16 logging.basicConfig(level=logging.DEBUG)
17
18 setattr(pickle, 'Unpickler', SafeUnpickler)
19
20 cli(sys.argv[1:])
21
22 ludwig_output_directory_name = "experiment_run"
23
24 make_visualizations(ludwig_output_directory_name)
25 convert_parquet_to_csv(
26 ludwig_output_directory_name
27 )
28 title = "Ludwig Train"
29 generate_html_report(title, ludwig_output_directory_name)