neuralogic.nn.base

class AbstractNeuraLogic(dataset_builder: DatasetBuilder, template: Template, settings: SettingsProxy)[source]

Bases: object

ground(dataset: BaseDataset, *, batch_size: int = 1, learnable_facts: bool = False) → GroundedDataset[source]
build_dataset(dataset: BaseDataset | GroundedDataset, *, batch_size: int = 1, learnable_facts: bool = False, progress: bool = False) → BuiltDataset[source]
set_hooks(hooks)[source]
run_hook(hook: str, value)[source]
sync_template(state_dict: Dict | None = None, weights=None)[source]
parameters() → Dict[source]
state_dict() → Dict[source]
load_state_dict(state_dict: Dict)[source]
draw(filename: str | None = None, show=True, img_type='png', value_detail: int = 0, graphviz_path: str | None = None, *args, **kwargs)[source]
class AbstractEvaluator(template: Template, settings: Settings)[source]

Bases: object

build_dataset(dataset: BaseDataset | BuiltDataset, *, batch_size: int = 1, learnable_facts: bool = False, progress: bool = False)[source]
property model: AbstractNeuraLogic
train(dataset: BaseDataset | BuiltDataset | None = None, *, generator: bool = True)[source]
test(dataset: BaseDataset | BuiltDataset | None = None, *, generator: bool = True)[source]
parameters() → Dict[source]
state_dict() → Dict[source]
load_state_dict(state_dict: Dict)[source]
reset_parameters()[source]
draw(filename: str | None = None, show=True, img_type='png', value_detail: int = 0, graphviz_path: str | None = None, *args, **kwargs)[source]