zhibinlu
/

vgcn-bert-distilbert-base-uncased

Feature Extraction

Model card Files Files and versions Community

zhibinlu commited on Jun 9, 2023

Commit

cbf28bd

·

1 Parent(s): a308509

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ import transformers as tfr
 from transformers.models.vgcn_bert.modeling_graph import WordGraph
 tokenizer = tfr.AutoTokenizer.from_pretrained(
-    "zhibinlu/distilbert-base-uncased"
 )
 # 1st method: Build graph using NPMI statistical method from training corpus
 wgraph = WordGraph(rows=train_valid_df["text"], tokenizer=tokenizer)

 from transformers.models.vgcn_bert.modeling_graph import WordGraph
 tokenizer = tfr.AutoTokenizer.from_pretrained(
+    "distilbert-base-uncased"
 )
 # 1st method: Build graph using NPMI statistical method from training corpus
 wgraph = WordGraph(rows=train_valid_df["text"], tokenizer=tokenizer)