# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("SkyAsl/Bert-base-Comment_classifier")
model = AutoModelForSequenceClassification.from_pretrained("SkyAsl/Bert-base-Comment_classifier")Quick Links
Model Card for Model ID
Training Details
Training Data
https://huggingface.co/datasets/stanfordnlp/imdb
Training Hyperparameters
-learning_rate = 2e-4 -batch_size = 128 -num_epochs = 3 -weight_decay = 0.01
Evaluation
---Training_loss: 0.355100 ---Validation_loss: 0.330199 ---Accuracy: 0.857 ---Roc_auc: 0.936
- Downloads last month
- 4
Model tree for SkyAsl/Bert-base-Comment_classifier
Base model
google-bert/bert-base-uncased
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="SkyAsl/Bert-base-Comment_classifier")