diff --git a/Code/S_run_aifeynman.py b/Code/S_run_aifeynman.py index 671c8fa..0cce00b 100644 --- a/Code/S_run_aifeynman.py +++ b/Code/S_run_aifeynman.py @@ -25,7 +25,7 @@ from S_add_bf_on_numbers_on_pareto import add_bf_on_numbers_on_pareto from dimensionalAnalysis import dimensionalAnalysis PA = ParetoSet() -def run_AI_all(pathdir,filename,BF_try_time=60,BF_ops_file_type="14ops", polyfit_deg=4, NN_epochs=4000, PA=PA): +def run_AI_all(pathdir,filename,BF_try_time=60,BF_ops_file_type="14ops", polyfit_deg=3, NN_epochs=4000, PA=PA): try: os.mkdir("results/") except: @@ -135,7 +135,7 @@ def run_AI_all(pathdir,filename,BF_try_time=60,BF_ops_file_type="14ops", polyfit return PA # this runs snap on the output of aifeynman -def run_aifeynman(pathdir,filename,BF_try_time,BF_ops_file_type, polyfit_deg=4, NN_epochs=4000, vars_name=[],test_percentage=20): +def run_aifeynman(pathdir,filename,BF_try_time,BF_ops_file_type, polyfit_deg=3, NN_epochs=4000, vars_name=[],test_percentage=20): # If the variable names are passed, do the dimensional analysis first filename_orig = filename try: diff --git a/Code/S_run_bf_polyfit.py b/Code/S_run_bf_polyfit.py index 66ca513..795e653 100644 --- a/Code/S_run_bf_polyfit.py +++ b/Code/S_run_bf_polyfit.py @@ -18,7 +18,7 @@ import os from os import path -def run_bf_polyfit(pathdir,pathdir_transformed,filename,BF_try_time,BF_ops_file_type, PA, polyfit_deg=4, output_type=""): +def run_bf_polyfit(pathdir,pathdir_transformed,filename,BF_try_time,BF_ops_file_type, PA, polyfit_deg=3, output_type=""): ############################################################################################################################# diff --git a/Code/ai_feynman_example.py b/Code/ai_feynman_example.py index d3d2523..5c20291 100644 --- a/Code/ai_feynman_example.py +++ b/Code/ai_feynman_example.py @@ -1,4 +1,4 @@ from S_run_aifeynman import run_aifeynman -run_aifeynman("../example_data/","example2.txt",30,"14ops.txt", polyfit_deg=4, NN_epochs=400) +run_aifeynman("../example_data/","example2.txt",30,"14ops.txt", polyfit_deg=3, NN_epochs=400) diff --git a/Code/ai_feynman_terminal_example.py b/Code/ai_feynman_terminal_example.py index 4dd8e13..1644c80 100644 --- a/Code/ai_feynman_terminal_example.py +++ b/Code/ai_feynman_terminal_example.py @@ -7,7 +7,7 @@ parser.add_argument("--pathdir", type=str, help="Path to the directory containin parser.add_argument("--filename", type=str, help="Name of the file containing the data") parser.add_argument("--BF_try_time", type=float, default=60, help="Time limit for each brute force code call") parser.add_argument("--BF_ops_file_type", type=str, default="14ops.txt", help="File containing the symbols to be used in the brute force code") -parser.add_argument("--polyfit_deg", type=int, default=4, help="Maximum degree of the polynomial tried by the polynomial fit routine") +parser.add_argument("--polyfit_deg", type=int, default=3, help="Maximum degree of the polynomial tried by the polynomial fit routine") parser.add_argument("--NN_epochs", type=int, default=2000, help="Number of epochs for the training") parser.add_argument("--vars_name", type=list, default=[], help="List with the names of the variables") parser.add_argument("--test_percentage", type=float, default=0, help="Percentage of the input data to be kept as the test set")