|
|
|
@@ -19,19 +19,20 @@ |
|
|
|
"source": [ |
|
|
|
"# Import necessary libraries and modules\n", |
|
|
|
"import os.path as osp\n", |
|
|
|
"\n", |
|
|
|
"import matplotlib.pyplot as plt\n", |
|
|
|
"import torch\n", |
|
|
|
"import torch.nn as nn\n", |
|
|
|
"import matplotlib.pyplot as plt\n", |
|
|
|
"\n", |
|
|
|
"from torch.optim import RMSprop, lr_scheduler\n", |
|
|
|
"\n", |
|
|
|
"from datasets import get_dataset\n", |
|
|
|
"from models.nn import LeNet5\n", |
|
|
|
"from abl.bridge import SimpleBridge\n", |
|
|
|
"from abl.data.evaluation import ReasoningMetric, SymbolAccuracy\n", |
|
|
|
"from abl.learning import ABLModel, BasicNN\n", |
|
|
|
"from abl.reasoning import KBBase, Reasoner\n", |
|
|
|
"from abl.data.evaluation import ReasoningMetric, SymbolAccuracy\n", |
|
|
|
"from abl.utils import ABLLogger, print_log\n", |
|
|
|
"from abl.bridge import SimpleBridge" |
|
|
|
"\n", |
|
|
|
"from datasets import get_dataset\n", |
|
|
|
"from models.nn import LeNet5" |
|
|
|
] |
|
|
|
}, |
|
|
|
{ |
|
|
|
|