Bluepearls commited on
Commit
7dd5f4b
·
1 Parent(s): a995fa9

commit from bluepearl

Browse files
Files changed (1) hide show
  1. config.json +25 -18
config.json CHANGED
@@ -1,20 +1,27 @@
1
  {
2
- "model_name": "Random-Forest-Classification"
3
- "description": "A pre-trained Random Forest model",
4
- "author": "Bluepearl",
5
- "version": "1.0.0",
6
- "architecture": "Tabular Classification",
7
- "parameters": {
8
- "hidden_size": 768,
9
- "num_attention_heads": 12,
10
- "num_hidden_layers": 12
11
- },
12
- "training_data": "",
13
- "hyperparameters": {
14
- "learning_rate": 0.0001,
15
- "batch_size": 32,
16
- "max_seq_length": 128
17
- },
18
- "output_format": "Json",
19
- "license": "MIT License"
 
 
 
 
 
 
 
20
  }
 
1
  {
2
+ "sklearn": {
3
+ "columns": [
4
+ "PassengerId",
5
+ "Survived",
6
+ "Pclass",
7
+ "Name",
8
+ "Sex",
9
+ "Age",
10
+ "SibSp",
11
+ "Parch",
12
+ "Ticket",
13
+ "Fare",
14
+ "Cabin",
15
+ "Embarked"
16
+ ],
17
+ "environment": [
18
+ "scikit-learn=1.3.0"
19
+ ],
20
+ "model": {
21
+ "file": "model.pkl"
22
+ },
23
+ "model_format": "pickle",
24
+ "task": "tabular-classification",
25
+ "use_intelex": false
26
+ }
27
  }