Hugging Face's logo

Spaces:
jesse-tong
/
vietnamese_hate_speech_detection
Sleeping

App Files Community
jesse-tong commited on
Commit
0227fba
·
1 Parent(s): 24f7930

Fix an error

Browse files
Files changed (1) hide show
  1. distill_bert_to_lstm.py +1 -1
distill_bert_to_lstm.py CHANGED
@@ -85,7 +85,7 @@ def main():
85
  num_categories = len(args.label_column) if isinstance(args.label_column, list) else 1
86
 
87
  train_data, _, _ = load_data(
88
- args.data_path,
89
  text_col=args.text_column,
90
  label_col=label_column,
91
  validation_split=args.val_split,
 
85
  num_categories = len(args.label_column) if isinstance(args.label_column, list) else 1
86
 
87
  train_data, _, _ = load_data(
88
+ args.train_data_path,
89
  text_col=args.text_column,
90
  label_col=label_column,
91
  validation_split=args.val_split,