Skip to main content
Fig. 6 | BMC Bioinformatics

Fig. 6

From: A new method for exploring gene–gene and gene–environment interactions in GWAS with tree ensemble methods and SHAP values

Fig. 6

Given a table of ranked feature importances, XGBoost models based on the top K features are trained in a new cross-validation procedure based on an independent set of individuals, namely the fitting data. We search for the XGBoost models that on average performs the best for a given set of hyperparameters (including the value of K) based on test data

Back to article page