Spaces:
Sleeping
Sleeping
File size: 546 Bytes
249a203 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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 |