Python numpy_to_ivec_refの例

コード例 #1

ファイル: jsgd.py プロジェクト: 2021rahul/Bachelor_Thesis_Project

def _numpy_to_xmatrix(m):
  xm = jsgd_wrap.x_matrix_t()
  xm.n, xm.d = m.shape
  if not hasattr(m, 'indptr'): # dense
    _check_row_float32(m)  
    xm.encoding = jsgd_wrap.JSGD_X_FULL
    xm.data = jsgd_wrap.numpy_to_fvec_ref(m)
  else:     # sparse    
    if m.format != 'csr': 
      raise TypeError('expected sparse CSR matrix')
    xm.encoding = jsgd_wrap.JSGD_X_SPARSE
    xm.sparse_data = jsgd_wrap.numpy_to_fvec_ref(m.data)
    xm.indptr = jsgd_wrap.numpy_to_ivec_ref(m.indptr)
    xm.indices = jsgd_wrap.numpy_to_ivec_ref(m.indices)
  return xm

コード例 #2

ファイル: jsgd.py プロジェクト: plsang/opensource

def _numpy_to_xmatrix(m):
  xm = jsgd_wrap.x_matrix_t()
  xm.n, xm.d = m.shape
  if not hasattr(m, 'indptr'): # dense
    _check_row_float32(m)  
    xm.encoding = jsgd_wrap.JSGD_X_FULL
    xm.data = jsgd_wrap.numpy_to_fvec_ref(m)
  else:     # sparse    
    if m.format != 'csr': 
      raise TypeError('expected sparse CSR matrix')
    xm.encoding = jsgd_wrap.JSGD_X_SPARSE
    xm.sparse_data = jsgd_wrap.numpy_to_fvec_ref(m.data)
    xm.indptr = jsgd_wrap.numpy_to_ivec_ref(m.indptr)
    xm.indices = jsgd_wrap.numpy_to_ivec_ref(m.indices)
  return xm

コード例 #3

ファイル: jsgd.py プロジェクト: 2021rahul/Bachelor_Thesis_Project

def _get_ptr(m):
  if m.dtype == numpy.int32: return jsgd_wrap.numpy_to_ivec_ref(m)
  if m.dtype == numpy.float32: return jsgd_wrap.numpy_to_fvec_ref(m)
  if m.dtype == numpy.float64: return jsgd_wrap.numpy_to_dvec_ref(m)
  raise TypeError('type %s not handled' % m.dtype)

コード例 #4

ファイル: jsgd.py プロジェクト: plsang/opensource

def _get_ptr(m):
  if m.dtype == numpy.int32: return jsgd_wrap.numpy_to_ivec_ref(m)
  if m.dtype == numpy.float32: return jsgd_wrap.numpy_to_fvec_ref(m)
  if m.dtype == numpy.float64: return jsgd_wrap.numpy_to_dvec_ref(m)
  raise TypeError('type %s not handled' % m.dtype)