diff --git a/mlfromscratch/supervised_learning/xgboost.py b/mlfromscratch/supervised_learning/xgboost.py index 75b7ccf9..afbc1f38 100644 --- a/mlfromscratch/supervised_learning/xgboost.py +++ b/mlfromscratch/supervised_learning/xgboost.py @@ -55,7 +55,7 @@ def __init__(self, n_estimators=200, learning_rate=0.001, min_samples_split=2, min_impurity=1e-7, max_depth=2): self.n_estimators = n_estimators # Number of trees self.learning_rate = learning_rate # Step size for weight update - self.min_samples_split = min_samples_split # The minimum n of sampels to justify split + self.min_samples_split = min_samples_split # The minimum n of samples to justify split self.min_impurity = min_impurity # Minimum variance reduction to continue self.max_depth = max_depth # Maximum depth for tree