├── .gitignore ├── LICENSE ├── README.md ├── datasets ├── __init__.py ├── append │ ├── __init__.py │ ├── append.bk │ ├── append.cl │ ├── append.lang │ ├── append.neg │ ├── append.pos │ ├── append.temp │ └── generate.py ├── delete │ ├── __init__.py │ ├── delete.bk │ ├── delete.cl │ ├── delete.lang │ ├── delete.neg │ ├── delete.pos │ └── generate.py ├── member │ ├── archive │ │ ├── generate.py │ │ ├── member.bk │ │ ├── member.cl │ │ ├── member.lang │ │ ├── member.neg │ │ ├── member.pos │ │ └── member.temp │ ├── generate.py │ ├── member.bk │ ├── member.cl │ ├── member.lang │ ├── member.neg │ ├── member.pos │ └── member.temp ├── plus │ ├── generate.py │ ├── plus.bk │ ├── plus.cl │ ├── plus.lang │ ├── plus.neg │ ├── plus.pos │ └── plus.temp ├── problem.py └── subtree │ ├── generate.py │ ├── subtree.bk │ ├── subtree.cl │ ├── subtree.lang │ ├── subtree.neg │ ├── subtree.pos │ └── subtree.temp ├── demo.ipynb ├── experiments ├── __init__.py ├── __main__.py ├── eval_utils.py ├── noise.py ├── normal.py ├── parameters.py ├── plot.py ├── softor.py └── step.py ├── imgs ├── .gitkeep ├── flow.png ├── noise │ └── loss │ │ └── .gitkeep ├── softor │ └── loss │ │ └── .gitkeep └── step │ └── loss │ └── .gitkeep ├── requirements.txt ├── results └── .gitkeep └── src ├── __init__.py ├── clause_generator.py ├── data_utils.py ├── exp.lark ├── exp_parser.py ├── fact_enumerator.py ├── ilp_problem.py ├── ilp_solver.py ├── infer.py ├── language.py ├── logic.py ├── logic_ops.py ├── optimizer.py ├── refinement.py ├── tensor_encoder.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | *.py[cod] 3 | .ipynb_checkpoints/ 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/append/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/append/append.bk: -------------------------------------------------------------------------------- 1 | append(*,*,*) 2 | -------------------------------------------------------------------------------- /datasets/append/append.cl: -------------------------------------------------------------------------------- 1 | append(X,Y,Z):-. 2 | -------------------------------------------------------------------------------- /datasets/append/append.lang: -------------------------------------------------------------------------------- 1 | append:3 2 | f:2 3 | a,b,c 4 | * 5 | -------------------------------------------------------------------------------- /datasets/append/append.neg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/append/append.neg -------------------------------------------------------------------------------- /datasets/append/append.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/append/append.pos -------------------------------------------------------------------------------- /datasets/append/append.temp: -------------------------------------------------------------------------------- 1 | 1,0 2 | 0,1 3 | -------------------------------------------------------------------------------- /datasets/append/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/append/generate.py -------------------------------------------------------------------------------- /datasets/delete/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/delete/delete.bk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/delete/delete.bk -------------------------------------------------------------------------------- /datasets/delete/delete.cl: -------------------------------------------------------------------------------- 1 | delete(X,Y,Z):-. 2 | -------------------------------------------------------------------------------- /datasets/delete/delete.lang: -------------------------------------------------------------------------------- 1 | delete:3 2 | f:2 3 | a,b,c 4 | 5 | -------------------------------------------------------------------------------- /datasets/delete/delete.neg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/delete/delete.neg -------------------------------------------------------------------------------- /datasets/delete/delete.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/delete/delete.pos -------------------------------------------------------------------------------- /datasets/delete/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/delete/generate.py -------------------------------------------------------------------------------- /datasets/member/archive/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/archive/generate.py -------------------------------------------------------------------------------- /datasets/member/archive/member.bk: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/member/archive/member.cl: -------------------------------------------------------------------------------- 1 | member(X,Y):-. 2 | -------------------------------------------------------------------------------- /datasets/member/archive/member.lang: -------------------------------------------------------------------------------- 1 | member:2 2 | f:2 3 | a,b,c,d,e 4 | 5 | -------------------------------------------------------------------------------- /datasets/member/archive/member.neg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/archive/member.neg -------------------------------------------------------------------------------- /datasets/member/archive/member.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/archive/member.pos -------------------------------------------------------------------------------- /datasets/member/archive/member.temp: -------------------------------------------------------------------------------- 1 | 1,0 2 | 0,0 3 | -------------------------------------------------------------------------------- /datasets/member/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/generate.py -------------------------------------------------------------------------------- /datasets/member/member.bk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/member.bk -------------------------------------------------------------------------------- /datasets/member/member.cl: -------------------------------------------------------------------------------- 1 | member(X,Y):-. 2 | -------------------------------------------------------------------------------- /datasets/member/member.lang: -------------------------------------------------------------------------------- 1 | member:2 2 | f:2 3 | a,b,c 4 | 5 | -------------------------------------------------------------------------------- /datasets/member/member.neg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/member.neg -------------------------------------------------------------------------------- /datasets/member/member.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/member/member.pos -------------------------------------------------------------------------------- /datasets/member/member.temp: -------------------------------------------------------------------------------- 1 | 1,0 2 | 0,0 3 | -------------------------------------------------------------------------------- /datasets/plus/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/plus/generate.py -------------------------------------------------------------------------------- /datasets/plus/plus.bk: -------------------------------------------------------------------------------- 1 | plus(0,0,0) 2 | -------------------------------------------------------------------------------- /datasets/plus/plus.cl: -------------------------------------------------------------------------------- 1 | plus(X,Y,Z):-. 2 | -------------------------------------------------------------------------------- /datasets/plus/plus.lang: -------------------------------------------------------------------------------- 1 | plus:3 2 | s:1 3 | 0 4 | 0 5 | -------------------------------------------------------------------------------- /datasets/plus/plus.neg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/plus/plus.neg -------------------------------------------------------------------------------- /datasets/plus/plus.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/plus/plus.pos -------------------------------------------------------------------------------- /datasets/plus/plus.temp: -------------------------------------------------------------------------------- 1 | 1,0 2 | 0,1 3 | -------------------------------------------------------------------------------- /datasets/problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/problem.py -------------------------------------------------------------------------------- /datasets/subtree/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/subtree/generate.py -------------------------------------------------------------------------------- /datasets/subtree/subtree.bk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/subtree/subtree.bk -------------------------------------------------------------------------------- /datasets/subtree/subtree.cl: -------------------------------------------------------------------------------- 1 | subtree(X,Y):-. 2 | -------------------------------------------------------------------------------- /datasets/subtree/subtree.lang: -------------------------------------------------------------------------------- 1 | subtree:2 2 | f:2 3 | a,b,c 4 | 5 | -------------------------------------------------------------------------------- /datasets/subtree/subtree.neg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/subtree/subtree.neg -------------------------------------------------------------------------------- /datasets/subtree/subtree.pos: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/subtree/subtree.pos -------------------------------------------------------------------------------- /datasets/subtree/subtree.temp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/datasets/subtree/subtree.temp -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/demo.ipynb -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/__main__.py -------------------------------------------------------------------------------- /experiments/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/eval_utils.py -------------------------------------------------------------------------------- /experiments/noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/noise.py -------------------------------------------------------------------------------- /experiments/normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/normal.py -------------------------------------------------------------------------------- /experiments/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/parameters.py -------------------------------------------------------------------------------- /experiments/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/plot.py -------------------------------------------------------------------------------- /experiments/softor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/softor.py -------------------------------------------------------------------------------- /experiments/step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/experiments/step.py -------------------------------------------------------------------------------- /imgs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imgs/flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/imgs/flow.png -------------------------------------------------------------------------------- /imgs/noise/loss/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imgs/softor/loss/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /imgs/step/loss/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | lark-parser 4 | seaborn 5 | tqdm 6 | sklearn 7 | -------------------------------------------------------------------------------- /results/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/clause_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/clause_generator.py -------------------------------------------------------------------------------- /src/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/data_utils.py -------------------------------------------------------------------------------- /src/exp.lark: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/exp.lark -------------------------------------------------------------------------------- /src/exp_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/exp_parser.py -------------------------------------------------------------------------------- /src/fact_enumerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/fact_enumerator.py -------------------------------------------------------------------------------- /src/ilp_problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/ilp_problem.py -------------------------------------------------------------------------------- /src/ilp_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/ilp_solver.py -------------------------------------------------------------------------------- /src/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/infer.py -------------------------------------------------------------------------------- /src/language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/language.py -------------------------------------------------------------------------------- /src/logic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/logic.py -------------------------------------------------------------------------------- /src/logic_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/logic_ops.py -------------------------------------------------------------------------------- /src/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/optimizer.py -------------------------------------------------------------------------------- /src/refinement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/refinement.py -------------------------------------------------------------------------------- /src/tensor_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/tensor_encoder.py -------------------------------------------------------------------------------- /src/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hkrsnd/dilp-st/HEAD/src/visualize.py --------------------------------------------------------------------------------