OpenAlex
/

distilbert-base-cased-finetuned-topic-classification-title-abstract

@@ -15,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 2.8565
-- Validation Loss: 3.2441
-- Train Accuracy: 0.4598
 - Epoch: 9
 ## Model description
@@ -44,16 +44,16 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 5.2222     | 3.9668          | 0.3397         | 0     |
-| 3.8730     | 3.5696          | 0.4003         | 1     |
-| 3.5292     | 3.4159          | 0.4255         | 2     |
-| 3.3343     | 3.3431          | 0.4383         | 3     |
-| 3.1978     | 3.2940          | 0.4470         | 4     |
-| 3.0928     | 3.2639          | 0.4521         | 5     |
-| 3.0089     | 3.2509          | 0.4557         | 6     |
-| 2.9438     | 3.2432          | 0.4578         | 7     |
-| 2.8928     | 3.2416          | 0.4593         | 8     |
-| 2.8565     | 3.2441          | 0.4598         | 9     |
 ### Framework versions

 This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Train Loss: 2.8398
+- Validation Loss: 3.2378
+- Train Accuracy: 0.4618
 - Epoch: 9
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 5.1779     | 3.9338          | 0.3457         | 0     |
+| 3.8441     | 3.5523          | 0.4044         | 1     |
+| 3.5070     | 3.4169          | 0.4267         | 2     |
+| 3.3152     | 3.3286          | 0.4402         | 3     |
+| 3.1797     | 3.2789          | 0.4488         | 4     |
+| 3.0756     | 3.2612          | 0.4537         | 5     |
+| 2.9929     | 3.2459          | 0.4575         | 6     |
+| 2.9266     | 3.2380          | 0.4598         | 7     |
+| 2.8758     | 3.2390          | 0.4611         | 8     |
+| 2.8398     | 3.2378          | 0.4618         | 9     |
 ### Framework versions

config.json CHANGED Viewed

The diff for this file is too large to render. See raw diff

tf_model.h5 CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:65bc7363e3aff225b9dbb9497dd4024b4e76e0632942fbf82043fe79867dee87
-size 277159264

 version https://git-lfs.github.com/spec/v1
+oid sha256:60fc4c026c190e1c1a85be8c5655e8532d42077d8f18317a9274a4bfec554ca4
+size 277162340