odin-engine 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (63) hide show
  1. benchmarks/__init__.py +17 -17
  2. benchmarks/datasets.py +284 -284
  3. benchmarks/metrics.py +275 -275
  4. benchmarks/run_ablation.py +279 -279
  5. benchmarks/run_npll_benchmark.py +270 -270
  6. npll/__init__.py +10 -10
  7. npll/bootstrap.py +474 -474
  8. npll/core/__init__.py +33 -33
  9. npll/core/knowledge_graph.py +308 -308
  10. npll/core/logical_rules.py +496 -496
  11. npll/core/mln.py +474 -474
  12. npll/inference/__init__.py +40 -40
  13. npll/inference/e_step.py +419 -419
  14. npll/inference/elbo.py +434 -434
  15. npll/inference/m_step.py +576 -576
  16. npll/npll_model.py +631 -631
  17. npll/scoring/__init__.py +42 -42
  18. npll/scoring/embeddings.py +441 -441
  19. npll/scoring/probability.py +402 -402
  20. npll/scoring/scoring_module.py +369 -369
  21. npll/training/__init__.py +24 -24
  22. npll/training/evaluation.py +496 -496
  23. npll/training/npll_trainer.py +520 -520
  24. npll/utils/__init__.py +47 -47
  25. npll/utils/batch_utils.py +492 -492
  26. npll/utils/config.py +144 -144
  27. npll/utils/math_utils.py +338 -338
  28. odin/__init__.py +21 -20
  29. odin/engine.py +264 -264
  30. odin/schema.py +210 -0
  31. {odin_engine-0.1.0.dist-info → odin_engine-0.2.0.dist-info}/METADATA +503 -456
  32. odin_engine-0.2.0.dist-info/RECORD +63 -0
  33. {odin_engine-0.1.0.dist-info → odin_engine-0.2.0.dist-info}/licenses/LICENSE +21 -21
  34. retrieval/__init__.py +50 -50
  35. retrieval/adapters.py +140 -140
  36. retrieval/adapters_arango.py +1418 -1418
  37. retrieval/aggregators.py +707 -707
  38. retrieval/beam.py +127 -127
  39. retrieval/budget.py +60 -60
  40. retrieval/cache.py +159 -159
  41. retrieval/confidence.py +88 -88
  42. retrieval/eval.py +49 -49
  43. retrieval/linker.py +87 -87
  44. retrieval/metrics.py +105 -105
  45. retrieval/metrics_motifs.py +36 -36
  46. retrieval/orchestrator.py +571 -571
  47. retrieval/ppr/__init__.py +12 -12
  48. retrieval/ppr/anchors.py +41 -41
  49. retrieval/ppr/bippr.py +61 -61
  50. retrieval/ppr/engines.py +257 -257
  51. retrieval/ppr/global_pr.py +76 -76
  52. retrieval/ppr/indexes.py +78 -78
  53. retrieval/ppr.py +156 -156
  54. retrieval/ppr_cache.py +25 -25
  55. retrieval/scoring.py +294 -294
  56. retrieval/utils/pii_redaction.py +36 -36
  57. retrieval/writers/__init__.py +9 -9
  58. retrieval/writers/arango_writer.py +28 -28
  59. retrieval/writers/base.py +21 -21
  60. retrieval/writers/janus_writer.py +36 -36
  61. odin_engine-0.1.0.dist-info/RECORD +0 -62
  62. {odin_engine-0.1.0.dist-info → odin_engine-0.2.0.dist-info}/WHEEL +0 -0
  63. {odin_engine-0.1.0.dist-info → odin_engine-0.2.0.dist-info}/top_level.txt +0 -0
npll/core/__init__.py CHANGED
@@ -1,34 +1,34 @@
1
- """
2
- Core components for NPLL implementation
3
- """
4
-
5
- from .knowledge_graph import KnowledgeGraph, Entity, Relation, Triple, load_knowledge_graph_from_triples
6
- from .logical_rules import (
7
- Variable, Atom, LogicalRule, GroundRule, RuleType, RuleGenerator,
8
- parse_rule_from_string
9
- )
10
- from .mln import MarkovLogicNetwork, MLNState, create_mln_from_kg_and_rules, verify_mln_implementation
11
-
12
- __all__ = [
13
- # Knowledge Graph
14
- 'KnowledgeGraph',
15
- 'Entity',
16
- 'Relation',
17
- 'Triple',
18
- 'load_knowledge_graph_from_triples',
19
-
20
- # Logical Rules
21
- 'Variable',
22
- 'Atom',
23
- 'LogicalRule',
24
- 'GroundRule',
25
- 'RuleType',
26
- 'RuleGenerator',
27
- 'parse_rule_from_string',
28
-
29
- # Markov Logic Networks
30
- 'MarkovLogicNetwork',
31
- 'MLNState',
32
- 'create_mln_from_kg_and_rules',
33
- 'verify_mln_implementation'
1
+ """
2
+ Core components for NPLL implementation
3
+ """
4
+
5
+ from .knowledge_graph import KnowledgeGraph, Entity, Relation, Triple, load_knowledge_graph_from_triples
6
+ from .logical_rules import (
7
+ Variable, Atom, LogicalRule, GroundRule, RuleType, RuleGenerator,
8
+ parse_rule_from_string
9
+ )
10
+ from .mln import MarkovLogicNetwork, MLNState, create_mln_from_kg_and_rules, verify_mln_implementation
11
+
12
+ __all__ = [
13
+ # Knowledge Graph
14
+ 'KnowledgeGraph',
15
+ 'Entity',
16
+ 'Relation',
17
+ 'Triple',
18
+ 'load_knowledge_graph_from_triples',
19
+
20
+ # Logical Rules
21
+ 'Variable',
22
+ 'Atom',
23
+ 'LogicalRule',
24
+ 'GroundRule',
25
+ 'RuleType',
26
+ 'RuleGenerator',
27
+ 'parse_rule_from_string',
28
+
29
+ # Markov Logic Networks
30
+ 'MarkovLogicNetwork',
31
+ 'MLNState',
32
+ 'create_mln_from_kg_and_rules',
33
+ 'verify_mln_implementation'
34
34
  ]