From 01385c2cf8f6d21e21deba497628999e336db670 Mon Sep 17 00:00:00 2001 From: Alex Sherstinsky Date: Wed, 28 Feb 2024 08:08:45 -0800 Subject: [PATCH] [BUGFIX] Correct typos and wrong description in zero-shot LLM examples. (#348) --- docs/examples/llms/llm_zero_shot_batch_inference.md | 4 ++-- docs/examples/llms/llm_zero_shot_text_generation.md | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docs/examples/llms/llm_zero_shot_batch_inference.md b/docs/examples/llms/llm_zero_shot_batch_inference.md index cf7a6d00..c6318114 100644 --- a/docs/examples/llms/llm_zero_shot_batch_inference.md +++ b/docs/examples/llms/llm_zero_shot_batch_inference.md @@ -94,7 +94,7 @@ output_features: """ ) -# Define Ludwig model object that drive model training +# Define Ludwig model object that drives model training model = LudwigModel(config=config, logging_level=logging.INFO) # initiate model training @@ -111,4 +111,4 @@ training_set, val_set, test_set, _ = preprocessed_data # batch prediction preds, _ = model.predict(test_set, skip_save_predictions=False) print(preds) -``` \ No newline at end of file +``` diff --git a/docs/examples/llms/llm_zero_shot_text_generation.md b/docs/examples/llms/llm_zero_shot_text_generation.md index 14002575..524f8903 100644 --- a/docs/examples/llms/llm_zero_shot_text_generation.md +++ b/docs/examples/llms/llm_zero_shot_text_generation.md @@ -70,10 +70,10 @@ config = yaml.safe_load( """ ) -# Define Ludwig model object that drive model training +# Define Ludwig model object that drives model training model = LudwigModel(config=config, logging_level=logging.INFO) -# Loads the model and performs no training. +# initiate model training ( train_stats, # dictionary containing training statistics preprocessed_data, # tuple Ludwig Dataset objects of pre-processed training data @@ -87,4 +87,4 @@ training_set, val_set, test_set, _ = preprocessed_data # batch prediction preds, _ = model.predict(test_set, skip_save_predictions=False) print(preds) -``` \ No newline at end of file +```