diff 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
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/ludwig_train.py	Tue Jan 07 22:44:09 2025 +0000
@@ -0,0 +1,29 @@
+import logging
+import pickle
+import sys
+
+from ludwig.train import cli
+
+from ludwig_experiment import (
+    convert_parquet_to_csv,
+    generate_html_report,
+    make_visualizations
+)
+
+from model_unpickler import SafeUnpickler
+
+
+logging.basicConfig(level=logging.DEBUG)
+
+setattr(pickle, 'Unpickler', SafeUnpickler)
+
+cli(sys.argv[1:])
+
+ludwig_output_directory_name = "experiment_run"
+
+make_visualizations(ludwig_output_directory_name)
+convert_parquet_to_csv(
+    ludwig_output_directory_name
+)
+title = "Ludwig Train"
+generate_html_report(title, ludwig_output_directory_name)