svdspace.py 文件源码

def main():
    parser = argparse.ArgumentParser('Transforms the space by applying the SVD')
    parser.add_argument('--input', '-i', help='Input file')
    parser.add_argument('--output', '-o', help='Output file')
    args = parser.parse_args()

    space = load_mikolov_text(args.input)
    U, s, Vh = svd(space.matrix, full_matrices=False)
    transformed = U.dot(np.diag(s))

    newspace = VectorSpace(transformed, space.vocab)
    newspace.save_mikolov_text(args.output)