Updated readme

ThilinaRajapakse · Apr 22, 2020 · 5aee622 · 5aee622
2 parents 8a134c7 + a2711b9
commit 5aee622
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -135,3 +135,5 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+#Stale Notebooks
+Untitled.ipynb
diff --git a/README.md b/README.md
@@ -324,7 +324,7 @@ transformers_logger = logging.getLogger("transformers")
 transformers_logger.setLevel(logging.WARNING)
 
 # Train and Evaluation data needs to be in a Pandas Dataframe containing at least two columns, a 'text' and a 'labels' column. The `labels` column should contain multi-hot encoded lists.
-train_data = [['Example sentence 1 for multilabel classification.', [1, 1, 1, 1, 0, 1]]] + [['This is another example sentence. ', [0, 1, 1, 0, 0, 0]]]
+train_data = [['Example sentence 1 for multilabel classification.', [1, 1, 1, 1, 0, 1]]], [['This is another example sentence. ', [0, 1, 1, 0, 0, 0]]]
 train_df = pd.DataFrame(train_data, columns=['text', 'labels'])
 
 eval_data = [['Example eval sentence for multilabel classification.', [1, 1, 1, 1, 0, 1]], ['Example eval senntence belonging to class 2', [0, 1, 1, 0, 0, 0]]]