diff --git a/ds/trec09mq/refiners.qtypes.py b/ds/trec09mq/refiners.qtypes.py index 37f12a6..df6fe41 100644 --- a/ds/trec09mq/refiners.qtypes.py +++ b/ds/trec09mq/refiners.qtypes.py @@ -5,7 +5,7 @@ def get_refiners_by_qtypes(all, dataset, qtypes, output): - refiners = pd.read_csv(all, index_col=0, nrows=0).columns[4::3] + refiners = pd.read_csv(all, index_col=0, nrows=0).columns[3::3] q_ = pd.read_csv(dataset, sep=',') qtypes = pd.read_csv(qtypes, sep='\t', usecols=['Topic', 'Class']) qtypes['Topic'] = qtypes['Topic'].astype(int)