Aartik
/

distilbert-base-uncased-yelp-new-aarti

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

distilbert-base-uncased-yelp-new-aarti

1 contributor

History: 2 commits

Aartik's picture

Yelp review classification

a31f47f verified 5 days ago

.DS_Store

10.2 kB

Yelp review classification 5 days ago
.gitattributes

1.52 kB

initial commit 5 days ago
README.md

1.16 kB

Yelp review classification 5 days ago
config.json

845 Bytes

Yelp review classification 5 days ago
model.safetensors
268 MB
LFS

Yelp review classification 5 days ago
training_args.bin
Detected Pickle imports (9)
- "transformers.trainer_utils.HubStrategy",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.OptimizerNames",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "transformers.training_args.TrainingArguments",
- "transformers.trainer_utils.IntervalStrategy",
- "accelerate.state.PartialState",
- "transformers.trainer_utils.SchedulerType",
- "torch.device"
How to fix it?
5.05 kB
LFS

Yelp review classification 5 days ago