absa-indobert-web / config.py
zdannn2808's picture
merapikan file, menambahkan komentar penting pada config.py, preprocessing.py, visualization.py, dan app.py
9e73d6d verified
raw
history blame contribute delete
649 Bytes
"""
config.py
---------
Berisi konfigurasi utama untuk model ABSA (Aspect-Based Sentiment Analysis)
menggunakan IndoBERT, termasuk parameter model dan daftar kolom aspek.
"""
CONFIG = {
"model_name": "indobenchmark/indobert-base-p1",
"dropout_rate": 0.3,
"max_len": 128,
"hf_model_repo": "zdannn2808/absa_indobert",
"hf_model_subfolder": "Indobert_Model/model",
"hf_tokenizer_subfolder": "Indobert_Model/tokenizer",
"batch_size": 32, # Ukuran batch untuk prediksi
"num_workers": 0, # Jumlah worker untuk DataLoader (0 = main process only)
}
ASPEK_COLUMNS = ["tugas", "pengajaran", "materi", "metode", "interaksi"]