Skip to main content

Table 2 The optimal hyperparameters of each ML model obtained through a grid search for TNB and NBA traits in 20 replicates of 5-fold CV

From: Using machine learning to improve the accuracy of genomic prediction of reproduction traits in pigs

Method Optimal hyperparametersa
SVR kernel = ‘rbf’, C = 7, gamma = 0.0001
KRR kernel = ‘rbf’, λ =0.1, gamma = 0.0001
RF n_estimators = 250, max_depth = None
Adaboost.R2_SVR n_estimators = 50, kernel = ‘rbf’, C = 7, gamma = 0.0001
Adaboost.R2_KRR n_estimators = 50, kernel = ‘rbf’, λ =0.01, gamma = 0.0001
  1. a Optimal hyperparameters: The optimal hyperparameters of each machine learning method obtained by using a grid search