0489238f5b5db28a6a6d3bcb13406f20e9749d79,clairvoyante/predict.py,,Test,#Any#Any#,23

Before Change

def Test(args, m):
    logging.info("Loading the dataset ...")
    total, XArrayCompressed, posArrayCompressed = \
    utils.GetAlnArray(args.tensor_fn)

    logging.info("Predicing the dataset ...")

After Change

    for num, XBatch, posBatch in utils.GetTensor( args.tensor_fn, param.predictBatchSize ):
        base, t = m.predict(XBatch)
        if num != len(base):
            sys.exit("Inconsistent shape between input tensor and output predictions %d/%d" % (num, len(base)))
        for j in range(len(base)):
            print >> call_fh, posBatch[j], np.argmax(base[j]), np.argmax(t[j])
    logging.info("Prediciton time elapsed: %.2f s" % (time.time() - predictStart))

Instances