Mercurial > repos > goeckslab > ludwig_evaluate

view ludwig_train.py @ 2:e55790a3e8a7 draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/Galaxy-Ludwig.git commit 008bc5a8ee2c80f70fc224a23b5d2cd0c8ef1810
author goeckslab
date Fri, 14 Mar 2025 16:48:25 +0000
parents 70a4d910f09a
children
line wrap: on
line source

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)