Skip to content

Commit

Permalink
Update ParallelSentencesDataset.py
Browse files Browse the repository at this point in the history
  • Loading branch information
nreimers authored May 28, 2021
1 parent f9ee270 commit abf3639
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sentence_transformers/datasets/ParallelSentencesDataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ def load_data(self, filepath: str, weight: int = 100, max_sentences: int = None,
Reads in a tab-seperated .txt/.csv/.tsv or .gz file. The different columns contain the different translations of the sentence in the first column
:param filepath: Filepath to the file
:param weight: If more that one dataset is loaded with load_data: With which frequency should data be sampled from this dataset?
:param weight: If more than one dataset is loaded with load_data: With which frequency should data be sampled from this dataset?
:param max_sentences: Max number of lines to be read from filepath
:param max_sentence_length: Skip the example if one of the sentences is has more characters than max_sentence_length
:param batch_size: Size for encoding parallel sentences
Expand Down

0 comments on commit abf3639

Please sign in to comment.