Skip to content

Commit

Permalink
Add examples
Browse files Browse the repository at this point in the history
  • Loading branch information
LukasZahradnik committed Aug 11, 2021
1 parent 7fbbb35 commit ed84665
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions examples/DistinguishingNonRegularGraphs.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -340,14 +340,14 @@
},
{
"cell_type": "code",
"execution_count": 26,
"execution_count": 27,
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Graph a is predicted to be class: 1 | 0.9564615488052368\n",
"Graph b is predicted to be class: 0 | 0.04621277377009392\n"
"Graph Bicyclopentyl is predicted to be class: 1 | 0.9578434228897095\n",
"Graph Decalin is predicted to be class: 0 | 0.04274271801114082\n"
]
}
],
Expand All @@ -357,7 +357,7 @@
"for _ in neuralogic_evaluator.train(train_dataset):\n",
" pass\n",
"\n",
"graphs = [\"a\", \"b\"]\n",
"graphs = [\"Bicyclopentyl\", \"Decalin\"]\n",
"\n",
"for graph_id, (label, predicted) in enumerate(neuralogic_evaluator.test(train_dataset)):\n",
" print(f\"Graph {graphs[graph_id]} is predicted to be class: {int(round(predicted))} | {predicted}\")\n"
Expand Down

0 comments on commit ed84665

Please sign in to comment.