Mercurial > repos > goeckslab > ludwig_train
comparison ludwig_train.py @ 0:f0be10937f5c 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:09 +0000 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:f0be10937f5c |
|---|---|
| 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) |