rycerzes commited on
Commit
e5443cd
·
verified ·
1 Parent(s): bc32363

Upload folder using huggingface_hub

Browse files
Files changed (9) hide show
  1. .gitattributes +1 -0
  2. README.md +59 -0
  3. config.json +4 -0
  4. gate_meta.json +7 -0
  5. metrics.json +7 -0
  6. model.safetensors +3 -0
  7. modules.json +14 -0
  8. pipeline.skops +3 -0
  9. tokenizer.json +0 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ pipeline.skops filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unknown
3
+ library_name: model2vec
4
+ license: mit
5
+ model_name: trained_gate
6
+ tags:
7
+ - embeddings
8
+ - static-embeddings
9
+ - sentence-transformers
10
+ ---
11
+
12
+ # trained_gate Model Card
13
+
14
+ This [Model2Vec](https://github.com/MinishLab/model2vec) model is a fine-tuned version of the [unknown](https://huggingface.co/unknown) Model2Vec model. It also includes a classifier head on top.
15
+
16
+ ## Installation
17
+
18
+ Install model2vec using pip:
19
+ ```
20
+ pip install model2vec[inference]
21
+ ```
22
+
23
+ ## Usage
24
+ Load this model using the `from_pretrained` method:
25
+ ```python
26
+ from model2vec.inference import StaticModelPipeline
27
+
28
+ # Load a pretrained Model2Vec model
29
+ model = StaticModelPipeline.from_pretrained("trained_gate")
30
+
31
+ # Predict labels
32
+ predicted = model.predict(["Example sentence"])
33
+ ```
34
+
35
+ ## Additional Resources
36
+
37
+ - [Model2Vec Repo](https://github.com/MinishLab/model2vec)
38
+ - [Model2Vec Base Models](https://huggingface.co/collections/minishlab/model2vec-base-models-66fd9dd9b7c3b3c0f25ca90e)
39
+ - [Model2Vec Results](https://github.com/MinishLab/model2vec/tree/main/results)
40
+ - [Model2Vec Docs](https://minish.ai/packages/model2vec/introduction)
41
+
42
+ ## Library Authors
43
+
44
+ Model2Vec was developed by the [Minish Lab](https://github.com/MinishLab) team consisting of [Stephan Tulkens](https://github.com/stephantul) and [Thomas van Dongen](https://github.com/Pringled).
45
+
46
+ ## Citation
47
+
48
+ Please cite the [Model2Vec repository](https://github.com/MinishLab/model2vec) if you use this model in your work.
49
+ ```
50
+ @software{minishlab2024model2vec,
51
+ author = {Stephan Tulkens and {van Dongen}, Thomas},
52
+ title = {Model2Vec: Fast State-of-the-Art Static Embeddings},
53
+ year = {2024},
54
+ publisher = {Zenodo},
55
+ doi = {10.5281/zenodo.17270888},
56
+ url = {https://github.com/MinishLab/model2vec},
57
+ license = {MIT}
58
+ }
59
+ ```
config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "normalize": true,
3
+ "embedding_dtype": "float32"
4
+ }
gate_meta.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_model": "minishlab/potion-base-32M",
3
+ "license_threshold": 0.2,
4
+ "test_license_recall": 0.9952,
5
+ "test_false_positive_rate": 0.0227,
6
+ "note": "recall-first gate; label 1 == license_related"
7
+ }
metrics.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_model": "minishlab/potion-base-32M",
3
+ "license_threshold": 0.2,
4
+ "test_license_recall": 0.9952,
5
+ "test_false_positive_rate": 0.0227,
6
+ "note": "recall-first gate; label 1 == license_related"
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36d9735bee66fdfd2ca1ad3580ae64cac7685aa118e8bdad9a8c330dd9cf21e4
3
+ size 129210456
modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": ".",
6
+ "type": "sentence_transformers.models.StaticEmbedding"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Normalize",
12
+ "type": "sentence_transformers.models.Normalize"
13
+ }
14
+ ]
pipeline.skops ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7931d6483759c0d84019fe51bb68a439fce42099820fa84c3d70015c89631951
3
+ size 7440557
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff