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, )