MilesCranmer commited on
Commit
4f80ce5
1 Parent(s): 121e6ac

Update hyperopt script to new API

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. hyperparamopt.py +1 -1
README.md CHANGED
@@ -161,7 +161,7 @@ name as a string.
161
 
162
  What follows is the API reference for running the numpy interface.
163
  You likely don't need to tune the hyperparameters yourself,
164
- but if you would like, you can use `hyperopt.py` as an example.
165
  However, you should adjust `procs`, `niterations`,
166
  `binary_operators`, `unary_operators`, and `maxsize`
167
  to your requirements.
 
161
 
162
  What follows is the API reference for running the numpy interface.
163
  You likely don't need to tune the hyperparameters yourself,
164
+ but if you would like, you can use `hyperparamopt.py` as an example.
165
  However, you should adjust `procs`, `niterations`,
166
  `binary_operators`, `unary_operators`, and `maxsize`
167
  to your requirements.
hyperparamopt.py CHANGED
@@ -79,7 +79,7 @@ def run_trial(args):
79
  print(f"Starting trial {j}")
80
  trial = pysr.pysr(
81
  test=f"simple{i}",
82
- threads=4,
83
  binary_operators=["plus", "mult", "pow", "div"],
84
  unary_operators=["cos", "exp", "sin", "loga", "abs"],
85
  equation_file=equation_file,
 
79
  print(f"Starting trial {j}")
80
  trial = pysr.pysr(
81
  test=f"simple{i}",
82
+ procs=4,
83
  binary_operators=["plus", "mult", "pow", "div"],
84
  unary_operators=["cos", "exp", "sin", "loga", "abs"],
85
  equation_file=equation_file,