├── .gitignore ├── LICENSE ├── Pipfile ├── README.md ├── configs ├── datasets │ ├── feynman │ │ ├── easy_set.yaml │ │ ├── hard_set.yaml │ │ └── medium_set.yaml │ └── random │ │ └── feynman_lm.yaml └── experiments │ ├── symbolic_transformer-pretraining.yaml │ ├── symbolic_transformer-srsd-feynman_easy.yaml │ ├── symbolic_transformer-srsd-feynman_hard.yaml │ └── symbolic_transformer-srsd-feynman_medium.yaml ├── dataset_analyzer.py ├── dataset_comparator.py ├── dataset_converter.py ├── dataset_generator.py ├── datasets ├── __init__.py ├── base.py ├── feynman.py ├── registry.py └── sampling.py ├── dummy_column_mixer.py ├── eq ├── __init__.py ├── conversion.py ├── dataset.py ├── eval.py ├── op.py ├── pred.py ├── registry.py ├── tree.py └── vocabulary.py ├── eq_analyzer.py ├── eq_comparator.py ├── external ├── README.md ├── ai_feynman2 │ ├── Dockerfile │ ├── README.md │ ├── ai_feynman2.def │ ├── ai_feynman_runner.py │ └── equation_converter.py ├── dso │ ├── Dockerfile │ ├── Pipfile │ ├── README.md │ ├── configs │ │ ├── dsr │ │ │ ├── config_w_const1.json │ │ │ ├── config_w_const2.json │ │ │ ├── config_w_const3.json │ │ │ ├── config_w_const4.json │ │ │ ├── config_w_const5.json │ │ │ ├── config_wo_const1.json │ │ │ ├── config_wo_const2.json │ │ │ ├── config_wo_const3.json │ │ │ ├── config_wo_const4.json │ │ │ └── config_wo_const5.json │ │ └── udsr │ │ │ ├── config_w_poly1.json │ │ │ ├── config_w_poly10.json │ │ │ ├── config_w_poly2.json │ │ │ ├── config_w_poly3.json │ │ │ ├── config_w_poly4.json │ │ │ ├── config_w_poly5.json │ │ │ ├── config_w_poly6.json │ │ │ ├── config_w_poly7.json │ │ │ ├── config_w_poly8.json │ │ │ └── config_w_poly9.json │ ├── dso.def │ └── equation_converter.py ├── e2e │ ├── Pipfile │ ├── README.md │ └── runner.py ├── ellyn │ ├── Dockerfile │ ├── README.md │ ├── configs │ │ ├── afp_default.yaml │ │ ├── afp_optuna.yaml │ │ ├── fe_afp_default.yaml │ │ └── fe_afp_optuna.yaml │ ├── ellyn.def │ └── ellyn_runner.py ├── gplearn │ ├── Pipfile │ ├── README.md │ ├── configs │ │ ├── default_w_const.yaml │ │ ├── default_wo_const.yaml │ │ └── optuna.yaml │ └── gp_runner.py └── pysr │ ├── README.md │ ├── config.yaml │ └── runner.py ├── model_selector.py ├── r2_evaluator.py ├── random_dataset_generator.ipynb └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | .ipynb_checkpoints/ 3 | __pycache__/ 4 | .editorconfig -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/LICENSE -------------------------------------------------------------------------------- /Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/Pipfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/README.md -------------------------------------------------------------------------------- /configs/datasets/feynman/easy_set.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/datasets/feynman/easy_set.yaml -------------------------------------------------------------------------------- /configs/datasets/feynman/hard_set.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/datasets/feynman/hard_set.yaml -------------------------------------------------------------------------------- /configs/datasets/feynman/medium_set.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/datasets/feynman/medium_set.yaml -------------------------------------------------------------------------------- /configs/datasets/random/feynman_lm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/datasets/random/feynman_lm.yaml -------------------------------------------------------------------------------- /configs/experiments/symbolic_transformer-pretraining.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/experiments/symbolic_transformer-pretraining.yaml -------------------------------------------------------------------------------- /configs/experiments/symbolic_transformer-srsd-feynman_easy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/experiments/symbolic_transformer-srsd-feynman_easy.yaml -------------------------------------------------------------------------------- /configs/experiments/symbolic_transformer-srsd-feynman_hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/experiments/symbolic_transformer-srsd-feynman_hard.yaml -------------------------------------------------------------------------------- /configs/experiments/symbolic_transformer-srsd-feynman_medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/configs/experiments/symbolic_transformer-srsd-feynman_medium.yaml -------------------------------------------------------------------------------- /dataset_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/dataset_analyzer.py -------------------------------------------------------------------------------- /dataset_comparator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/dataset_comparator.py -------------------------------------------------------------------------------- /dataset_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/dataset_converter.py -------------------------------------------------------------------------------- /dataset_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/dataset_generator.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from datasets import sampling, feynman 2 | -------------------------------------------------------------------------------- /datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/datasets/base.py -------------------------------------------------------------------------------- /datasets/feynman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/datasets/feynman.py -------------------------------------------------------------------------------- /datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/datasets/registry.py -------------------------------------------------------------------------------- /datasets/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/datasets/sampling.py -------------------------------------------------------------------------------- /dummy_column_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/dummy_column_mixer.py -------------------------------------------------------------------------------- /eq/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/__init__.py -------------------------------------------------------------------------------- /eq/conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/conversion.py -------------------------------------------------------------------------------- /eq/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/dataset.py -------------------------------------------------------------------------------- /eq/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/eval.py -------------------------------------------------------------------------------- /eq/op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/op.py -------------------------------------------------------------------------------- /eq/pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/pred.py -------------------------------------------------------------------------------- /eq/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/registry.py -------------------------------------------------------------------------------- /eq/tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/tree.py -------------------------------------------------------------------------------- /eq/vocabulary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq/vocabulary.py -------------------------------------------------------------------------------- /eq_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq_analyzer.py -------------------------------------------------------------------------------- /eq_comparator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/eq_comparator.py -------------------------------------------------------------------------------- /external/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/README.md -------------------------------------------------------------------------------- /external/ai_feynman2/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ai_feynman2/Dockerfile -------------------------------------------------------------------------------- /external/ai_feynman2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ai_feynman2/README.md -------------------------------------------------------------------------------- /external/ai_feynman2/ai_feynman2.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ai_feynman2/ai_feynman2.def -------------------------------------------------------------------------------- /external/ai_feynman2/ai_feynman_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ai_feynman2/ai_feynman_runner.py -------------------------------------------------------------------------------- /external/ai_feynman2/equation_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ai_feynman2/equation_converter.py -------------------------------------------------------------------------------- /external/dso/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/Dockerfile -------------------------------------------------------------------------------- /external/dso/Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/Pipfile -------------------------------------------------------------------------------- /external/dso/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/README.md -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_w_const1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_w_const1.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_w_const2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_w_const2.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_w_const3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_w_const3.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_w_const4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_w_const4.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_w_const5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_w_const5.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_wo_const1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_wo_const1.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_wo_const2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_wo_const2.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_wo_const3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_wo_const3.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_wo_const4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_wo_const4.json -------------------------------------------------------------------------------- /external/dso/configs/dsr/config_wo_const5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/dsr/config_wo_const5.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly1.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly10.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly10.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly2.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly3.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly4.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly5.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly6.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly6.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly7.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly7.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly8.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly8.json -------------------------------------------------------------------------------- /external/dso/configs/udsr/config_w_poly9.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/configs/udsr/config_w_poly9.json -------------------------------------------------------------------------------- /external/dso/dso.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/dso.def -------------------------------------------------------------------------------- /external/dso/equation_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/dso/equation_converter.py -------------------------------------------------------------------------------- /external/e2e/Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/e2e/Pipfile -------------------------------------------------------------------------------- /external/e2e/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/e2e/README.md -------------------------------------------------------------------------------- /external/e2e/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/e2e/runner.py -------------------------------------------------------------------------------- /external/ellyn/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/Dockerfile -------------------------------------------------------------------------------- /external/ellyn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/README.md -------------------------------------------------------------------------------- /external/ellyn/configs/afp_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/configs/afp_default.yaml -------------------------------------------------------------------------------- /external/ellyn/configs/afp_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/configs/afp_optuna.yaml -------------------------------------------------------------------------------- /external/ellyn/configs/fe_afp_default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/configs/fe_afp_default.yaml -------------------------------------------------------------------------------- /external/ellyn/configs/fe_afp_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/configs/fe_afp_optuna.yaml -------------------------------------------------------------------------------- /external/ellyn/ellyn.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/ellyn.def -------------------------------------------------------------------------------- /external/ellyn/ellyn_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/ellyn/ellyn_runner.py -------------------------------------------------------------------------------- /external/gplearn/Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/gplearn/Pipfile -------------------------------------------------------------------------------- /external/gplearn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/gplearn/README.md -------------------------------------------------------------------------------- /external/gplearn/configs/default_w_const.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/gplearn/configs/default_w_const.yaml -------------------------------------------------------------------------------- /external/gplearn/configs/default_wo_const.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/gplearn/configs/default_wo_const.yaml -------------------------------------------------------------------------------- /external/gplearn/configs/optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/gplearn/configs/optuna.yaml -------------------------------------------------------------------------------- /external/gplearn/gp_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/gplearn/gp_runner.py -------------------------------------------------------------------------------- /external/pysr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/pysr/README.md -------------------------------------------------------------------------------- /external/pysr/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/pysr/config.yaml -------------------------------------------------------------------------------- /external/pysr/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/external/pysr/runner.py -------------------------------------------------------------------------------- /model_selector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/model_selector.py -------------------------------------------------------------------------------- /r2_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/r2_evaluator.py -------------------------------------------------------------------------------- /random_dataset_generator.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/random_dataset_generator.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/omron-sinicx/srsd-benchmark/HEAD/requirements.txt --------------------------------------------------------------------------------