Efficient Hyperparameter Tuning with Grid Search for Text Categorization using kNN Approach with BM25 Similarity

Raji Ghawi, Jürgen Pfeffer

Research output: Contribution to journalArticlepeer-review

89 Scopus citations

Abstract

In machine learning, hyperparameter tuning is the problem of choosing a set of optimal hyperparameters for a learning algorithm. Several approaches have been widely adopted for hyperparameter tuning, which is typically a time consuming process. We propose an efficient technique to speed up the process of hyperparameter tuning with Grid Search. We applied this technique on text categorization using kNN algorithm with BM25 similarity, where three hyperparameters need to be tuned. Our experiments show that our proposed technique is at least an order of magnitude faster than conventional tuning.

Original languageEnglish
Pages (from-to)160-180
Number of pages21
JournalOpen Computer Science
Volume9
Issue number1
DOIs
StatePublished - 1 Jan 2019

Keywords

  • BM25
  • grid search
  • hyperparameter tuning
  • kNN
  • text categorization

Fingerprint

Dive into the research topics of 'Efficient Hyperparameter Tuning with Grid Search for Text Categorization using kNN Approach with BM25 Similarity'. Together they form a unique fingerprint.

Cite this