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 |