Text Classification
Transformers
TensorBoard
Safetensors
bert
Generated from Trainer
text-embeddings-inference
Instructions to use vat75/PhishGuard-AI with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use vat75/PhishGuard-AI with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="vat75/PhishGuard-AI")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("vat75/PhishGuard-AI") model = AutoModelForSequenceClassification.from_pretrained("vat75/PhishGuard-AI") - Notebooks
- Google Colab
- Kaggle
Training in progress, epoch 1
Browse files- config.json +1 -0
- model.safetensors +1 -1
- training_args.bin +1 -1
config.json
CHANGED
|
@@ -28,6 +28,7 @@
|
|
| 28 |
"pooler_size_per_head": 128,
|
| 29 |
"pooler_type": "first_token_transform",
|
| 30 |
"position_embedding_type": "absolute",
|
|
|
|
| 31 |
"tie_word_embeddings": true,
|
| 32 |
"transformers_version": "5.0.0",
|
| 33 |
"type_vocab_size": 2,
|
|
|
|
| 28 |
"pooler_size_per_head": 128,
|
| 29 |
"pooler_type": "first_token_transform",
|
| 30 |
"position_embedding_type": "absolute",
|
| 31 |
+
"problem_type": "single_label_classification",
|
| 32 |
"tie_word_embeddings": true,
|
| 33 |
"transformers_version": "5.0.0",
|
| 34 |
"type_vocab_size": 2,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 651395048
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:92bfe4840f5d89368bde9bca158e2451abd8ecdd975df5c88ac3d98428e863c9
|
| 3 |
size 651395048
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5265
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d22be77fac8977dd45219b1d7e209515b83fcb8ca6a7f25ae57b930bb4df4213
|
| 3 |
size 5265
|