Update README.md
Browse files
README.md
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
license: mit
|
3 |
language:
|
4 |
- en
|
5 |
-
pipeline_tag:
|
6 |
tags:
|
7 |
- pytorch
|
8 |
- mlflow
|
@@ -12,7 +12,7 @@ tags:
|
|
12 |
---
|
13 |
## Scaling-ML
|
14 |
Scaling-ML is a project that classifies news headlines into 10 groups.
|
15 |
-
The main part of the project fine-tuning of the [BERT](https://huggingface.co/allenai/scibert_scivocab_uncased)[1] model and including tools like MLflow for tracking experiments, Ray for scaling and distibuted computing, and MLOps components for seamless management of machine learning workflows
|
16 |
|
17 |
### Set Up
|
18 |
|
|
|
2 |
license: mit
|
3 |
language:
|
4 |
- en
|
5 |
+
pipeline_tag: text-classification
|
6 |
tags:
|
7 |
- pytorch
|
8 |
- mlflow
|
|
|
12 |
---
|
13 |
## Scaling-ML
|
14 |
Scaling-ML is a project that classifies news headlines into 10 groups.
|
15 |
+
The main part of the project fine-tuning of the [BERT](https://huggingface.co/allenai/scibert_scivocab_uncased)[1] model and including tools like MLflow for tracking experiments, Ray for scaling and distibuted computing, and MLOps components for seamless management of machine learning workflows.
|
16 |
|
17 |
### Set Up
|
18 |
|