added weights for milo strats

This commit is contained in:
Mikkel Milo 2019-04-06 17:40:51 +02:00
parent 945b0c969f
commit 9997aca958
2 changed files with 4 additions and 4 deletions

View File

@ -12,7 +12,6 @@ source venv/bin/activate
echo Installing required Python packages
pip install -Ur requirements.txt
function run() {
python -m nightr
}

View File

@ -14,8 +14,9 @@ def camImgStrat(context : Context) -> Prediction:
"""
img = cv2.imread(str(Path(__file__).parent.joinpath("night.jpg")), 0)
average = img.mean(axis=0).mean(axis=0)
print(average)
p = Prediction()
p.weight = 0.7
if average < 100:
p.probability = 1.0
p.reasons.append('Image was dark')
@ -33,6 +34,7 @@ def australiaStrat(context : Context) -> Prediction:
t = datetime.now().astimezone(australia)
hour = t.hour
p = Prediction()
if hour > 22 or hour < 6:
p.probability = 0.0
p.reasons.append('It\'s night-time in Australia')
@ -55,9 +57,8 @@ def tv2newsStrat(context : Context) -> Prediction:
avg_delta += d
avg_timestamp = avg_delta // len(delta_times) // 60
p = Prediction()
print('average time between articles on tv2:', avg_timestamp, 'minutes')
p.weight = 0.7
p.probability = 1.0 if avg_timestamp > 50 else 0.0
p.reasons.append('There were ' + ('few' if avg_timestamp > 50 else 'many') + ' recent articles on TV2 News')
print(p.reasons[0])
return p