added weights for milo strats
This commit is contained in:
parent
945b0c969f
commit
9997aca958
|
@ -12,7 +12,6 @@ source venv/bin/activate
|
|||
echo Installing required Python packages
|
||||
pip install -Ur requirements.txt
|
||||
|
||||
|
||||
function run() {
|
||||
python -m nightr
|
||||
}
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in a new issue