Skip to content

Instantly share code, notes, and snippets.

@hamletbatista

hamletbatista/ludwig_get_questions_predictions.py

Created June 3, 2019 02:57
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save hamletbatista/cff01cda629019630a675c38ab1c3c79 to your computer and use it in GitHub Desktop.
Save hamletbatista/cff01cda629019630a675c38ab1c3c79 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ludwig_get_questions_predictions.py
model = LudwigModel.load("results/experiment_run_3/model")
test_df = pd.read_csv("Question Report_Page 1_Table.csv")
#we rename Query to Questions to match what the model expects
predictions = model.predict(test_df.rename(columns={'Query': 'Questions'} ))
test_df.join(predictions)[["Query", "Category0_predictions"]]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment