Kolibri Games interview question

Give some improvement suggestions for the python model (prediction)