For example line 216 : input_data['risk'] = 'bad' is a hard coded string value. this can be improved by using encoding like LabelEncoder