intentgraph-lm / src /graph_builder.py
hmnshudhmn24's picture
Upload 23 files
602b5b1 verified
import networkx as nx
class GraphBuilder:
def build(self, intents):
graph = nx.DiGraph()
for i, intent in enumerate(intents):
graph.add_node(intent, type='intent')
if i > 0:
graph.add_edge(intents[i-1], intent, weight=1.0)
return graph