Naaturel/chess-engine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Min samples and leaf max features

Browse Source
This commit is contained in:
Matthias Guillitte
2026-09-27 17:08:00 +02:00
parent 9de8906e0e
commit bacf54b8db
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
doc/hyperparametersOptimisation/min_samples_leaf_max_features.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 72 KiB

After

Width:  |  Height:  |  Size: 74 KiB

4
src/training/decision_tree/main.py
View File

@@ -106,7 +106,7 @@ if __name__ == "__main__":
"label": "Max features",
"values": [
round(value, 2)
for value in np.arange(0.2, 0.5, 0.05)
for value in np.arange(0.45, 1, 0.05)
]
}
base_model_parameters = {
@@ -129,7 +129,7 @@ if __name__ == "__main__":
# Plot graph
fig, ax = plt.subplots()
scatter = ax.scatter([], [], c=[], cmap="RdYlGn_r", vmin=0, vmax=2)
scatter = ax.scatter([], [], c=[], cmap="RdYlGn_r", vmin=1.3, vmax=1.6)
colorbar = fig.colorbar(scatter, ax=ax)
colorbar.set_label("Test MAE")
ax.set_xlabel(parameter_1["label"])