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

Before Change

    call_fh = open(args.call_fn, "w")
    predictStart = time.time()
    predictBatchSize = param.predictBatchSize
    for i in range(0, total, predictBatchSize):
        XBatch, _, _ = utils.DecompressArray(XArrayCompressed, i, predictBatchSize, total)
        posBatch, _, _ = utils.DecompressArray(posArrayCompressed, i, predictBatchSize, total)
        base, t = m.predict(XBatch)
        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))


if __name__ == "__main__":

After Change

    predictStart = time.time()
    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