lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

join function TypeError fixed

Browse Source
This commit is contained in:
mizdler 2015-04-24 22:35:09 +04:30
parent aa57cce3ef
commit e172e12182
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/kaggle_otto_nn.py
View File

@ -65,7 +65,7 @@ def preprocess_labels(y, encoder=None, categorical=True):
def make_submission(y_prob, ids, encoder, fname): def make_submission(y_prob, ids, encoder, fname):
with open(fname, 'w') as f: with open(fname, 'w') as f:
f.write('id,') f.write('id,')
f.write(','.join(encoder.classes_)) f.write(','.join([str(i) for i in encoder.classes_]))
f.write('\n') f.write('\n')
for i, probs in zip(ids, y_prob): for i, probs in zip(ids, y_prob):
probas = ','.join([i] + [str(p) for p in probs.tolist()]) probas = ','.join([i] + [str(p) for p in probs.tolist()])