caspervk/nightr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added weights for milo strats

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/activate.sh
View File

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

7
server/nightr/strategies/miloStrats.py
View File

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