Spaces:
Sleeping
Sleeping
| import numpy as np | |
| import model_2 | |
| model = model_2.model | |
| cols = model_2.target2.columns | |
| def encode(encoders, input_data): | |
| model_input=[] | |
| for i in range(len(input_data)): | |
| row=[] | |
| for j in range(18): | |
| row.extend(encoders[j].transform([input_data.iloc[i][j]])) | |
| model_input.append(row) | |
| model_input=np.array(model_input) | |
| return model_input | |
| def input_output(user_input): | |
| model_input=encode(model_2.encoders, user_input) | |
| output=model.predict(model_input) | |
| final_result = cols[np.argmax(output)] | |
| return final_result |