Skip to content

Instantly share code, notes, and snippets.

@initcron
Created May 14, 2025 06:22
Show Gist options
  • Save initcron/702de323bab9a3b85ee3cde295d06d49 to your computer and use it in GitHub Desktop.
Save initcron/702de323bab9a3b85ee3cde295d06d49 to your computer and use it in GitHub Desktop.
Sample Model Config with XGBoost
model:
best_model: GradientBoosting
feature_sets:
rfe:
- '0'
- '1'
- '2'
- '3'
- '4'
- '5'
- '9'
- '10'
- '13'
- '14'
mae: 12310.638379356631
name: house_price_model
parameters:
alpha: 0.9
ccp_alpha: 0.0
criterion: friedman_mse
init: null
learning_rate: 0.1
loss: squared_error
max_depth: 3
max_features: null
max_leaf_nodes: null
min_impurity_decrease: 0.0
min_samples_leaf: 1
min_samples_split: 2
min_weight_fraction_leaf: 0.0
n_estimators: 100
n_iter_no_change: null
random_state: null
subsample: 1.0
tol: 0.0001
validation_fraction: 0.1
verbose: 0
warm_start: false
r2_score: 0.9937776455685025
target_variable: price
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment