Skip to main content

Table 3 parameters of all machine learning models in this study

From: Application of machine learning model to predict osteoporosis based on abdominal computed tomography images of the psoas muscle: a retrospective study

Model

parameters

LR

penalty = 'l2', dual = False, tol = 0.0001, C = 1.0, fit_intercept = True, intercept_scaling = 1, class_weight = None, random_state = None, solver = 'lbfgs', max_iter = 100, multi_class = 'auto', verbose = 0, warm_start = False, n_jobs = None, l1_ratio = None

XGBoost

n_estimators = 500, learning_rate = 0.5, objective = 'binary:logistic', use_label_encoder = True

GNB

priors = None, var_smoothing = 1e-09

GBM

loss = 'deviance', learning_rate = 0.5, n_estimators = 500, subsample = 1.0, criterion = 'friedman_mse', min_samples_split = 2, min_samples_leaf = 1, min_weight_fraction_leaf = 0.0, max_depth = 3, min_impurity_decrease = 0.0, min_impurity_split = None, init = None, random_state = None, max_features = None, verbose = 0, max_leaf_nodes = None, warm_start = False, validation_fraction = 0.1, n_iter_no_change = None, tol = 0.0001, ccp_alpha = 0.0

RF

n_estimators = 100, criterion = 'gini', max_depth = None, min_samples_split = 2, min_samples_leaf = 1, min_weight_fraction_leaf = 0.0, max_features = 'auto', max_leaf_nodes = None, min_impurity_decrease = 0.0, min_impurity_split = None, bootstrap = True, oob_score = False, n_jobs = None, random_state = None, verbose = 0, warm_start = False, class_weight = None, ccp_alpha = 0.0, max_samples = None

SVM

C = 2.33, kernel = 'rbf', degree = 3, gamma = 2.15e-04, coef0 = 0.0, shrinking = True, probability = False, tol = 0.001, cache_size = 200, class_weight = None, verbose = False, max_iter = -1, decision_function_shape = 'ovr', break_ties = False, random_state = None

  1. Abbreviations: LR Logistic regression, GBM Gradient boosting machine, RF Random forest, GNB Gaussian naïve Bayes, XGBoost Extreme gradient boosting, SVM Support vector machines