Skip to content

Commit

Permalink
Remove signal runtime alarm from symbolic conversion
Browse files Browse the repository at this point in the history
  • Loading branch information
Sarah Krebs committed Jan 3, 2024
1 parent 1e75d3c commit 87b02bd
Showing 1 changed file with 17 additions and 23 deletions.
40 changes: 17 additions & 23 deletions deepcave/plugins/hyperparameter/symbolic_explanations.py
Original file line number Diff line number Diff line change
Expand Up @@ -319,32 +319,26 @@ def process(run, inputs):
symb_model = SymbolicRegressor(**symb_params)
symb_model.fit(x, y)

def handler(signo, frame):
raise RuntimeError

signal.signal(signal.SIGALRM, handler)
signal.alarm(6) # seconds
while True:
try:
conv_expr = (
f"{objective.name} = "
f"{convert_symb(symb_model, n_decimals=3, hp_names=selected_hyperparameters)}"
)
except:
conv_expr = (
"Conversion of the expression failed. Please try another seed or increase "
"the parsimony hyperparameter."
)
try:
conv_expr = (
f"{objective.name} = "
f"{convert_symb(symb_model, n_decimals=3, hp_names=selected_hyperparameters)}"
)
except:
conv_expr = (
"Conversion of the expression failed. Please try another seed or increase "
"the parsimony hyperparameter."
)

if len(conv_expr) > 115:
conv_expr = (
"Expression is too long to display. Please try another seed or increase "
"the parsimony hyperparameter."
)
if len(conv_expr) > 115:
conv_expr = (
"Expression is too long to display. Please try another seed or increase "
"the parsimony hyperparameter."
)

y_symbolic = symb_model.predict(x).tolist()
y_symbolic = symb_model.predict(x).tolist()

return {"x": x, "y": y_symbolic, "expr": conv_expr}
return {"x": x, "y": y_symbolic, "expr": conv_expr}

@staticmethod
def get_output_layout(register):
Expand Down

0 comments on commit 87b02bd

Please sign in to comment.