dialog2graph.pipelines.core.d2g_generator#

Dialog2Graph base generator to reuse evaluate method in different algorithms

Classes#

DGBaseGenerator

Dialog2graph base generator to reuse evaluate method in different algorithms

Module Contents#

class dialog2graph.pipelines.core.d2g_generator.DGBaseGenerator(/, **data: Any)[source]#

Bases: dialog2graph.pipelines.core.algorithms.GraphGenerator

Dialog2graph base generator to reuse evaluate method in different algorithms .. attribute:: sim_model

str, model name in storage

Parameters:

data (Any)

sim_model: str = None#
evaluate(graph, true_graph, eval_stage: str) → dialog2graph.metrics.DGReportType[source]#

Call metrics and return report

Parameters:

  • graph – generated graph

  • true_graph – expected graph

  • eval_stage (str) – string defining eval stage, like step2 or end

Returns:

result}

Return type:

dictionary with report like {“metric_name”