7.1.9. mclearn.classifier.grid_search_svm_poly¶
-
mclearn.classifier.
grid_search_svm_poly
(X, y, train_size=300, test_size=300, fig_path=None, pickle_path=None)[source]¶ Do a grid search on SVM with polynomial transformation of the features.
Parameters: - X (array) – The feature matrix of the data.
- y (array) – The target column.
- train_size (int) – The size of the training set in each iteration.
- test_size (int) – The size of the test set in each iteration.
- fig_path (str) – The path where the heat map plot can be saved.
- pickle_path (str) – The path where the pickled scores can be saved.