From e172e12182868f70d39ff27b307663d6fe9422e1 Mon Sep 17 00:00:00 2001 From: mizdler Date: Fri, 24 Apr 2015 22:35:09 +0430 Subject: [PATCH] join function TypeError fixed --- examples/kaggle_otto_nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/kaggle_otto_nn.py b/examples/kaggle_otto_nn.py index 120041b35..f1b64cfb0 100644 --- a/examples/kaggle_otto_nn.py +++ b/examples/kaggle_otto_nn.py @@ -65,7 +65,7 @@ def preprocess_labels(y, encoder=None, categorical=True): def make_submission(y_prob, ids, encoder, fname): with open(fname, 'w') as f: f.write('id,') - f.write(','.join(encoder.classes_)) + f.write(','.join([str(i) for i in encoder.classes_])) f.write('\n') for i, probs in zip(ids, y_prob): probas = ','.join([i] + [str(p) for p in probs.tolist()])