[catboost/quality_benchmarks] fix target column parsing
Note: mandatory check (NEED_CHECK) was skipped
@@ -44,7 +44,7 @@ class Experiment(object):
def read_data(self):
cols = pd.read_csv(self.cd_path, sep='\t', header=None)
- target_col = np.where(cols[1] == 'Target')[0][0]
+ target_col = cols[0][np.where(cols[1] == 'Target')[0][0]]
cat_cols = cols[cols[1] == "Categ"][0].values
X_train, y_train = self.read_file(self.train_path, target_col)
