Tal vez , pero tenga en cuenta que este es uno de esos casos en los que el aprendizaje automático no es la respuesta . Existe una tendencia a intentar el aprendizaje automático de calzador en casos donde realmente, las soluciones basadas en reglas estándar son más rápidas, más simples y, en general, la elección correcta: P
Solo porque puedas, no significa que debas
Editar : Originalmente escribí esto como "Sí, pero tenga en cuenta que ...", pero luego comencé a dudar de mí mismo, ya que nunca lo había visto. Lo probé esta tarde y ciertamente es factible:
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())
La salida es 0.74576, por lo que está encontrando correctamente el máximo 74.5% del tiempo. No tengo dudas de que eso podría mejorarse, pero como digo, este no es un caso de uso que recomendaría para ML.
EDIT 2 : en realidad volví a ejecutar esta mañana usando el RandomForestClassifier de sklearn y funcionó significativamente mejor:
# 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())
Y la puntuación aquí es del 94,4% de las muestras con el máximo correctamente identificado, lo cual es bastante bueno.