module.py 文件源码

def _predict_proba(mod, val):
    mod.incache = True

    val.reset()
    true = val.gesture.copy()
    segment = val.segment.copy()
    val.reset()
    assert np.all(true == val.gesture.copy())
    assert np.all(segment == val.segment.copy())

    out = mod.predict(val).asnumpy()
    #  assert_equal(out.ndim, 2)
    #  assert_equal(out.shape[1], mod.num_gesture)
    #  pred = out.argmax(axis=1)
    if mod.lstm:
        cut = lambda a: np.hstack(a[begin:end + mod.num_channel - mod.data_shape_1]
                                  for begin, end in utils.continuous_segments(segment))
        true = cut(true)
        segment = cut(segment)
    #  assert_equal(true.shape, pred.shape)
    return out, true, segment