arnie23 commited on
Commit
9ce1e56
·
verified ·
1 Parent(s): ce57433

Delete geo_metals_model_fixed.py

Browse files
Files changed (1) hide show
  1. geo_metals_model_fixed.py +0 -33
geo_metals_model_fixed.py DELETED
@@ -1,33 +0,0 @@
1
- # geo_metals_model_fixed.py
2
- from sklearn.pipeline import Pipeline
3
- from sklearn.compose import ColumnTransformer
4
- from sklearn.preprocessing import OneHotEncoder
5
- from sklearn.multioutput import MultiOutputRegressor
6
- from sklearn.ensemble import GradientBoostingRegressor
7
-
8
- # Define preprocessing for categorical and numeric features
9
- preprocessor = ColumnTransformer(
10
- transformers=[
11
- ("cat", OneHotEncoder(handle_unknown="ignore"), ["type"]),
12
- ("num", "passthrough", ["lat", "lon"])
13
- ]
14
- )
15
-
16
- # Define the model pipeline
17
- model = Pipeline(steps=[
18
- ("preprocessor", preprocessor),
19
- ("regressor", MultiOutputRegressor(
20
- GradientBoostingRegressor(
21
- learning_rate=0.1,
22
- loss="squared_error",
23
- max_depth=3,
24
- max_features=None,
25
- min_samples_split=2,
26
- min_samples_leaf=1,
27
- subsample=1.0,
28
- n_estimators=100,
29
- random_state=None,
30
- verbose=0
31
- )
32
- ))
33
- ])