Add fixes identified by test cases
parent
4cec9b0767
commit
a37958f3f2
|
@ -82,4 +82,6 @@ def asigmoid(x):
|
||||||
|
|
||||||
def pdf(x, mu, std):
|
def pdf(x, mu, std):
|
||||||
"""Probability density function (normal distribution)"""
|
"""Probability density function (normal distribution)"""
|
||||||
|
if std == 0:
|
||||||
|
return 0
|
||||||
return (1.0 / (std * sqrt(2 * pi))) * np.exp(-(x - mu) ** 2 / (2 * std ** 2))
|
return (1.0 / (std * sqrt(2 * pi))) * np.exp(-(x - mu) ** 2 / (2 * std ** 2))
|
||||||
|
|
|
@ -62,6 +62,8 @@ class EngineScript(BaseScript):
|
||||||
stdout.buffer.flush()
|
stdout.buffer.flush()
|
||||||
except (EOFError, KeyboardInterrupt):
|
except (EOFError, KeyboardInterrupt):
|
||||||
pass
|
pass
|
||||||
|
finally:
|
||||||
|
sys.stdout = stdout
|
||||||
|
|
||||||
|
|
||||||
main = EngineScript.run_main
|
main = EngineScript.run_main
|
||||||
|
|
|
@ -34,9 +34,12 @@ class ThresholdDecoder:
|
||||||
def decode(self, raw_output: float) -> float:
|
def decode(self, raw_output: float) -> float:
|
||||||
if raw_output == 1.0 or raw_output == 0.0:
|
if raw_output == 1.0 or raw_output == 0.0:
|
||||||
return raw_output
|
return raw_output
|
||||||
ratio = (asigmoid(raw_output) - self.min_out) / self.out_range
|
if self.out_range == 0:
|
||||||
ratio = min(max(ratio, 0.0), 1.0)
|
cp = int(raw_output > self.min_out)
|
||||||
cp = self.cd[int(ratio * (len(self.cd) - 1) + 0.5)]
|
else:
|
||||||
|
ratio = (asigmoid(raw_output) - self.min_out) / self.out_range
|
||||||
|
ratio = min(max(ratio, 0.0), 1.0)
|
||||||
|
cp = self.cd[int(ratio * (len(self.cd) - 1) + 0.5)]
|
||||||
if cp < self.center:
|
if cp < self.center:
|
||||||
return 0.5 * cp / self.center
|
return 0.5 * cp / self.center
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -107,6 +107,10 @@ class ReadWriteStream(object):
|
||||||
self.buffer += s
|
self.buffer += s
|
||||||
self.write_event.set()
|
self.write_event.set()
|
||||||
|
|
||||||
|
def flush(self):
|
||||||
|
"""Makes compatible with sys.stdout"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
class TriggerDetector:
|
class TriggerDetector:
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue