Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,10 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
# My Model
|
| 5 |
|
|
@@ -21,7 +26,7 @@ model = load_model("my_model.h5")
|
|
| 21 |
predictions = model.predict(X)
|
| 22 |
Model Details
|
| 23 |
```
|
| 24 |
-
Framework: TensorFlow / Keras
|
| 25 |
Input features: Text and numerical features
|
| 26 |
Output: Duration category (short, medium, long)
|
| 27 |
Performance
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
library_name: keras
|
| 6 |
+
pipeline_tag: tabular-classification
|
| 7 |
+
inference: true
|
| 8 |
---
|
| 9 |
# My Model
|
| 10 |
|
|
|
|
| 26 |
predictions = model.predict(X)
|
| 27 |
Model Details
|
| 28 |
```
|
| 29 |
+
Framework: TensorFlow / Keras / Bert
|
| 30 |
Input features: Text and numerical features
|
| 31 |
Output: Duration category (short, medium, long)
|
| 32 |
Performance
|