view embeddings/one-per-line.py @ 504:19ab9ce916e3

slightly more sophisticated system for finding the mnist data
author James Bergstra <bergstrj@iro.umontreal.ca>
date Wed, 29 Oct 2008 11:38:49 -0400
parents a07948f780b9
children
line wrap: on
line source

#!/usr/bin/python

import string
#import psyco

weightsfile = "lm-weights.txt"
vocabfile = "words.asc"
size = 30000
dimensions = 50

import numpy, math
import sys
from common.str import percent

word_to_vector = {}

f = open(weightsfile)
f.readline()
vals = [float(v) for v in string.split(f.readline())]
assert len(vals) == size * dimensions
vals.reverse()
#for i in range(size):
r = range(size)
r.reverse()
for i in r:
    l = vals[dimensions*i:dimensions*(i+1)]
    print string.join([`s` for s in l], "\t")