Je cours un GridSearchCV (Grid Search Cross Validation) à partir de la bibliothèque Sklearn sur un SGDClassifier (classificateur de descente de gradient stochastique). J'utilise un DataFrame de Pandas pour les fonctionnalités et la cible. Voici le code:Sklearn GridSearchCV utilisant la colonne Pandas DataFrame
from sklearn.grid_search import GridSearchCV
parameters = {'loss': [ 'hinge', 'log', 'modified_huber', 'squared_hinge', 'perceptron'], 'alpha': [0.1, 0.01, 0.001, 0.0001, 0.00001, 0.000001, 0.0000001], 'n_iter': list(np.arange(1,1001))}
clf = GridSearchCV(estimator = SGDClassifier(), param_grid = parameters, scoring = 'f1')
print(clf)
clf.fit(X_train, y_train)
Où est X_train est un 300 lignes x 31 colonnes Pandas dataframe avec chaque colonne nommée par ce qui suit:
['school', 'sex', 'age', 'address', 'famsize', 'Pstatus', 'Medu', 'Fedu', 'Mjob', 'Fjob', 'reason', 'guardian', 'traveltime', 'studytime', 'failures', 'schoolsup', 'famsup', 'paid', 'activities', 'nursery', 'higher', 'internet', 'romantic', 'famrel', 'freetime', 'goout', 'Dalc', 'Walc', 'health', 'absences']
Et y_train est un 300 lignes x 1 colonne Pandas série nommée par les éléments suivants:
['passed']
Lorsque je tente l'algorithme de GridSearchCV, je me fais la déclaration d'erreur suivant:
IndexError: too many indices for array