farrukhshahzad25 commited on
Commit
85fbc65
1 Parent(s): 5c7e9aa

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - tabular
6
+ - regression
7
+ - tabular-regression
8
+ datasets:
9
+ - autotrain-vessel-eta/autotrain-data
10
+ ---
11
+
12
+ # Model Trained Using AutoTrain
13
+
14
+ - Problem type: Tabular regression
15
+
16
+ ## Validation Metrics
17
+
18
+ - r2: 0.2033836841583252
19
+ - mse: 53092.500978994
20
+ - mae: 150.96381290340423
21
+ - rmse: 230.41810037189788
22
+ - rmsle: 0.9569819523414094
23
+ - loss: 230.41810037189788
24
+
25
+ ## Best Params
26
+
27
+ - learning_rate: 0.0695392185390836
28
+ - reg_lambda: 0.00017542491817558795
29
+ - reg_alpha: 0.6577124531542021
30
+ - subsample: 0.3632574815242663
31
+ - colsample_bytree: 0.8007491192913739
32
+ - max_depth: 4
33
+ - early_stopping_rounds: 166
34
+ - n_estimators: 15000
35
+ - eval_metric: rmse
36
+
37
+ ## Usage
38
+
39
+ ```python
40
+ import json
41
+ import joblib
42
+ import pandas as pd
43
+
44
+ model = joblib.load('model.joblib')
45
+ config = json.load(open('config.json'))
46
+
47
+ features = config['features']
48
+
49
+ # data = pd.read_csv("data.csv")
50
+ data = data[features]
51
+
52
+ predictions = model.predict(data) # or model.predict_proba(data)
53
+
54
+ # predictions can be converted to original labels using label_encoders.pkl
55
+
56
+ ```
autotrain-data/dataset_dict.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"splits": ["train", "validation"]}
autotrain-data/train/data-00000-of-00001.arrow ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45498615635aa8b54deb3913db993b6e65aba4aa5492a449d7450899aed6501f
3
+ size 6528672
autotrain-data/train/dataset_info.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "citation": "",
3
+ "description": "",
4
+ "features": {
5
+ "course": {
6
+ "dtype": "float64",
7
+ "_type": "Value"
8
+ },
9
+ "heading": {
10
+ "dtype": "float64",
11
+ "_type": "Value"
12
+ },
13
+ "latitude": {
14
+ "dtype": "float64",
15
+ "_type": "Value"
16
+ },
17
+ "longitude": {
18
+ "dtype": "float64",
19
+ "_type": "Value"
20
+ },
21
+ "speed": {
22
+ "dtype": "float64",
23
+ "_type": "Value"
24
+ },
25
+ "timestamp": {
26
+ "dtype": "int64",
27
+ "_type": "Value"
28
+ },
29
+ "destination_port_index": {
30
+ "dtype": "int64",
31
+ "_type": "Value"
32
+ },
33
+ "autotrain_id": {
34
+ "dtype": "int64",
35
+ "_type": "Value"
36
+ },
37
+ "autotrain_label": {
38
+ "dtype": "int64",
39
+ "_type": "Value"
40
+ }
41
+ },
42
+ "homepage": "",
43
+ "license": ""
44
+ }
autotrain-data/train/state.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_data_files": [
3
+ {
4
+ "filename": "data-00000-of-00001.arrow"
5
+ }
6
+ ],
7
+ "_fingerprint": "100187c468cd0ab1",
8
+ "_format_columns": null,
9
+ "_format_kwargs": {},
10
+ "_format_type": null,
11
+ "_output_all_columns": false,
12
+ "_split": null
13
+ }
autotrain-data/validation/data-00000-of-00001.arrow ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd08f9903599b1da8a1a6e54bfee050deb06a2bba2087389a821cd3e201c78ce
3
+ size 726432
autotrain-data/validation/dataset_info.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "citation": "",
3
+ "description": "",
4
+ "features": {
5
+ "course": {
6
+ "dtype": "float64",
7
+ "_type": "Value"
8
+ },
9
+ "heading": {
10
+ "dtype": "float64",
11
+ "_type": "Value"
12
+ },
13
+ "latitude": {
14
+ "dtype": "float64",
15
+ "_type": "Value"
16
+ },
17
+ "longitude": {
18
+ "dtype": "float64",
19
+ "_type": "Value"
20
+ },
21
+ "speed": {
22
+ "dtype": "float64",
23
+ "_type": "Value"
24
+ },
25
+ "timestamp": {
26
+ "dtype": "int64",
27
+ "_type": "Value"
28
+ },
29
+ "destination_port_index": {
30
+ "dtype": "int64",
31
+ "_type": "Value"
32
+ },
33
+ "autotrain_id": {
34
+ "dtype": "int64",
35
+ "_type": "Value"
36
+ },
37
+ "autotrain_label": {
38
+ "dtype": "int64",
39
+ "_type": "Value"
40
+ }
41
+ },
42
+ "homepage": "",
43
+ "license": ""
44
+ }
autotrain-data/validation/state.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_data_files": [
3
+ {
4
+ "filename": "data-00000-of-00001.arrow"
5
+ }
6
+ ],
7
+ "_fingerprint": "f6d897442f042e24",
8
+ "_format_columns": null,
9
+ "_format_kwargs": {},
10
+ "_format_type": null,
11
+ "_output_all_columns": false,
12
+ "_split": null
13
+ }
model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9688b8dd5ca1c633c21a3107b7338010b793a8203c998aa74cad47e52511c50
3
+ size 413606
target_encoders.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:926248e52d1fa532c317e37da24ed652ae64110f8219cb5e061668bd3091f048
3
+ size 5
training_params.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"data_path": "autotrain-vessel-eta/autotrain-data", "model": "xgboost", "username": "farrukhshahzad25", "seed": 42, "train_split": "train", "valid_split": "validation", "project_name": "autotrain-vessel-eta", "push_to_hub": true, "id_column": "autotrain_id", "target_columns": ["autotrain_label"], "categorical_columns": null, "numerical_columns": ["course", "heading", "latitude", "longitude", "speed", "timestamp", "destination_port_index"], "task": "regression", "num_trials": 10, "time_limit": 600, "categorical_imputer": null, "numerical_imputer": null, "numeric_scaler": null}