How to use ketut/IndoBERTkbli with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="ketut/IndoBERTkbli")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("ketut/IndoBERTkbli") model = AutoModelForSequenceClassification.from_pretrained("ketut/IndoBERTkbli")
Model klasifikasi kode kelompok KBLI (5 digit).
++untuk label encoder
import joblib
le = joblib.load("label_encoder_base_p2_augmented.pkl")
Files info
Base model
