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 |