search

yukihito-jokyu / RAG-1

0 stars 0 forks source link

NormalSearchクラスを使用する際、検証かそうでないか指定できるようにする #14

Closed yukihito-jokyu closed 1 month ago

yukihito-jokyu commented 1 month ago

NormalSearchの_setupメソッドで使用しているget_text関数(utils.py内で定義)の中身は以下の通りになっている・

def get_text() -> List[str]:
    """
    説明
    ----------
    datasetから文章を取得し、リストとして返す。

    Returns
    ----------
    List[str]
        各txtファイルから取り出した文章をリストとして保管したもの

    """

    doc_list = []

    for i in range(1, 8):
        with open(f"dataset/novels/{i}.txt", "r") as file:
            document = file.read()
        doc_list.append(document)

    return doc_list

これは、test用のtxtファイルを参照するコードになっているので、検証用のtxtファイルを読み込むことができるように、引数でしてするコードを書くようにしたい。

yukihito-jokyu commented 1 month ago

実装したので閉じる