@ -15,6 +15,7 @@ import scipy as sc
import tensorflow as tf
import _pickle as cPickle
import hickle as hkl
@ -132,21 +133,21 @@ class FASTsearch(object):
uiOZ = uiOZ . transpose ( )
sess = tf . Session ( )
with tf . device ( ' /gpu:0 ' ) :
with sess . as_default ( ) :
with sess . as_default ( ) :
uiOZ_tensor = tf . constant ( uiOZ )
uiOZ_tensor = tf . constant ( uiOZ )
dbOZ_tensor_sparse = convert_sparse_matrix_to_sparse_tensor ( self . dbOZ )
#uiOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(uiOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#dbOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(dbOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#wordCountDoku = tf.matmul(uiOZ_tensor, dbOZ_tensor)
wordCountDoku = tf . sparse_tensor_dense_matmul ( dbOZ_tensor_sparse , uiOZ_tensor )
wCD = np . array ( wordCountDoku . eval ( ) )
dbOZ_tensor_sparse = convert_sparse_matrix_to_sparse_tensor ( self . dbOZ )
#uiOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(uiOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#dbOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(dbOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#wordCountDoku = tf.matmul(uiOZ_tensor, dbOZ_tensor)
wordCountDoku = tf . sparse_tensor_dense_matmul ( dbOZ_tensor_sparse , uiOZ_tensor )
wCD = np . array ( wordCountDoku . eval ( ) )
indexedwCD = [ ]
for n in range ( len ( wCD ) ) :
@ -206,21 +207,21 @@ class FASTsearch(object):
uiOZ = uiOZ . transpose ( )
sess = tf . Session ( )
with tf . device ( ' /gpu:0 ' ) :
with sess . as_default ( ) :
with sess . as_default ( ) :
uiOZ_tensor = tf . constant ( uiOZ )
uiOZ_tensor = tf . constant ( uiOZ )
dbOZ_tensor_sparse = convert_sparse_matrix_to_sparse_tensor ( self . dbOZ )
#uiOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(uiOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#dbOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(dbOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#wordCountDoku = tf.matmul(uiOZ_tensor, dbOZ_tensor)
wordCountDoku = tf . sparse_tensor_dense_matmul ( dbOZ_tensor_sparse , uiOZ_tensor )
wCD = np . array ( wordCountDoku . eval ( ) )
dbOZ_tensor_sparse = convert_sparse_matrix_to_sparse_tensor ( self . dbOZ )
#uiOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(uiOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#dbOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(dbOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#wordCountDoku = tf.matmul(uiOZ_tensor, dbOZ_tensor)
wordCountDoku = tf . sparse_tensor_dense_matmul ( dbOZ_tensor_sparse , uiOZ_tensor )
wCD = np . array ( wordCountDoku . eval ( ) )
indexedwCD = [ ]
for n in range ( len ( wCD ) ) :
@ -257,21 +258,21 @@ class FASTsearch(object):
uiOZ = uiOZ . transpose ( )
sess = tf . Session ( )
with tf . device ( ' /gpu:0 ' ) :
with sess . as_default ( ) :
with sess . as_default ( ) :
uiOZ_tensor = tf . constant ( uiOZ )
uiOZ_tensor = tf . constant ( uiOZ )
dbOZ_tensor_sparse = convert_sparse_matrix_to_sparse_tensor ( self . dbOZ )
#uiOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(uiOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#dbOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(dbOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#wordCountDoku = tf.matmul(uiOZ_tensor, dbOZ_tensor)
wordCountDoku = tf . sparse_tensor_dense_matmul ( dbOZ_tensor_sparse , uiOZ_tensor )
wCD = np . array ( wordCountDoku . eval ( ) )
dbOZ_tensor_sparse = convert_sparse_matrix_to_sparse_tensor ( self . dbOZ )
#uiOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(uiOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#dbOZ_tensor_sparse =tf.contrib.layers.dense_to_sparse(dbOZ_tensor, eos_token=0, outputs_collections=None, scope=None )
#wordCountDoku = tf.matmul(uiOZ_tensor, dbOZ_tensor)
wordCountDoku = tf . sparse_tensor_dense_matmul ( dbOZ_tensor_sparse , uiOZ_tensor )
wCD = np . array ( wordCountDoku . eval ( ) )
indexedwCD = [ ]
for n in range ( len ( wCD ) ) :