読書メモ〜『情報検索の基礎』 第1章 1

boolean_retrieval_model.py

# p.3、論理検索モデル(Boolean retrieval model)

import glob
import os

DATA_PATH = '../data/Boolean_Retrieval/shakespeare_collection'

DOCUMENTS = [
    'Tragedies/Antony and Cleopatra',
    'Tragedies/Julius Caesar',
    'Comedies/The Tempest',
    'Tragedies/Hamlet',
    'Tragedies/Othello',
    'Tragedies/Macbeth'
]

# 用語と文章の関係(term-document)からなる結合行列をつくる
matrix = utils.create_incidence_matrix(
    [os.path.join(DATA_PATH, d) for d in DOCUMENTS])