AutonLabTruth commited on
Commit
e2a7e95
1 Parent(s): 271a473

Added timing feature

Browse files
Files changed (1) hide show
  1. pysr/Problems.py +5 -1
pysr/Problems.py CHANGED
@@ -93,9 +93,13 @@ def run_on_problem(problem, verbosity=0):
93
  """
94
  Takes in a problem and returns a tuple: (equations, best predicted equation, actual equation)
95
  """
 
96
  from . import pysr, best
 
97
  equations = pysr(problem.X, problem.y, variable_names=problem.variable_names, verbosity=verbosity)
98
- return equations, best(equations), problem.form
 
 
99
 
100
  if __name__ == "__main__":
101
  ret = FeynmanProblem.mk_problems(first=100, gen=True)
 
93
  """
94
  Takes in a problem and returns a tuple: (equations, best predicted equation, actual equation)
95
  """
96
+ from time import time
97
  from . import pysr, best
98
+ starting = time()
99
  equations = pysr(problem.X, problem.y, variable_names=problem.variable_names, verbosity=verbosity)
100
+ timing = time()-starting
101
+ others = {"equations": equations, "time": timing}
102
+ return best(equations), problem.form, others
103
 
104
  if __name__ == "__main__":
105
  ret = FeynmanProblem.mk_problems(first=100, gen=True)