Upload README.md
611e26a verified - 1.52 kB initial commit
- 3.91 kB Upload README.md
- 1.5 kB DistilBERT multi-label HR classifier
- 1.75 kB Upload inference.py
- 482 Bytes Upload label_config.json
- 268 MB DistilBERT multi-label HR classifier v2
setfit_classifier.pkl Detected Pickle imports (5)
- "sklearn.linear_model._logistic.LogisticRegression",
- "numpy._core.multiarray._reconstruct",
- "numpy.ndarray",
- "sklearn.multioutput.MultiOutputClassifier",
- "numpy.dtype"
How to fix it?
67.1 kB Add setfit_classifier.pkl from SETFit training - 333 Bytes Add setfit_cv_results.json from SETFit training
setfit_encoder.pkl Detected Pickle imports (35)
- "sentence_transformers.base.modality.InputFormatter",
- "sentence_transformers.sentence_transformer.modules.normalize.Normalize",
- "sentence_transformers.sentence_transformer.modules.pooling.Pooling",
- "transformers.models.bert.modeling_bert.BertPooler",
- "torch.nn.modules.activation.Tanh",
- "sentence_transformers.sentence_transformer.model_card.SentenceTransformerModelCardData",
- "transformers.models.bert.modeling_bert.BertEncoder",
- "transformers.models.bert.tokenization_bert.BertTokenizer",
- "torch._utils._rebuild_parameter_with_state",
- "transformers.models.bert.modeling_bert.BertEmbeddings",
- "tokenizers.models.Model",
- "transformers.models.bert.modeling_bert.BertSelfOutput",
- "torch._utils._rebuild_tensor_v2",
- "transformers.activations.GELUActivation",
- "torch.nn.modules.linear.Linear",
- "transformers.models.bert.modeling_bert.BertModel",
- "torch.nn.modules.dropout.Dropout",
- "transformers.models.bert.configuration_bert.BertConfig",
- "collections.OrderedDict",
- "tokenizers.Tokenizer",
- "tokenizers.AddedToken",
- "transformers.models.bert.modeling_bert.BertLayer",
- "transformers.models.bert.modeling_bert.BertIntermediate",
- "transformers.models.bert.modeling_bert.BertSelfAttention",
- "transformers.models.bert.modeling_bert.BertOutput",
- "torch.nn.modules.normalization.LayerNorm",
- "sentence_transformers.sentence_transformer.model.SentenceTransformer",
- "torch.nn.modules.sparse.Embedding",
- "torch.float32",
- "torch.torch_version.TorchVersion",
- "transformers.models.bert.modeling_bert.BertAttention",
- "sentence_transformers.base.modules.transformer.Transformer",
- "torch.nn.modules.container.ModuleList",
- "torch._C._nn.gelu",
- "torch.storage._load_from_bytes"
How to fix it?
91.8 MB Add setfit_encoder.pkl from SETFit training - 390 Bytes Add setfit_label_config.json from SETFit training
- 711 kB DistilBERT multi-label HR classifier
- 351 Bytes DistilBERT multi-label HR classifier
training_args.bin Detected Pickle imports (10)
- "torch.device",
- "transformers.trainer_utils.IntervalStrategy",
- "transformers.trainer_utils.HubStrategy",
- "transformers.training_args.TrainingArguments",
- "transformers.trainer_utils.SchedulerType",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "transformers.training_args.OptimizerNames",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.trainer_utils.SaveStrategy",
- "accelerate.state.PartialState"
How to fix it?
5.33 kB DistilBERT multi-label HR classifier v2 - 22.8 kB Add training_script.py from SETFit training