Hanish2007/brand-safety-model

By Hanish2007 · ↓ 8 · ♥ 0 · Updated 2025-02-04

What's in this repo

This page lists the files in Hanish2007/brand-safety-model, including model.onnx.

1 ONNX models (.onnx files)
120 files, including 54 .json, 20 .pt, 11 .safetensors, 11 .bin, 11 .txt, 10 .pth
Last updated 2025-02-04

ONNX files

File	Size	Download
onnx/model.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

library_name: transformers license: apache-2.0 base_model: distilbert/distilbert-base-uncased tags:

generated_from_trainer metrics:
accuracy model-index:
name: brand-safety-model results: []

brand-safety-model

This model is a fine-tuned version of distilbert/distilbert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:

Loss: 0.4779
Accuracy: 0.8657

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

learning_rate: 2e-05
train_batch_size: 64
eval_batch_size: 64
seed: 42
optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: linear
num_epochs: 15

Training results

Training Loss	Epoch	Step	Validation Loss	Accuracy
1.581	1.0	112	1.4547	0.6798
0.9156	2.0	224	0.8586	0.7904
0.6402	3.0	336	0.6679	0.8292
0.5405	4.0	448	0.5595	0.8551
0.4553	5.0	560	0.5183	0.8534
0.3509	6.0	672	0.4825	0.8629
0.316	7.0	784	0.4786	0.8584
0.2506	8.0	896	0.4710	0.8618
0.2049	9.0	1008	0.4912	0.8567
0.1416	10.0	1120	0.4881	0.8567
0.1572	11.0	1232	0.4779	0.8657
0.1407	12.0	1344	0.4886	0.8596
0.132	13.0	1456	0.4933	0.8618
0.1319	14.0	1568	0.4807	0.8635
0.1277	15.0	1680	0.4846	0.8618

Framework versions

Transformers 4.46.3
Pytorch 2.4.0
Datasets 2.20.0
Tokenizers 0.20.3

Model info

text-classification transformers

transformers onnx safetensors distilbert text-classification generated_from_trainer base_model:distilbert/distilbert-base-uncased base_model:quantized:distilbert/distilbert-base-uncased license:apache-2.0 text-embeddings-inference endpoints_compatible region:us

↓ 8 ♥ 0

Created: 2025-01-27

Updated: 2025-02-04

View on Hugging Face

Files (120)

.gitattributes

README.md

checkpoint-130/config.json

checkpoint-130/model.safetensors

checkpoint-130/optimizer.pt

checkpoint-130/rng_state.pth

checkpoint-130/scheduler.pt

checkpoint-130/special_tokens_map.json

checkpoint-130/tokenizer.json

checkpoint-130/tokenizer_config.json

checkpoint-130/trainer_state.json

checkpoint-130/training_args.bin

checkpoint-130/vocab.txt

checkpoint-195/config.json

checkpoint-195/model.safetensors

checkpoint-195/optimizer.pt

checkpoint-195/rng_state.pth

checkpoint-195/scheduler.pt

checkpoint-195/special_tokens_map.json

checkpoint-195/tokenizer.json

checkpoint-195/tokenizer_config.json

checkpoint-195/trainer_state.json

checkpoint-195/training_args.bin

checkpoint-195/vocab.txt

checkpoint-260/config.json

checkpoint-260/model.safetensors

checkpoint-260/optimizer.pt

checkpoint-260/rng_state.pth

checkpoint-260/scheduler.pt

checkpoint-260/special_tokens_map.json

checkpoint-260/tokenizer.json

checkpoint-260/tokenizer_config.json

checkpoint-260/trainer_state.json

checkpoint-260/training_args.bin

checkpoint-260/vocab.txt

checkpoint-325/config.json

checkpoint-325/model.safetensors

checkpoint-325/optimizer.pt

checkpoint-325/rng_state.pth

checkpoint-325/scheduler.pt

checkpoint-325/special_tokens_map.json

checkpoint-325/tokenizer.json

checkpoint-325/tokenizer_config.json

checkpoint-325/trainer_state.json

checkpoint-325/training_args.bin

checkpoint-325/vocab.txt

checkpoint-390/config.json

checkpoint-390/model.safetensors

checkpoint-390/optimizer.pt

checkpoint-390/rng_state.pth

checkpoint-390/scheduler.pt

checkpoint-390/special_tokens_map.json

checkpoint-390/tokenizer.json

checkpoint-390/tokenizer_config.json

checkpoint-390/trainer_state.json

checkpoint-390/training_args.bin

checkpoint-390/vocab.txt

checkpoint-455/config.json

checkpoint-455/model.safetensors

checkpoint-455/optimizer.pt

checkpoint-455/rng_state.pth

checkpoint-455/scheduler.pt

checkpoint-455/special_tokens_map.json

checkpoint-455/tokenizer.json

checkpoint-455/tokenizer_config.json

checkpoint-455/trainer_state.json

checkpoint-455/training_args.bin

checkpoint-455/vocab.txt

checkpoint-520/config.json

checkpoint-520/model.safetensors

checkpoint-520/optimizer.pt

checkpoint-520/rng_state.pth

checkpoint-520/scheduler.pt

checkpoint-520/special_tokens_map.json

checkpoint-520/tokenizer.json

checkpoint-520/tokenizer_config.json

checkpoint-520/trainer_state.json

checkpoint-520/training_args.bin

checkpoint-520/vocab.txt

checkpoint-585/config.json

checkpoint-585/model.safetensors

checkpoint-585/optimizer.pt

checkpoint-585/rng_state.pth

checkpoint-585/scheduler.pt

checkpoint-585/special_tokens_map.json

checkpoint-585/tokenizer.json

checkpoint-585/tokenizer_config.json

checkpoint-585/trainer_state.json

checkpoint-585/training_args.bin

checkpoint-585/vocab.txt

checkpoint-65/config.json

checkpoint-65/model.safetensors

checkpoint-65/optimizer.pt

checkpoint-65/rng_state.pth

checkpoint-65/scheduler.pt

checkpoint-65/special_tokens_map.json

checkpoint-65/tokenizer.json

checkpoint-65/tokenizer_config.json

checkpoint-65/trainer_state.json

checkpoint-65/training_args.bin

checkpoint-65/vocab.txt

checkpoint-650/config.json

checkpoint-650/model.safetensors

checkpoint-650/optimizer.pt

checkpoint-650/rng_state.pth

checkpoint-650/scheduler.pt

checkpoint-650/special_tokens_map.json

checkpoint-650/tokenizer.json

checkpoint-650/tokenizer_config.json

checkpoint-650/trainer_state.json

checkpoint-650/training_args.bin

checkpoint-650/vocab.txt

config.json

model.safetensors

onnx/model.onnx ONNX

special_tokens_map.json

tokenizer.json

tokenizer_config.json

training_args.bin

vocab.txt

Related models

More text-classification models

Browse all text-classification models

Hanish2007/brand-safety-model

What's in this repo

ONNX files

How to use

README

brand-safety-model

Model description

Intended uses & limitations

Training and evaluation data

Training procedure

Training hyperparameters

Training results

Framework versions

Model info

Files (120)

Related models

More from Hanish2007

More text-classification models

Popular models