Views
No views yet
networkx.Graphpip install networkx numpy torch1import networkx as nx
2from ai_meetup_model.model import ChallengeModel
3
4model = ChallengeModel("ai_meetup_model", device="cpu")
5
6G = nx.Graph()
7G.add_nodes_from(list("ABCDEFGHIJ"))
8G.add_edge("A","C", weight=0.8)
9G.add_edge("C","B", weight=0.4)
10
11print(model.custom_predict(G))ai_meetup_model/
├── model.pt # váhy modelu
├── config.json # konfigurácia architektúry
├── model.py # definícia modelu + wrapper
└── __init__.py