Commit 6eec6918 authored by Noric Couderc's avatar Noric Couderc

Fixed bug in train_model.py

The jmh_with_best variable was undefined.
parent a91a367a
......@@ -138,7 +138,7 @@ def compute_sample_weights(jmh_data):
The weight is computer by computing the ratio of improvement,
and adding it's log to 1/N, where N is the number of samples
"""
ratios = jmh_data["Lowest score_best"] / jmh_with_best["Lowest score"]
ratios = jmh_data["Lowest score_best"] / jmh_data["Lowest score"]
sample_weights = (1 / ratios.shape[0]) + numpy.log(ratios)
return sample_weights
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment