Merge pull request #19 from google/for_demo

More consistent demo
This commit is contained in:
Shuhei Iitsuka
2018-09-11 15:23:55 +09:00
committed by GitHub

View File

@@ -16,6 +16,12 @@
# limitations under the License. # limitations under the License.
from __future__ import print_function from __future__ import print_function
# Surpress warnings. Please unvail these warnings when development.
import warnings
warnings.filterwarnings("ignore", category=RuntimeWarning)
warnings.filterwarnings("ignore", category=FutureWarning)
from nazoru.led import LED_BLUE, LED_RED, LED_CHASSIS from nazoru.led import LED_BLUE, LED_RED, LED_CHASSIS
import time import time
LED_RED.blink(1) LED_RED.blink(1)
@@ -67,7 +73,16 @@ def main():
LED_CHASSIS.set_brightness(100) LED_CHASSIS.set_brightness(100)
LED_RED.on() LED_RED.on()
try:
result = predictor.predict_top_n(data, 5) result = predictor.predict_top_n(data, 5)
except IndexError:
# This is possible when pressing keys we don't use for learning.
print('Invalid input. (out of range)')
continue
except ValueError:
# This is possible when pressing only one key.
print('Invalid input. (one key)')
continue
LED_RED.off() LED_RED.off()
LED_CHASSIS.set_brightness(5) LED_CHASSIS.set_brightness(5)