Talvez , mas observe que este é um daqueles casos em que o aprendizado de máquina não é a resposta . Há uma tendência de tentar aprender o aprendizado de máquina em casos em que as soluções baseadas em regras padrão são mais rápidas, mais simples e geralmente a escolha certa: P
Só porque você pode, não significa que você deveria
Edit : Eu originalmente escrevi isso como "Sim, mas note que ...", mas comecei a duvidar de mim mesmo, nunca tendo visto isso. Eu tentei esta tarde e é certamente factível:
import numpy as np
from keras.models import Model
from keras.layers import Input, Dense, Dropout
from keras.utils import to_categorical
from sklearn.model_selection import train_test_split
from keras.callbacks import EarlyStopping
# Create an input array of 50,000 samples of 20 random numbers each
x = np.random.randint(0, 100, size=(50000, 20))
# And a one-hot encoded target denoting the index of the maximum of the inputs
y = to_categorical(np.argmax(x, axis=1), num_classes=20)
# Split into training and testing datasets
x_train, x_test, y_train, y_test = train_test_split(x, y)
# Build a network, probaly needlessly complicated since it needs a lot of dropout to
# perform even reasonably well.
i = Input(shape=(20, ))
a = Dense(1024, activation='relu')(i)
b = Dense(512, activation='relu')(a)
ba = Dropout(0.3)(b)
c = Dense(256, activation='relu')(ba)
d = Dense(128, activation='relu')(c)
o = Dense(20, activation='softmax')(d)
model = Model(inputs=i, outputs=o)
es = EarlyStopping(monitor='val_loss', patience=3)
model.compile(optimizer='adam', loss='categorical_crossentropy')
model.fit(x_train, y_train, epochs=15, batch_size=8, validation_data=[x_test, y_test], callbacks=[es])
print(np.where(np.argmax(model.predict(x_test), axis=1) == np.argmax(y_test, axis=1), 1, 0).mean())
A saída é 0.74576, por isso está localizando corretamente o máximo de 74,5% do tempo. Não tenho dúvidas de que isso poderia ser melhorado, mas, como digo, esse não é um caso que eu recomendaria para o ML.
EDIT 2 : Na verdade, eu re-executei esta manhã usando o RandomForestClassifier do sklearn e o desempenho foi significativamente melhor:
# instantiation of the arrays is identical
rfc = RandomForestClassifier(n_estimators=1000, verbose=1)
rfc.fit(x_train, y_train)
yhat_proba = rfc.predict_proba(x_test)
# We have some annoying transformations to do because this .predict_proba() call returns the data in a weird format of shape (20, 12500, 2).
for i in range(len(yhat_proba)):
yhat_proba[i] = yhat_proba[i][:, 1]
pyhat = np.reshape(np.ravel(yhat_proba), (12500,20), order='F')
print(np.where(np.argmax(pyhat, axis=1) == np.argmax(y_test, axis=1), 1, 0).mean())
E a pontuação aqui é de 94,4% das amostras com o máximo identificado corretamente, o que é realmente muito bom.