Skip to content

Instantly share code, notes, and snippets.

@hamletbatista

hamletbatista/ludwig_get_predictions.py

Created June 3, 2019 02:06
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save hamletbatista/81d9f9359c4e3a31a687abebd1b26d69 to your computer and use it in GitHub Desktop.
Save hamletbatista/81d9f9359c4e3a31a687abebd1b26d69 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ludwig_get_predictions.py
from ludwig.api import LudwigModel
model = LudwigModel.load("results/experiment_run_0/model")
predictions = model.predict(test_df)
test_df.join(predictions)[["text", "category_predictions"]]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment