Commit aa220d66 authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

fixed y shape in triplets

parent 6c6ae475
Pipeline #2704 passed with stages
in 19 minutes and 54 seconds
......@@ -231,7 +231,7 @@ class NeighborTripletTraintest(object):
fh.create_dataset('t_%s' % split_name,
data=split_triplets)
fh.create_dataset('y_%s' % split_name,
data=np.zeros((len(split_triplets), 1)))
data=np.zeros((len(split_triplets),)))
NeighborTripletTraintest.__log.info(
'NeighborTripletTraintest saved to %s', out_file)
'''
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment