Fixed code review comments on return types
@@ -7,9 +7,7 @@ import numpy as np
import pandas as pd
from scenarios.sentence_similarity.gensen_train import train
-from utils_nlp.gensen.create_gensen_model import (
- create_multiseq2seq_model,
-)
+from utils_nlp.gensen.create_gensen_model import create_multiseq2seq_model
from utils_nlp.gensen.gensen import GenSenSingle
from utils_nlp.gensen.preprocess_utils import gensen_preprocess
@@ -72,6 +70,7 @@ class GenSenClassifier:
test set.
Returns:
+ str: Path to the folder containing all preprocessed token files.
"""
return gensen_preprocess(train_df, dev_df, test_df, self.cache_dir)
@@ -83,7 +82,8 @@ class GenSenClassifier:
Args:
config_file: Path to the config file.
- Returns(dict): The loaded json file as python object
+ Returns
+ dict: The loaded json file as python object
"""
json_object = json.load(open(config_file, "r", encoding="utf-8"))
@@ -131,8 +131,9 @@ class GenSenClassifier:
Args:
sentences(list) : List of sentences.
- Returns(array): A pairwise cosine similarity for the sentences provided based
- on their gensen vector representations.
+ Returns
+ array: A pairwise cosine similarity for the sentences provided based on their gensen
+ vector representations.
"""
