def test_BeagleComposite_2(): from inphosemantics import load_picklez, dump_matrix root = 'test-data/iep/plato/' corpus_filename =\ root + 'corpus/iep-plato.pickle.bz2' env_filename =\ root + 'models/iep-plato-beagleenviroment-sentences.npy' matrix_filename =\ root + 'models/iep-plato-beaglecomposite-sentences.npy' print 'Loading corpus\n'\ ' ', corpus_filename c = load_picklez(corpus_filename) print 'Loading environment model\n'\ ' ', env_filename e = BeagleEnvironment() e.load_matrix(env_filename) print e.matrix print 'Training model' m = BeagleComposite() m.train(c, env_matrix=e.matrix) print m.matrix print 'Dumping matrix to\n'\ ' ', matrix_filename m.dump_matrix(matrix_filename) return m