def plot_core_results(output_directory: Text, number_of_examples: List[int]) -> None:
"""Plot core model comparison graph.
Args:
output_directory: path to the output directory
number_of_examples: number of examples per run
"""
import rasa.utils.plotting as plotting_utils
graph_path = os.path.join(output_directory, "core_model_comparison_graph.pdf")
plotting_utils.plot_curve(
output_directory,
number_of_examples,
x_label_text="Number of stories present during training",
y_label_text="Number of correct test stories",
graph_path=graph_path,
)
def plot_nlu_results(output_directory: Text, number_of_examples: List[int]) -> None:
"""Plot NLU model comparison graph.
Args:
output_directory: path to the output directory
number_of_examples: number of examples per run
"""
import rasa.utils.plotting as plotting_utils
graph_path = os.path.join(output_directory, "nlu_model_comparison_graph.pdf")
plotting_utils.plot_curve(
output_directory,
number_of_examples,
x_label_text="Number of intent examples present during training",
y_label_text="Label-weighted average F1 score on test set",
graph_path=graph_path,
)
