/* Generated by Cython 0.29.36 */
/* BEGIN: Cython Metadata
{
"distutils": {
"depends": [
"gensim/models/voidptr.h"
],
"extra_compile_args": [
"-std=c++11"
],
"extra_link_args": [
"-std=c++11"
],
"include_dirs": [
"gensim/models",
"./gensim/models"
],
"language": "c++",
"name": "gensim.models.doc2vec_inner",
"sources": [
"gensim/models/doc2vec_inner.pyx"
]
},
"module_name": "gensim.models.doc2vec_inner"
}
END: Cython Metadata */
#ifndef PY_SSIZE_T_CLEAN
#define PY_SSIZE_T_CLEAN
#endif /* PY_SSIZE_T_CLEAN */
#include "Python.h"
#ifndef Py_PYTHON_H
#error Python headers needed to compile C extensions, please install development version of Python.
#elif PY_VERSION_HEX < 0x02060000 || (0x03000000 <= PY_VERSION_HEX && PY_VERSION_HEX < 0x03030000)
#error Cython requires Python 2.6+ or Python 3.3+.
#else
#define CYTHON_ABI "0_29_36"
#define CYTHON_HEX_VERSION 0x001D24F0
#define CYTHON_FUTURE_DIVISION 1
#include
#ifndef offsetof
#define offsetof(type, member) ( (size_t) & ((type*)0) -> member )
#endif
#if !defined(WIN32) && !defined(MS_WINDOWS)
#ifndef __stdcall
#define __stdcall
#endif
#ifndef __cdecl
#define __cdecl
#endif
#ifndef __fastcall
#define __fastcall
#endif
#endif
#ifndef DL_IMPORT
#define DL_IMPORT(t) t
#endif
#ifndef DL_EXPORT
#define DL_EXPORT(t) t
#endif
#define __PYX_COMMA ,
#ifndef HAVE_LONG_LONG
#if PY_VERSION_HEX >= 0x02070000
#define HAVE_LONG_LONG
#endif
#endif
#ifndef PY_LONG_LONG
#define PY_LONG_LONG LONG_LONG
#endif
#ifndef Py_HUGE_VAL
#define Py_HUGE_VAL HUGE_VAL
#endif
#ifdef PYPY_VERSION
#define CYTHON_COMPILING_IN_PYPY 1
#define CYTHON_COMPILING_IN_PYSTON 0
#define CYTHON_COMPILING_IN_CPYTHON 0
#define CYTHON_COMPILING_IN_NOGIL 0
#undef CYTHON_USE_TYPE_SLOTS
#define CYTHON_USE_TYPE_SLOTS 0
#undef CYTHON_USE_PYTYPE_LOOKUP
#define CYTHON_USE_PYTYPE_LOOKUP 0
#if PY_VERSION_HEX < 0x03050000
#undef CYTHON_USE_ASYNC_SLOTS
#define CYTHON_USE_ASYNC_SLOTS 0
#elif !defined(CYTHON_USE_ASYNC_SLOTS)
#define CYTHON_USE_ASYNC_SLOTS 1
#endif
#undef CYTHON_USE_PYLIST_INTERNALS
#define CYTHON_USE_PYLIST_INTERNALS 0
#undef CYTHON_USE_UNICODE_INTERNALS
#define CYTHON_USE_UNICODE_INTERNALS 0
#undef CYTHON_USE_UNICODE_WRITER
#define CYTHON_USE_UNICODE_WRITER 0
#undef CYTHON_USE_PYLONG_INTERNALS
#define CYTHON_USE_PYLONG_INTERNALS 0
#undef CYTHON_AVOID_BORROWED_REFS
#define CYTHON_AVOID_BORROWED_REFS 1
#undef CYTHON_ASSUME_SAFE_MACROS
#define CYTHON_ASSUME_SAFE_MACROS 0
#undef CYTHON_UNPACK_METHODS
#define CYTHON_UNPACK_METHODS 0
#undef CYTHON_FAST_THREAD_STATE
#define CYTHON_FAST_THREAD_STATE 0
#undef CYTHON_FAST_PYCALL
#define CYTHON_FAST_PYCALL 0
#if PY_VERSION_HEX < 0x03090000
#undef CYTHON_PEP489_MULTI_PHASE_INIT
#define CYTHON_PEP489_MULTI_PHASE_INIT 0
#elif !defined(CYTHON_PEP489_MULTI_PHASE_INIT)
#define CYTHON_PEP489_MULTI_PHASE_INIT 1
#endif
#undef CYTHON_USE_TP_FINALIZE
#define CYTHON_USE_TP_FINALIZE (PY_VERSION_HEX >= 0x030400a1 && PYPY_VERSION_NUM >= 0x07030C00)
#undef CYTHON_USE_DICT_VERSIONS
#define CYTHON_USE_DICT_VERSIONS 0
#undef CYTHON_USE_EXC_INFO_STACK
#define CYTHON_USE_EXC_INFO_STACK 0
#ifndef CYTHON_UPDATE_DESCRIPTOR_DOC
#define CYTHON_UPDATE_DESCRIPTOR_DOC 0
#endif
#elif defined(PYSTON_VERSION)
#define CYTHON_COMPILING_IN_PYPY 0
#define CYTHON_COMPILING_IN_PYSTON 1
#define CYTHON_COMPILING_IN_CPYTHON 0
#define CYTHON_COMPILING_IN_NOGIL 0
#ifndef CYTHON_USE_TYPE_SLOTS
#define CYTHON_USE_TYPE_SLOTS 1
#endif
#undef CYTHON_USE_PYTYPE_LOOKUP
#define CYTHON_USE_PYTYPE_LOOKUP 0
#undef CYTHON_USE_ASYNC_SLOTS
#define CYTHON_USE_ASYNC_SLOTS 0
#undef CYTHON_USE_PYLIST_INTERNALS
#define CYTHON_USE_PYLIST_INTERNALS 0
#ifndef CYTHON_USE_UNICODE_INTERNALS
#define CYTHON_USE_UNICODE_INTERNALS 1
#endif
#undef CYTHON_USE_UNICODE_WRITER
#define CYTHON_USE_UNICODE_WRITER 0
#undef CYTHON_USE_PYLONG_INTERNALS
#define CYTHON_USE_PYLONG_INTERNALS 0
#ifndef CYTHON_AVOID_BORROWED_REFS
#define CYTHON_AVOID_BORROWED_REFS 0
#endif
#ifndef CYTHON_ASSUME_SAFE_MACROS
#define CYTHON_ASSUME_SAFE_MACROS 1
#endif
#ifndef CYTHON_UNPACK_METHODS
#define CYTHON_UNPACK_METHODS 1
#endif
#undef CYTHON_FAST_THREAD_STATE
#define CYTHON_FAST_THREAD_STATE 0
#undef CYTHON_FAST_PYCALL
#define CYTHON_FAST_PYCALL 0
#undef CYTHON_PEP489_MULTI_PHASE_INIT
#define CYTHON_PEP489_MULTI_PHASE_INIT 0
#undef CYTHON_USE_TP_FINALIZE
#define CYTHON_USE_TP_FINALIZE 0
#undef CYTHON_USE_DICT_VERSIONS
#define CYTHON_USE_DICT_VERSIONS 0
#undef CYTHON_USE_EXC_INFO_STACK
#define CYTHON_USE_EXC_INFO_STACK 0
#ifndef CYTHON_UPDATE_DESCRIPTOR_DOC
#define CYTHON_UPDATE_DESCRIPTOR_DOC 0
#endif
#elif defined(PY_NOGIL)
#define CYTHON_COMPILING_IN_PYPY 0
#define CYTHON_COMPILING_IN_PYSTON 0
#define CYTHON_COMPILING_IN_CPYTHON 0
#define CYTHON_COMPILING_IN_NOGIL 1
#ifndef CYTHON_USE_TYPE_SLOTS
#define CYTHON_USE_TYPE_SLOTS 1
#endif
#undef CYTHON_USE_PYTYPE_LOOKUP
#define CYTHON_USE_PYTYPE_LOOKUP 0
#ifndef CYTHON_USE_ASYNC_SLOTS
#define CYTHON_USE_ASYNC_SLOTS 1
#endif
#undef CYTHON_USE_PYLIST_INTERNALS
#define CYTHON_USE_PYLIST_INTERNALS 0
#ifndef CYTHON_USE_UNICODE_INTERNALS
#define CYTHON_USE_UNICODE_INTERNALS 1
#endif
#undef CYTHON_USE_UNICODE_WRITER
#define CYTHON_USE_UNICODE_WRITER 0
#undef CYTHON_USE_PYLONG_INTERNALS
#define CYTHON_USE_PYLONG_INTERNALS 0
#ifndef CYTHON_AVOID_BORROWED_REFS
#define CYTHON_AVOID_BORROWED_REFS 0
#endif
#ifndef CYTHON_ASSUME_SAFE_MACROS
#define CYTHON_ASSUME_SAFE_MACROS 1
#endif
#ifndef CYTHON_UNPACK_METHODS
#define CYTHON_UNPACK_METHODS 1
#endif
#undef CYTHON_FAST_THREAD_STATE
#define CYTHON_FAST_THREAD_STATE 0
#undef CYTHON_FAST_PYCALL
#define CYTHON_FAST_PYCALL 0
#ifndef CYTHON_PEP489_MULTI_PHASE_INIT
#define CYTHON_PEP489_MULTI_PHASE_INIT 1
#endif
#ifndef CYTHON_USE_TP_FINALIZE
#define CYTHON_USE_TP_FINALIZE 1
#endif
#undef CYTHON_USE_DICT_VERSIONS
#define CYTHON_USE_DICT_VERSIONS 0
#undef CYTHON_USE_EXC_INFO_STACK
#define CYTHON_USE_EXC_INFO_STACK 0
#else
#define CYTHON_COMPILING_IN_PYPY 0
#define CYTHON_COMPILING_IN_PYSTON 0
#define CYTHON_COMPILING_IN_CPYTHON 1
#define CYTHON_COMPILING_IN_NOGIL 0
#ifndef CYTHON_USE_TYPE_SLOTS
#define CYTHON_USE_TYPE_SLOTS 1
#endif
#if PY_VERSION_HEX < 0x02070000
#undef CYTHON_USE_PYTYPE_LOOKUP
#define CYTHON_USE_PYTYPE_LOOKUP 0
#elif !defined(CYTHON_USE_PYTYPE_LOOKUP)
#define CYTHON_USE_PYTYPE_LOOKUP 1
#endif
#if PY_MAJOR_VERSION < 3
#undef CYTHON_USE_ASYNC_SLOTS
#define CYTHON_USE_ASYNC_SLOTS 0
#elif !defined(CYTHON_USE_ASYNC_SLOTS)
#define CYTHON_USE_ASYNC_SLOTS 1
#endif
#if PY_VERSION_HEX < 0x02070000
#undef CYTHON_USE_PYLONG_INTERNALS
#define CYTHON_USE_PYLONG_INTERNALS 0
#elif !defined(CYTHON_USE_PYLONG_INTERNALS)
#define CYTHON_USE_PYLONG_INTERNALS (PY_VERSION_HEX < 0x030C00A5)
#endif
#ifndef CYTHON_USE_PYLIST_INTERNALS
#define CYTHON_USE_PYLIST_INTERNALS 1
#endif
#ifndef CYTHON_USE_UNICODE_INTERNALS
#define CYTHON_USE_UNICODE_INTERNALS 1
#endif
#if PY_VERSION_HEX < 0x030300F0 || PY_VERSION_HEX >= 0x030B00A2
#undef CYTHON_USE_UNICODE_WRITER
#define CYTHON_USE_UNICODE_WRITER 0
#elif !defined(CYTHON_USE_UNICODE_WRITER)
#define CYTHON_USE_UNICODE_WRITER 1
#endif
#ifndef CYTHON_AVOID_BORROWED_REFS
#define CYTHON_AVOID_BORROWED_REFS 0
#endif
#ifndef CYTHON_ASSUME_SAFE_MACROS
#define CYTHON_ASSUME_SAFE_MACROS 1
#endif
#ifndef CYTHON_UNPACK_METHODS
#define CYTHON_UNPACK_METHODS 1
#endif
#if PY_VERSION_HEX >= 0x030B00A4
#undef CYTHON_FAST_THREAD_STATE
#define CYTHON_FAST_THREAD_STATE 0
#elif !defined(CYTHON_FAST_THREAD_STATE)
#define CYTHON_FAST_THREAD_STATE 1
#endif
#ifndef CYTHON_FAST_PYCALL
#define CYTHON_FAST_PYCALL (PY_VERSION_HEX < 0x030A0000)
#endif
#ifndef CYTHON_PEP489_MULTI_PHASE_INIT
#define CYTHON_PEP489_MULTI_PHASE_INIT (PY_VERSION_HEX >= 0x03050000)
#endif
#ifndef CYTHON_USE_TP_FINALIZE
#define CYTHON_USE_TP_FINALIZE (PY_VERSION_HEX >= 0x030400a1)
#endif
#ifndef CYTHON_USE_DICT_VERSIONS
#define CYTHON_USE_DICT_VERSIONS ((PY_VERSION_HEX >= 0x030600B1) && (PY_VERSION_HEX < 0x030C00A5))
#endif
#if PY_VERSION_HEX >= 0x030B00A4
#undef CYTHON_USE_EXC_INFO_STACK
#define CYTHON_USE_EXC_INFO_STACK 0
#elif !defined(CYTHON_USE_EXC_INFO_STACK)
#define CYTHON_USE_EXC_INFO_STACK (PY_VERSION_HEX >= 0x030700A3)
#endif
#ifndef CYTHON_UPDATE_DESCRIPTOR_DOC
#define CYTHON_UPDATE_DESCRIPTOR_DOC 1
#endif
#endif
#if !defined(CYTHON_FAST_PYCCALL)
#define CYTHON_FAST_PYCCALL (CYTHON_FAST_PYCALL && PY_VERSION_HEX >= 0x030600B1)
#endif
#if CYTHON_USE_PYLONG_INTERNALS
#if PY_MAJOR_VERSION < 3
#include "longintrepr.h"
#endif
#undef SHIFT
#undef BASE
#undef MASK
#ifdef SIZEOF_VOID_P
enum { __pyx_check_sizeof_voidp = 1 / (int)(SIZEOF_VOID_P == sizeof(void*)) };
#endif
#endif
#ifndef __has_attribute
#define __has_attribute(x) 0
#endif
#ifndef __has_cpp_attribute
#define __has_cpp_attribute(x) 0
#endif
#ifndef CYTHON_RESTRICT
#if defined(__GNUC__)
#define CYTHON_RESTRICT __restrict__
#elif defined(_MSC_VER) && _MSC_VER >= 1400
#define CYTHON_RESTRICT __restrict
#elif defined (__STDC_VERSION__) && __STDC_VERSION__ >= 199901L
#define CYTHON_RESTRICT restrict
#else
#define CYTHON_RESTRICT
#endif
#endif
#ifndef CYTHON_UNUSED
# if defined(__GNUC__)
# if !(defined(__cplusplus)) || (__GNUC__ > 3 || (__GNUC__ == 3 && __GNUC_MINOR__ >= 4))
# define CYTHON_UNUSED __attribute__ ((__unused__))
# else
# define CYTHON_UNUSED
# endif
# elif defined(__ICC) || (defined(__INTEL_COMPILER) && !defined(_MSC_VER))
# define CYTHON_UNUSED __attribute__ ((__unused__))
# else
# define CYTHON_UNUSED
# endif
#endif
#ifndef CYTHON_MAYBE_UNUSED_VAR
# if defined(__cplusplus)
template void CYTHON_MAYBE_UNUSED_VAR( const T& ) { }
# else
# define CYTHON_MAYBE_UNUSED_VAR(x) (void)(x)
# endif
#endif
#ifndef CYTHON_NCP_UNUSED
# if CYTHON_COMPILING_IN_CPYTHON
# define CYTHON_NCP_UNUSED
# else
# define CYTHON_NCP_UNUSED CYTHON_UNUSED
# endif
#endif
#define __Pyx_void_to_None(void_result) ((void)(void_result), Py_INCREF(Py_None), Py_None)
#ifdef _MSC_VER
#ifndef _MSC_STDINT_H_
#if _MSC_VER < 1300
typedef unsigned char uint8_t;
typedef unsigned int uint32_t;
#else
typedef unsigned __int8 uint8_t;
typedef unsigned __int32 uint32_t;
#endif
#endif
#else
#include
#endif
#ifndef CYTHON_FALLTHROUGH
#if defined(__cplusplus) && __cplusplus >= 201103L
#if __has_cpp_attribute(fallthrough)
#define CYTHON_FALLTHROUGH [[fallthrough]]
#elif __has_cpp_attribute(clang::fallthrough)
#define CYTHON_FALLTHROUGH [[clang::fallthrough]]
#elif __has_cpp_attribute(gnu::fallthrough)
#define CYTHON_FALLTHROUGH [[gnu::fallthrough]]
#endif
#endif
#ifndef CYTHON_FALLTHROUGH
#if __has_attribute(fallthrough)
#define CYTHON_FALLTHROUGH __attribute__((fallthrough))
#else
#define CYTHON_FALLTHROUGH
#endif
#endif
#if defined(__clang__ ) && defined(__apple_build_version__)
#if __apple_build_version__ < 7000000
#undef CYTHON_FALLTHROUGH
#define CYTHON_FALLTHROUGH
#endif
#endif
#endif
#ifndef __cplusplus
#error "Cython files generated with the C++ option must be compiled with a C++ compiler."
#endif
#ifndef CYTHON_INLINE
#if defined(__clang__)
#define CYTHON_INLINE __inline__ __attribute__ ((__unused__))
#else
#define CYTHON_INLINE inline
#endif
#endif
template
void __Pyx_call_destructor(T& x) {
x.~T();
}
template
class __Pyx_FakeReference {
public:
__Pyx_FakeReference() : ptr(NULL) { }
__Pyx_FakeReference(const T& ref) : ptr(const_cast(&ref)) { }
T *operator->() { return ptr; }
T *operator&() { return ptr; }
operator T&() { return *ptr; }
template bool operator ==(U other) { return *ptr == other; }
template bool operator !=(U other) { return *ptr != other; }
private:
T *ptr;
};
#define __PYX_BUILD_PY_SSIZE_T "n"
#define CYTHON_FORMAT_SSIZE_T "z"
#if PY_MAJOR_VERSION < 3
#define __Pyx_BUILTIN_MODULE_NAME "__builtin__"
#define __Pyx_PyCode_New(a, k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos)\
PyCode_New(a+k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos)
#define __Pyx_DefaultClassType PyClass_Type
#else
#define __Pyx_BUILTIN_MODULE_NAME "builtins"
#define __Pyx_DefaultClassType PyType_Type
#if PY_VERSION_HEX >= 0x030B00A1
static CYTHON_INLINE PyCodeObject* __Pyx_PyCode_New(int a, int k, int l, int s, int f,
PyObject *code, PyObject *c, PyObject* n, PyObject *v,
PyObject *fv, PyObject *cell, PyObject* fn,
PyObject *name, int fline, PyObject *lnos) {
PyObject *kwds=NULL, *argcount=NULL, *posonlyargcount=NULL, *kwonlyargcount=NULL;
PyObject *nlocals=NULL, *stacksize=NULL, *flags=NULL, *replace=NULL, *call_result=NULL, *empty=NULL;
const char *fn_cstr=NULL;
const char *name_cstr=NULL;
PyCodeObject* co=NULL;
PyObject *type, *value, *traceback;
PyErr_Fetch(&type, &value, &traceback);
if (!(kwds=PyDict_New())) goto end;
if (!(argcount=PyLong_FromLong(a))) goto end;
if (PyDict_SetItemString(kwds, "co_argcount", argcount) != 0) goto end;
if (!(posonlyargcount=PyLong_FromLong(0))) goto end;
if (PyDict_SetItemString(kwds, "co_posonlyargcount", posonlyargcount) != 0) goto end;
if (!(kwonlyargcount=PyLong_FromLong(k))) goto end;
if (PyDict_SetItemString(kwds, "co_kwonlyargcount", kwonlyargcount) != 0) goto end;
if (!(nlocals=PyLong_FromLong(l))) goto end;
if (PyDict_SetItemString(kwds, "co_nlocals", nlocals) != 0) goto end;
if (!(stacksize=PyLong_FromLong(s))) goto end;
if (PyDict_SetItemString(kwds, "co_stacksize", stacksize) != 0) goto end;
if (!(flags=PyLong_FromLong(f))) goto end;
if (PyDict_SetItemString(kwds, "co_flags", flags) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_code", code) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_consts", c) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_names", n) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_varnames", v) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_freevars", fv) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_cellvars", cell) != 0) goto end;
if (PyDict_SetItemString(kwds, "co_linetable", lnos) != 0) goto end;
if (!(fn_cstr=PyUnicode_AsUTF8AndSize(fn, NULL))) goto end;
if (!(name_cstr=PyUnicode_AsUTF8AndSize(name, NULL))) goto end;
if (!(co = PyCode_NewEmpty(fn_cstr, name_cstr, fline))) goto end;
if (!(replace = PyObject_GetAttrString((PyObject*)co, "replace"))) goto cleanup_code_too;
if (!(empty = PyTuple_New(0))) goto cleanup_code_too; // unfortunately __pyx_empty_tuple isn't available here
if (!(call_result = PyObject_Call(replace, empty, kwds))) goto cleanup_code_too;
Py_XDECREF((PyObject*)co);
co = (PyCodeObject*)call_result;
call_result = NULL;
if (0) {
cleanup_code_too:
Py_XDECREF((PyObject*)co);
co = NULL;
}
end:
Py_XDECREF(kwds);
Py_XDECREF(argcount);
Py_XDECREF(posonlyargcount);
Py_XDECREF(kwonlyargcount);
Py_XDECREF(nlocals);
Py_XDECREF(stacksize);
Py_XDECREF(replace);
Py_XDECREF(call_result);
Py_XDECREF(empty);
if (type) {
PyErr_Restore(type, value, traceback);
}
return co;
}
#else
#define __Pyx_PyCode_New(a, k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos)\
PyCode_New(a, k, l, s, f, code, c, n, v, fv, cell, fn, name, fline, lnos)
#endif
#define __Pyx_DefaultClassType PyType_Type
#endif
#if PY_VERSION_HEX >= 0x030900F0 && !CYTHON_COMPILING_IN_PYPY
#define __Pyx_PyObject_GC_IsFinalized(o) PyObject_GC_IsFinalized(o)
#else
#define __Pyx_PyObject_GC_IsFinalized(o) _PyGC_FINALIZED(o)
#endif
#ifndef Py_TPFLAGS_CHECKTYPES
#define Py_TPFLAGS_CHECKTYPES 0
#endif
#ifndef Py_TPFLAGS_HAVE_INDEX
#define Py_TPFLAGS_HAVE_INDEX 0
#endif
#ifndef Py_TPFLAGS_HAVE_NEWBUFFER
#define Py_TPFLAGS_HAVE_NEWBUFFER 0
#endif
#ifndef Py_TPFLAGS_HAVE_FINALIZE
#define Py_TPFLAGS_HAVE_FINALIZE 0
#endif
#ifndef METH_STACKLESS
#define METH_STACKLESS 0
#endif
#if PY_VERSION_HEX <= 0x030700A3 || !defined(METH_FASTCALL)
#ifndef METH_FASTCALL
#define METH_FASTCALL 0x80
#endif
typedef PyObject *(*__Pyx_PyCFunctionFast) (PyObject *self, PyObject *const *args, Py_ssize_t nargs);
typedef PyObject *(*__Pyx_PyCFunctionFastWithKeywords) (PyObject *self, PyObject *const *args,
Py_ssize_t nargs, PyObject *kwnames);
#else
#define __Pyx_PyCFunctionFast _PyCFunctionFast
#define __Pyx_PyCFunctionFastWithKeywords _PyCFunctionFastWithKeywords
#endif
#if CYTHON_FAST_PYCCALL
#define __Pyx_PyFastCFunction_Check(func)\
((PyCFunction_Check(func) && (METH_FASTCALL == (PyCFunction_GET_FLAGS(func) & ~(METH_CLASS | METH_STATIC | METH_COEXIST | METH_KEYWORDS | METH_STACKLESS)))))
#else
#define __Pyx_PyFastCFunction_Check(func) 0
#endif
#if CYTHON_COMPILING_IN_PYPY && !defined(PyObject_Malloc)
#define PyObject_Malloc(s) PyMem_Malloc(s)
#define PyObject_Free(p) PyMem_Free(p)
#define PyObject_Realloc(p) PyMem_Realloc(p)
#endif
#if CYTHON_COMPILING_IN_CPYTHON && PY_VERSION_HEX < 0x030400A1
#define PyMem_RawMalloc(n) PyMem_Malloc(n)
#define PyMem_RawRealloc(p, n) PyMem_Realloc(p, n)
#define PyMem_RawFree(p) PyMem_Free(p)
#endif
#if CYTHON_COMPILING_IN_PYSTON
#define __Pyx_PyCode_HasFreeVars(co) PyCode_HasFreeVars(co)
#define __Pyx_PyFrame_SetLineNumber(frame, lineno) PyFrame_SetLineNumber(frame, lineno)
#else
#define __Pyx_PyCode_HasFreeVars(co) (PyCode_GetNumFree(co) > 0)
#define __Pyx_PyFrame_SetLineNumber(frame, lineno) (frame)->f_lineno = (lineno)
#endif
#if !CYTHON_FAST_THREAD_STATE || PY_VERSION_HEX < 0x02070000
#define __Pyx_PyThreadState_Current PyThreadState_GET()
#elif PY_VERSION_HEX >= 0x03060000
#define __Pyx_PyThreadState_Current _PyThreadState_UncheckedGet()
#elif PY_VERSION_HEX >= 0x03000000
#define __Pyx_PyThreadState_Current PyThreadState_GET()
#else
#define __Pyx_PyThreadState_Current _PyThreadState_Current
#endif
#if PY_VERSION_HEX < 0x030700A2 && !defined(PyThread_tss_create) && !defined(Py_tss_NEEDS_INIT)
#include "pythread.h"
#define Py_tss_NEEDS_INIT 0
typedef int Py_tss_t;
static CYTHON_INLINE int PyThread_tss_create(Py_tss_t *key) {
*key = PyThread_create_key();
return 0;
}
static CYTHON_INLINE Py_tss_t * PyThread_tss_alloc(void) {
Py_tss_t *key = (Py_tss_t *)PyObject_Malloc(sizeof(Py_tss_t));
*key = Py_tss_NEEDS_INIT;
return key;
}
static CYTHON_INLINE void PyThread_tss_free(Py_tss_t *key) {
PyObject_Free(key);
}
static CYTHON_INLINE int PyThread_tss_is_created(Py_tss_t *key) {
return *key != Py_tss_NEEDS_INIT;
}
static CYTHON_INLINE void PyThread_tss_delete(Py_tss_t *key) {
PyThread_delete_key(*key);
*key = Py_tss_NEEDS_INIT;
}
static CYTHON_INLINE int PyThread_tss_set(Py_tss_t *key, void *value) {
return PyThread_set_key_value(*key, value);
}
static CYTHON_INLINE void * PyThread_tss_get(Py_tss_t *key) {
return PyThread_get_key_value(*key);
}
#endif
#if CYTHON_COMPILING_IN_CPYTHON || defined(_PyDict_NewPresized)
#define __Pyx_PyDict_NewPresized(n) ((n <= 8) ? PyDict_New() : _PyDict_NewPresized(n))
#else
#define __Pyx_PyDict_NewPresized(n) PyDict_New()
#endif
#if PY_MAJOR_VERSION >= 3 || CYTHON_FUTURE_DIVISION
#define __Pyx_PyNumber_Divide(x,y) PyNumber_TrueDivide(x,y)
#define __Pyx_PyNumber_InPlaceDivide(x,y) PyNumber_InPlaceTrueDivide(x,y)
#else
#define __Pyx_PyNumber_Divide(x,y) PyNumber_Divide(x,y)
#define __Pyx_PyNumber_InPlaceDivide(x,y) PyNumber_InPlaceDivide(x,y)
#endif
#if CYTHON_COMPILING_IN_CPYTHON && PY_VERSION_HEX >= 0x030500A1 && CYTHON_USE_UNICODE_INTERNALS
#define __Pyx_PyDict_GetItemStr(dict, name) _PyDict_GetItem_KnownHash(dict, name, ((PyASCIIObject *) name)->hash)
#else
#define __Pyx_PyDict_GetItemStr(dict, name) PyDict_GetItem(dict, name)
#endif
#if PY_VERSION_HEX > 0x03030000 && defined(PyUnicode_KIND)
#define CYTHON_PEP393_ENABLED 1
#if PY_VERSION_HEX >= 0x030C0000
#define __Pyx_PyUnicode_READY(op) (0)
#else
#define __Pyx_PyUnicode_READY(op) (likely(PyUnicode_IS_READY(op)) ?\
0 : _PyUnicode_Ready((PyObject *)(op)))
#endif
#define __Pyx_PyUnicode_GET_LENGTH(u) PyUnicode_GET_LENGTH(u)
#define __Pyx_PyUnicode_READ_CHAR(u, i) PyUnicode_READ_CHAR(u, i)
#define __Pyx_PyUnicode_MAX_CHAR_VALUE(u) PyUnicode_MAX_CHAR_VALUE(u)
#define __Pyx_PyUnicode_KIND(u) PyUnicode_KIND(u)
#define __Pyx_PyUnicode_DATA(u) PyUnicode_DATA(u)
#define __Pyx_PyUnicode_READ(k, d, i) PyUnicode_READ(k, d, i)
#define __Pyx_PyUnicode_WRITE(k, d, i, ch) PyUnicode_WRITE(k, d, i, ch)
#if PY_VERSION_HEX >= 0x030C0000
#define __Pyx_PyUnicode_IS_TRUE(u) (0 != PyUnicode_GET_LENGTH(u))
#else
#if CYTHON_COMPILING_IN_CPYTHON && PY_VERSION_HEX >= 0x03090000
#define __Pyx_PyUnicode_IS_TRUE(u) (0 != (likely(PyUnicode_IS_READY(u)) ? PyUnicode_GET_LENGTH(u) : ((PyCompactUnicodeObject *)(u))->wstr_length))
#else
#define __Pyx_PyUnicode_IS_TRUE(u) (0 != (likely(PyUnicode_IS_READY(u)) ? PyUnicode_GET_LENGTH(u) : PyUnicode_GET_SIZE(u)))
#endif
#endif
#else
#define CYTHON_PEP393_ENABLED 0
#define PyUnicode_1BYTE_KIND 1
#define PyUnicode_2BYTE_KIND 2
#define PyUnicode_4BYTE_KIND 4
#define __Pyx_PyUnicode_READY(op) (0)
#define __Pyx_PyUnicode_GET_LENGTH(u) PyUnicode_GET_SIZE(u)
#define __Pyx_PyUnicode_READ_CHAR(u, i) ((Py_UCS4)(PyUnicode_AS_UNICODE(u)[i]))
#define __Pyx_PyUnicode_MAX_CHAR_VALUE(u) ((sizeof(Py_UNICODE) == 2) ? 65535 : 1114111)
#define __Pyx_PyUnicode_KIND(u) (sizeof(Py_UNICODE))
#define __Pyx_PyUnicode_DATA(u) ((void*)PyUnicode_AS_UNICODE(u))
#define __Pyx_PyUnicode_READ(k, d, i) ((void)(k), (Py_UCS4)(((Py_UNICODE*)d)[i]))
#define __Pyx_PyUnicode_WRITE(k, d, i, ch) (((void)(k)), ((Py_UNICODE*)d)[i] = ch)
#define __Pyx_PyUnicode_IS_TRUE(u) (0 != PyUnicode_GET_SIZE(u))
#endif
#if CYTHON_COMPILING_IN_PYPY
#define __Pyx_PyUnicode_Concat(a, b) PyNumber_Add(a, b)
#define __Pyx_PyUnicode_ConcatSafe(a, b) PyNumber_Add(a, b)
#else
#define __Pyx_PyUnicode_Concat(a, b) PyUnicode_Concat(a, b)
#define __Pyx_PyUnicode_ConcatSafe(a, b) ((unlikely((a) == Py_None) || unlikely((b) == Py_None)) ?\
PyNumber_Add(a, b) : __Pyx_PyUnicode_Concat(a, b))
#endif
#if CYTHON_COMPILING_IN_PYPY && !defined(PyUnicode_Contains)
#define PyUnicode_Contains(u, s) PySequence_Contains(u, s)
#endif
#if CYTHON_COMPILING_IN_PYPY && !defined(PyByteArray_Check)
#define PyByteArray_Check(obj) PyObject_TypeCheck(obj, &PyByteArray_Type)
#endif
#if CYTHON_COMPILING_IN_PYPY && !defined(PyObject_Format)
#define PyObject_Format(obj, fmt) PyObject_CallMethod(obj, "__format__", "O", fmt)
#endif
#define __Pyx_PyString_FormatSafe(a, b) ((unlikely((a) == Py_None || (PyString_Check(b) && !PyString_CheckExact(b)))) ? PyNumber_Remainder(a, b) : __Pyx_PyString_Format(a, b))
#define __Pyx_PyUnicode_FormatSafe(a, b) ((unlikely((a) == Py_None || (PyUnicode_Check(b) && !PyUnicode_CheckExact(b)))) ? PyNumber_Remainder(a, b) : PyUnicode_Format(a, b))
#if PY_MAJOR_VERSION >= 3
#define __Pyx_PyString_Format(a, b) PyUnicode_Format(a, b)
#else
#define __Pyx_PyString_Format(a, b) PyString_Format(a, b)
#endif
#if PY_MAJOR_VERSION < 3 && !defined(PyObject_ASCII)
#define PyObject_ASCII(o) PyObject_Repr(o)
#endif
#if PY_MAJOR_VERSION >= 3
#define PyBaseString_Type PyUnicode_Type
#define PyStringObject PyUnicodeObject
#define PyString_Type PyUnicode_Type
#define PyString_Check PyUnicode_Check
#define PyString_CheckExact PyUnicode_CheckExact
#ifndef PyObject_Unicode
#define PyObject_Unicode PyObject_Str
#endif
#endif
#if PY_MAJOR_VERSION >= 3
#define __Pyx_PyBaseString_Check(obj) PyUnicode_Check(obj)
#define __Pyx_PyBaseString_CheckExact(obj) PyUnicode_CheckExact(obj)
#else
#define __Pyx_PyBaseString_Check(obj) (PyString_Check(obj) || PyUnicode_Check(obj))
#define __Pyx_PyBaseString_CheckExact(obj) (PyString_CheckExact(obj) || PyUnicode_CheckExact(obj))
#endif
#ifndef PySet_CheckExact
#define PySet_CheckExact(obj) (Py_TYPE(obj) == &PySet_Type)
#endif
#if PY_VERSION_HEX >= 0x030900A4
#define __Pyx_SET_REFCNT(obj, refcnt) Py_SET_REFCNT(obj, refcnt)
#define __Pyx_SET_SIZE(obj, size) Py_SET_SIZE(obj, size)
#else
#define __Pyx_SET_REFCNT(obj, refcnt) Py_REFCNT(obj) = (refcnt)
#define __Pyx_SET_SIZE(obj, size) Py_SIZE(obj) = (size)
#endif
#if CYTHON_ASSUME_SAFE_MACROS
#define __Pyx_PySequence_SIZE(seq) Py_SIZE(seq)
#else
#define __Pyx_PySequence_SIZE(seq) PySequence_Size(seq)
#endif
#if PY_MAJOR_VERSION >= 3
#define PyIntObject PyLongObject
#define PyInt_Type PyLong_Type
#define PyInt_Check(op) PyLong_Check(op)
#define PyInt_CheckExact(op) PyLong_CheckExact(op)
#define PyInt_FromString PyLong_FromString
#define PyInt_FromUnicode PyLong_FromUnicode
#define PyInt_FromLong PyLong_FromLong
#define PyInt_FromSize_t PyLong_FromSize_t
#define PyInt_FromSsize_t PyLong_FromSsize_t
#define PyInt_AsLong PyLong_AsLong
#define PyInt_AS_LONG PyLong_AS_LONG
#define PyInt_AsSsize_t PyLong_AsSsize_t
#define PyInt_AsUnsignedLongMask PyLong_AsUnsignedLongMask
#define PyInt_AsUnsignedLongLongMask PyLong_AsUnsignedLongLongMask
#define PyNumber_Int PyNumber_Long
#endif
#if PY_MAJOR_VERSION >= 3
#define PyBoolObject PyLongObject
#endif
#if PY_MAJOR_VERSION >= 3 && CYTHON_COMPILING_IN_PYPY
#ifndef PyUnicode_InternFromString
#define PyUnicode_InternFromString(s) PyUnicode_FromString(s)
#endif
#endif
#if PY_VERSION_HEX < 0x030200A4
typedef long Py_hash_t;
#define __Pyx_PyInt_FromHash_t PyInt_FromLong
#define __Pyx_PyInt_AsHash_t __Pyx_PyIndex_AsHash_t
#else
#define __Pyx_PyInt_FromHash_t PyInt_FromSsize_t
#define __Pyx_PyInt_AsHash_t __Pyx_PyIndex_AsSsize_t
#endif
#if PY_MAJOR_VERSION >= 3
#define __Pyx_PyMethod_New(func, self, klass) ((self) ? ((void)(klass), PyMethod_New(func, self)) : __Pyx_NewRef(func))
#else
#define __Pyx_PyMethod_New(func, self, klass) PyMethod_New(func, self, klass)
#endif
#if CYTHON_USE_ASYNC_SLOTS
#if PY_VERSION_HEX >= 0x030500B1
#define __Pyx_PyAsyncMethodsStruct PyAsyncMethods
#define __Pyx_PyType_AsAsync(obj) (Py_TYPE(obj)->tp_as_async)
#else
#define __Pyx_PyType_AsAsync(obj) ((__Pyx_PyAsyncMethodsStruct*) (Py_TYPE(obj)->tp_reserved))
#endif
#else
#define __Pyx_PyType_AsAsync(obj) NULL
#endif
#ifndef __Pyx_PyAsyncMethodsStruct
typedef struct {
unaryfunc am_await;
unaryfunc am_aiter;
unaryfunc am_anext;
} __Pyx_PyAsyncMethodsStruct;
#endif
#if defined(_WIN32) || defined(WIN32) || defined(MS_WINDOWS)
#if !defined(_USE_MATH_DEFINES)
#define _USE_MATH_DEFINES
#endif
#endif
#include
#ifdef NAN
#define __PYX_NAN() ((float) NAN)
#else
static CYTHON_INLINE float __PYX_NAN() {
float value;
memset(&value, 0xFF, sizeof(value));
return value;
}
#endif
#if defined(__CYGWIN__) && defined(_LDBL_EQ_DBL)
#define __Pyx_truncl trunc
#else
#define __Pyx_truncl truncl
#endif
#define __PYX_MARK_ERR_POS(f_index, lineno) \
{ __pyx_filename = __pyx_f[f_index]; (void)__pyx_filename; __pyx_lineno = lineno; (void)__pyx_lineno; __pyx_clineno = __LINE__; (void)__pyx_clineno; }
#define __PYX_ERR(f_index, lineno, Ln_error) \
{ __PYX_MARK_ERR_POS(f_index, lineno) goto Ln_error; }
#ifndef __PYX_EXTERN_C
#ifdef __cplusplus
#define __PYX_EXTERN_C extern "C"
#else
#define __PYX_EXTERN_C extern
#endif
#endif
#define __PYX_HAVE__gensim__models__doc2vec_inner
#define __PYX_HAVE_API__gensim__models__doc2vec_inner
/* Early includes */
#include
#include
#include "numpy/arrayobject.h"
#include "numpy/ndarrayobject.h"
#include "numpy/ndarraytypes.h"
#include "numpy/arrayscalars.h"
#include "numpy/ufuncobject.h"
/* NumPy API declarations from "numpy/__init__.pxd" */
#include "voidptr.h"
#ifdef _OPENMP
#include
#endif /* _OPENMP */
#if defined(PYREX_WITHOUT_ASSERTIONS) && !defined(CYTHON_WITHOUT_ASSERTIONS)
#define CYTHON_WITHOUT_ASSERTIONS
#endif
typedef struct {PyObject **p; const char *s; const Py_ssize_t n; const char* encoding;
const char is_unicode; const char is_str; const char intern; } __Pyx_StringTabEntry;
#define __PYX_DEFAULT_STRING_ENCODING_IS_ASCII 0
#define __PYX_DEFAULT_STRING_ENCODING_IS_UTF8 0
#define __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT (PY_MAJOR_VERSION >= 3 && __PYX_DEFAULT_STRING_ENCODING_IS_UTF8)
#define __PYX_DEFAULT_STRING_ENCODING ""
#define __Pyx_PyObject_FromString __Pyx_PyBytes_FromString
#define __Pyx_PyObject_FromStringAndSize __Pyx_PyBytes_FromStringAndSize
#define __Pyx_uchar_cast(c) ((unsigned char)c)
#define __Pyx_long_cast(x) ((long)x)
#define __Pyx_fits_Py_ssize_t(v, type, is_signed) (\
(sizeof(type) < sizeof(Py_ssize_t)) ||\
(sizeof(type) > sizeof(Py_ssize_t) &&\
likely(v < (type)PY_SSIZE_T_MAX ||\
v == (type)PY_SSIZE_T_MAX) &&\
(!is_signed || likely(v > (type)PY_SSIZE_T_MIN ||\
v == (type)PY_SSIZE_T_MIN))) ||\
(sizeof(type) == sizeof(Py_ssize_t) &&\
(is_signed || likely(v < (type)PY_SSIZE_T_MAX ||\
v == (type)PY_SSIZE_T_MAX))) )
static CYTHON_INLINE int __Pyx_is_valid_index(Py_ssize_t i, Py_ssize_t limit) {
return (size_t) i < (size_t) limit;
}
#if defined (__cplusplus) && __cplusplus >= 201103L
#include
#define __Pyx_sst_abs(value) std::abs(value)
#elif SIZEOF_INT >= SIZEOF_SIZE_T
#define __Pyx_sst_abs(value) abs(value)
#elif SIZEOF_LONG >= SIZEOF_SIZE_T
#define __Pyx_sst_abs(value) labs(value)
#elif defined (_MSC_VER)
#define __Pyx_sst_abs(value) ((Py_ssize_t)_abs64(value))
#elif defined (__STDC_VERSION__) && __STDC_VERSION__ >= 199901L
#define __Pyx_sst_abs(value) llabs(value)
#elif defined (__GNUC__)
#define __Pyx_sst_abs(value) __builtin_llabs(value)
#else
#define __Pyx_sst_abs(value) ((value<0) ? -value : value)
#endif
static CYTHON_INLINE const char* __Pyx_PyObject_AsString(PyObject*);
static CYTHON_INLINE const char* __Pyx_PyObject_AsStringAndSize(PyObject*, Py_ssize_t* length);
#define __Pyx_PyByteArray_FromString(s) PyByteArray_FromStringAndSize((const char*)s, strlen((const char*)s))
#define __Pyx_PyByteArray_FromStringAndSize(s, l) PyByteArray_FromStringAndSize((const char*)s, l)
#define __Pyx_PyBytes_FromString PyBytes_FromString
#define __Pyx_PyBytes_FromStringAndSize PyBytes_FromStringAndSize
static CYTHON_INLINE PyObject* __Pyx_PyUnicode_FromString(const char*);
#if PY_MAJOR_VERSION < 3
#define __Pyx_PyStr_FromString __Pyx_PyBytes_FromString
#define __Pyx_PyStr_FromStringAndSize __Pyx_PyBytes_FromStringAndSize
#else
#define __Pyx_PyStr_FromString __Pyx_PyUnicode_FromString
#define __Pyx_PyStr_FromStringAndSize __Pyx_PyUnicode_FromStringAndSize
#endif
#define __Pyx_PyBytes_AsWritableString(s) ((char*) PyBytes_AS_STRING(s))
#define __Pyx_PyBytes_AsWritableSString(s) ((signed char*) PyBytes_AS_STRING(s))
#define __Pyx_PyBytes_AsWritableUString(s) ((unsigned char*) PyBytes_AS_STRING(s))
#define __Pyx_PyBytes_AsString(s) ((const char*) PyBytes_AS_STRING(s))
#define __Pyx_PyBytes_AsSString(s) ((const signed char*) PyBytes_AS_STRING(s))
#define __Pyx_PyBytes_AsUString(s) ((const unsigned char*) PyBytes_AS_STRING(s))
#define __Pyx_PyObject_AsWritableString(s) ((char*) __Pyx_PyObject_AsString(s))
#define __Pyx_PyObject_AsWritableSString(s) ((signed char*) __Pyx_PyObject_AsString(s))
#define __Pyx_PyObject_AsWritableUString(s) ((unsigned char*) __Pyx_PyObject_AsString(s))
#define __Pyx_PyObject_AsSString(s) ((const signed char*) __Pyx_PyObject_AsString(s))
#define __Pyx_PyObject_AsUString(s) ((const unsigned char*) __Pyx_PyObject_AsString(s))
#define __Pyx_PyObject_FromCString(s) __Pyx_PyObject_FromString((const char*)s)
#define __Pyx_PyBytes_FromCString(s) __Pyx_PyBytes_FromString((const char*)s)
#define __Pyx_PyByteArray_FromCString(s) __Pyx_PyByteArray_FromString((const char*)s)
#define __Pyx_PyStr_FromCString(s) __Pyx_PyStr_FromString((const char*)s)
#define __Pyx_PyUnicode_FromCString(s) __Pyx_PyUnicode_FromString((const char*)s)
static CYTHON_INLINE size_t __Pyx_Py_UNICODE_strlen(const Py_UNICODE *u) {
const Py_UNICODE *u_end = u;
while (*u_end++) ;
return (size_t)(u_end - u - 1);
}
#define __Pyx_PyUnicode_FromUnicode(u) PyUnicode_FromUnicode(u, __Pyx_Py_UNICODE_strlen(u))
#define __Pyx_PyUnicode_FromUnicodeAndLength PyUnicode_FromUnicode
#define __Pyx_PyUnicode_AsUnicode PyUnicode_AsUnicode
#define __Pyx_NewRef(obj) (Py_INCREF(obj), obj)
#define __Pyx_Owned_Py_None(b) __Pyx_NewRef(Py_None)
static CYTHON_INLINE PyObject * __Pyx_PyBool_FromLong(long b);
static CYTHON_INLINE int __Pyx_PyObject_IsTrue(PyObject*);
static CYTHON_INLINE int __Pyx_PyObject_IsTrueAndDecref(PyObject*);
static CYTHON_INLINE PyObject* __Pyx_PyNumber_IntOrLong(PyObject* x);
#define __Pyx_PySequence_Tuple(obj)\
(likely(PyTuple_CheckExact(obj)) ? __Pyx_NewRef(obj) : PySequence_Tuple(obj))
static CYTHON_INLINE Py_ssize_t __Pyx_PyIndex_AsSsize_t(PyObject*);
static CYTHON_INLINE PyObject * __Pyx_PyInt_FromSize_t(size_t);
static CYTHON_INLINE Py_hash_t __Pyx_PyIndex_AsHash_t(PyObject*);
#if CYTHON_ASSUME_SAFE_MACROS
#define __pyx_PyFloat_AsDouble(x) (PyFloat_CheckExact(x) ? PyFloat_AS_DOUBLE(x) : PyFloat_AsDouble(x))
#else
#define __pyx_PyFloat_AsDouble(x) PyFloat_AsDouble(x)
#endif
#define __pyx_PyFloat_AsFloat(x) ((float) __pyx_PyFloat_AsDouble(x))
#if PY_MAJOR_VERSION >= 3
#define __Pyx_PyNumber_Int(x) (PyLong_CheckExact(x) ? __Pyx_NewRef(x) : PyNumber_Long(x))
#else
#define __Pyx_PyNumber_Int(x) (PyInt_CheckExact(x) ? __Pyx_NewRef(x) : PyNumber_Int(x))
#endif
#define __Pyx_PyNumber_Float(x) (PyFloat_CheckExact(x) ? __Pyx_NewRef(x) : PyNumber_Float(x))
#if PY_MAJOR_VERSION < 3 && __PYX_DEFAULT_STRING_ENCODING_IS_ASCII
static int __Pyx_sys_getdefaultencoding_not_ascii;
static int __Pyx_init_sys_getdefaultencoding_params(void) {
PyObject* sys;
PyObject* default_encoding = NULL;
PyObject* ascii_chars_u = NULL;
PyObject* ascii_chars_b = NULL;
const char* default_encoding_c;
sys = PyImport_ImportModule("sys");
if (!sys) goto bad;
default_encoding = PyObject_CallMethod(sys, (char*) "getdefaultencoding", NULL);
Py_DECREF(sys);
if (!default_encoding) goto bad;
default_encoding_c = PyBytes_AsString(default_encoding);
if (!default_encoding_c) goto bad;
if (strcmp(default_encoding_c, "ascii") == 0) {
__Pyx_sys_getdefaultencoding_not_ascii = 0;
} else {
char ascii_chars[128];
int c;
for (c = 0; c < 128; c++) {
ascii_chars[c] = c;
}
__Pyx_sys_getdefaultencoding_not_ascii = 1;
ascii_chars_u = PyUnicode_DecodeASCII(ascii_chars, 128, NULL);
if (!ascii_chars_u) goto bad;
ascii_chars_b = PyUnicode_AsEncodedString(ascii_chars_u, default_encoding_c, NULL);
if (!ascii_chars_b || !PyBytes_Check(ascii_chars_b) || memcmp(ascii_chars, PyBytes_AS_STRING(ascii_chars_b), 128) != 0) {
PyErr_Format(
PyExc_ValueError,
"This module compiled with c_string_encoding=ascii, but default encoding '%.200s' is not a superset of ascii.",
default_encoding_c);
goto bad;
}
Py_DECREF(ascii_chars_u);
Py_DECREF(ascii_chars_b);
}
Py_DECREF(default_encoding);
return 0;
bad:
Py_XDECREF(default_encoding);
Py_XDECREF(ascii_chars_u);
Py_XDECREF(ascii_chars_b);
return -1;
}
#endif
#if __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT && PY_MAJOR_VERSION >= 3
#define __Pyx_PyUnicode_FromStringAndSize(c_str, size) PyUnicode_DecodeUTF8(c_str, size, NULL)
#else
#define __Pyx_PyUnicode_FromStringAndSize(c_str, size) PyUnicode_Decode(c_str, size, __PYX_DEFAULT_STRING_ENCODING, NULL)
#if __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT
static char* __PYX_DEFAULT_STRING_ENCODING;
static int __Pyx_init_sys_getdefaultencoding_params(void) {
PyObject* sys;
PyObject* default_encoding = NULL;
char* default_encoding_c;
sys = PyImport_ImportModule("sys");
if (!sys) goto bad;
default_encoding = PyObject_CallMethod(sys, (char*) (const char*) "getdefaultencoding", NULL);
Py_DECREF(sys);
if (!default_encoding) goto bad;
default_encoding_c = PyBytes_AsString(default_encoding);
if (!default_encoding_c) goto bad;
__PYX_DEFAULT_STRING_ENCODING = (char*) malloc(strlen(default_encoding_c) + 1);
if (!__PYX_DEFAULT_STRING_ENCODING) goto bad;
strcpy(__PYX_DEFAULT_STRING_ENCODING, default_encoding_c);
Py_DECREF(default_encoding);
return 0;
bad:
Py_XDECREF(default_encoding);
return -1;
}
#endif
#endif
/* Test for GCC > 2.95 */
#if defined(__GNUC__) && (__GNUC__ > 2 || (__GNUC__ == 2 && (__GNUC_MINOR__ > 95)))
#define likely(x) __builtin_expect(!!(x), 1)
#define unlikely(x) __builtin_expect(!!(x), 0)
#else /* !__GNUC__ or GCC < 2.95 */
#define likely(x) (x)
#define unlikely(x) (x)
#endif /* __GNUC__ */
static CYTHON_INLINE void __Pyx_pretend_to_initialize(void* ptr) { (void)ptr; }
static PyObject *__pyx_m = NULL;
static PyObject *__pyx_d;
static PyObject *__pyx_b;
static PyObject *__pyx_cython_runtime = NULL;
static PyObject *__pyx_empty_tuple;
static PyObject *__pyx_empty_bytes;
static PyObject *__pyx_empty_unicode;
static int __pyx_lineno;
static int __pyx_clineno = 0;
static const char * __pyx_cfilenm= __FILE__;
static const char *__pyx_filename;
/* Header.proto */
#if !defined(CYTHON_CCOMPLEX)
#if defined(__cplusplus)
#define CYTHON_CCOMPLEX 1
#elif defined(_Complex_I)
#define CYTHON_CCOMPLEX 1
#else
#define CYTHON_CCOMPLEX 0
#endif
#endif
#if CYTHON_CCOMPLEX
#ifdef __cplusplus
#include
#else
#include
#endif
#endif
#if CYTHON_CCOMPLEX && !defined(__cplusplus) && defined(__sun__) && defined(__GNUC__)
#undef _Complex_I
#define _Complex_I 1.0fj
#endif
static const char *__pyx_f[] = {
"gensim/models/doc2vec_inner.pyx",
"__init__.pxd",
"type.pxd",
};
/* NoFastGil.proto */
#define __Pyx_PyGILState_Ensure PyGILState_Ensure
#define __Pyx_PyGILState_Release PyGILState_Release
#define __Pyx_FastGIL_Remember()
#define __Pyx_FastGIL_Forget()
#define __Pyx_FastGilFuncInit()
/* ForceInitThreads.proto */
#ifndef __PYX_FORCE_INIT_THREADS
#define __PYX_FORCE_INIT_THREADS 0
#endif
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":690
* # in Cython to enable them only on the right systems.
*
* ctypedef npy_int8 int8_t # <<<<<<<<<<<<<<
* ctypedef npy_int16 int16_t
* ctypedef npy_int32 int32_t
*/
typedef npy_int8 __pyx_t_5numpy_int8_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":691
*
* ctypedef npy_int8 int8_t
* ctypedef npy_int16 int16_t # <<<<<<<<<<<<<<
* ctypedef npy_int32 int32_t
* ctypedef npy_int64 int64_t
*/
typedef npy_int16 __pyx_t_5numpy_int16_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":692
* ctypedef npy_int8 int8_t
* ctypedef npy_int16 int16_t
* ctypedef npy_int32 int32_t # <<<<<<<<<<<<<<
* ctypedef npy_int64 int64_t
* #ctypedef npy_int96 int96_t
*/
typedef npy_int32 __pyx_t_5numpy_int32_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":693
* ctypedef npy_int16 int16_t
* ctypedef npy_int32 int32_t
* ctypedef npy_int64 int64_t # <<<<<<<<<<<<<<
* #ctypedef npy_int96 int96_t
* #ctypedef npy_int128 int128_t
*/
typedef npy_int64 __pyx_t_5numpy_int64_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":697
* #ctypedef npy_int128 int128_t
*
* ctypedef npy_uint8 uint8_t # <<<<<<<<<<<<<<
* ctypedef npy_uint16 uint16_t
* ctypedef npy_uint32 uint32_t
*/
typedef npy_uint8 __pyx_t_5numpy_uint8_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":698
*
* ctypedef npy_uint8 uint8_t
* ctypedef npy_uint16 uint16_t # <<<<<<<<<<<<<<
* ctypedef npy_uint32 uint32_t
* ctypedef npy_uint64 uint64_t
*/
typedef npy_uint16 __pyx_t_5numpy_uint16_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":699
* ctypedef npy_uint8 uint8_t
* ctypedef npy_uint16 uint16_t
* ctypedef npy_uint32 uint32_t # <<<<<<<<<<<<<<
* ctypedef npy_uint64 uint64_t
* #ctypedef npy_uint96 uint96_t
*/
typedef npy_uint32 __pyx_t_5numpy_uint32_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":700
* ctypedef npy_uint16 uint16_t
* ctypedef npy_uint32 uint32_t
* ctypedef npy_uint64 uint64_t # <<<<<<<<<<<<<<
* #ctypedef npy_uint96 uint96_t
* #ctypedef npy_uint128 uint128_t
*/
typedef npy_uint64 __pyx_t_5numpy_uint64_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":704
* #ctypedef npy_uint128 uint128_t
*
* ctypedef npy_float32 float32_t # <<<<<<<<<<<<<<
* ctypedef npy_float64 float64_t
* #ctypedef npy_float80 float80_t
*/
typedef npy_float32 __pyx_t_5numpy_float32_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":705
*
* ctypedef npy_float32 float32_t
* ctypedef npy_float64 float64_t # <<<<<<<<<<<<<<
* #ctypedef npy_float80 float80_t
* #ctypedef npy_float128 float128_t
*/
typedef npy_float64 __pyx_t_5numpy_float64_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":714
* # The int types are mapped a bit surprising --
* # numpy.int corresponds to 'l' and numpy.long to 'q'
* ctypedef npy_long int_t # <<<<<<<<<<<<<<
* ctypedef npy_longlong long_t
* ctypedef npy_longlong longlong_t
*/
typedef npy_long __pyx_t_5numpy_int_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":715
* # numpy.int corresponds to 'l' and numpy.long to 'q'
* ctypedef npy_long int_t
* ctypedef npy_longlong long_t # <<<<<<<<<<<<<<
* ctypedef npy_longlong longlong_t
*
*/
typedef npy_longlong __pyx_t_5numpy_long_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":716
* ctypedef npy_long int_t
* ctypedef npy_longlong long_t
* ctypedef npy_longlong longlong_t # <<<<<<<<<<<<<<
*
* ctypedef npy_ulong uint_t
*/
typedef npy_longlong __pyx_t_5numpy_longlong_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":718
* ctypedef npy_longlong longlong_t
*
* ctypedef npy_ulong uint_t # <<<<<<<<<<<<<<
* ctypedef npy_ulonglong ulong_t
* ctypedef npy_ulonglong ulonglong_t
*/
typedef npy_ulong __pyx_t_5numpy_uint_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":719
*
* ctypedef npy_ulong uint_t
* ctypedef npy_ulonglong ulong_t # <<<<<<<<<<<<<<
* ctypedef npy_ulonglong ulonglong_t
*
*/
typedef npy_ulonglong __pyx_t_5numpy_ulong_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":720
* ctypedef npy_ulong uint_t
* ctypedef npy_ulonglong ulong_t
* ctypedef npy_ulonglong ulonglong_t # <<<<<<<<<<<<<<
*
* ctypedef npy_intp intp_t
*/
typedef npy_ulonglong __pyx_t_5numpy_ulonglong_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":722
* ctypedef npy_ulonglong ulonglong_t
*
* ctypedef npy_intp intp_t # <<<<<<<<<<<<<<
* ctypedef npy_uintp uintp_t
*
*/
typedef npy_intp __pyx_t_5numpy_intp_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":723
*
* ctypedef npy_intp intp_t
* ctypedef npy_uintp uintp_t # <<<<<<<<<<<<<<
*
* ctypedef npy_double float_t
*/
typedef npy_uintp __pyx_t_5numpy_uintp_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":725
* ctypedef npy_uintp uintp_t
*
* ctypedef npy_double float_t # <<<<<<<<<<<<<<
* ctypedef npy_double double_t
* ctypedef npy_longdouble longdouble_t
*/
typedef npy_double __pyx_t_5numpy_float_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":726
*
* ctypedef npy_double float_t
* ctypedef npy_double double_t # <<<<<<<<<<<<<<
* ctypedef npy_longdouble longdouble_t
*
*/
typedef npy_double __pyx_t_5numpy_double_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":727
* ctypedef npy_double float_t
* ctypedef npy_double double_t
* ctypedef npy_longdouble longdouble_t # <<<<<<<<<<<<<<
*
* ctypedef npy_cfloat cfloat_t
*/
typedef npy_longdouble __pyx_t_5numpy_longdouble_t;
/* "gensim/models/word2vec_inner.pxd":20
* void* PyCObject_AsVoidPtr(object obj)
*
* ctypedef np.float32_t REAL_t # <<<<<<<<<<<<<<
*
* # BLAS routine signatures
*/
typedef __pyx_t_5numpy_float32_t __pyx_t_6gensim_6models_14word2vec_inner_REAL_t;
/* Declarations.proto */
#if CYTHON_CCOMPLEX
#ifdef __cplusplus
typedef ::std::complex< float > __pyx_t_float_complex;
#else
typedef float _Complex __pyx_t_float_complex;
#endif
#else
typedef struct { float real, imag; } __pyx_t_float_complex;
#endif
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float, float);
/* Declarations.proto */
#if CYTHON_CCOMPLEX
#ifdef __cplusplus
typedef ::std::complex< double > __pyx_t_double_complex;
#else
typedef double _Complex __pyx_t_double_complex;
#endif
#else
typedef struct { double real, imag; } __pyx_t_double_complex;
#endif
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double, double);
/*--- Type declarations ---*/
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":729
* ctypedef npy_longdouble longdouble_t
*
* ctypedef npy_cfloat cfloat_t # <<<<<<<<<<<<<<
* ctypedef npy_cdouble cdouble_t
* ctypedef npy_clongdouble clongdouble_t
*/
typedef npy_cfloat __pyx_t_5numpy_cfloat_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":730
*
* ctypedef npy_cfloat cfloat_t
* ctypedef npy_cdouble cdouble_t # <<<<<<<<<<<<<<
* ctypedef npy_clongdouble clongdouble_t
*
*/
typedef npy_cdouble __pyx_t_5numpy_cdouble_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":731
* ctypedef npy_cfloat cfloat_t
* ctypedef npy_cdouble cdouble_t
* ctypedef npy_clongdouble clongdouble_t # <<<<<<<<<<<<<<
*
* ctypedef npy_cdouble complex_t
*/
typedef npy_clongdouble __pyx_t_5numpy_clongdouble_t;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":733
* ctypedef npy_clongdouble clongdouble_t
*
* ctypedef npy_cdouble complex_t # <<<<<<<<<<<<<<
*
* cdef inline object PyArray_MultiIterNew1(a):
*/
typedef npy_cdouble __pyx_t_5numpy_complex_t;
struct __pyx_t_6gensim_6models_14word2vec_inner_Word2VecConfig;
struct __pyx_opt_args_6gensim_6models_14word2vec_inner_init_w2v_config;
/* "gensim/models/word2vec_inner.pxd":23
*
* # BLAS routine signatures
* ctypedef void (*scopy_ptr) (const int *N, const float *X, const int *incX, float *Y, const int *incY) nogil # <<<<<<<<<<<<<<
* ctypedef void (*saxpy_ptr) (const int *N, const float *alpha, const float *X, const int *incX, float *Y, const int *incY) nogil
* ctypedef float (*sdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
*/
typedef void (*__pyx_t_6gensim_6models_14word2vec_inner_scopy_ptr)(int const *, float const *, int const *, float *, int const *);
/* "gensim/models/word2vec_inner.pxd":24
* # BLAS routine signatures
* ctypedef void (*scopy_ptr) (const int *N, const float *X, const int *incX, float *Y, const int *incY) nogil
* ctypedef void (*saxpy_ptr) (const int *N, const float *alpha, const float *X, const int *incX, float *Y, const int *incY) nogil # <<<<<<<<<<<<<<
* ctypedef float (*sdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef double (*dsdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
*/
typedef void (*__pyx_t_6gensim_6models_14word2vec_inner_saxpy_ptr)(int const *, float const *, float const *, int const *, float *, int const *);
/* "gensim/models/word2vec_inner.pxd":25
* ctypedef void (*scopy_ptr) (const int *N, const float *X, const int *incX, float *Y, const int *incY) nogil
* ctypedef void (*saxpy_ptr) (const int *N, const float *alpha, const float *X, const int *incX, float *Y, const int *incY) nogil
* ctypedef float (*sdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil # <<<<<<<<<<<<<<
* ctypedef double (*dsdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef double (*snrm2_ptr) (const int *N, const float *X, const int *incX) nogil
*/
typedef float (*__pyx_t_6gensim_6models_14word2vec_inner_sdot_ptr)(int const *, float const *, int const *, float const *, int const *);
/* "gensim/models/word2vec_inner.pxd":26
* ctypedef void (*saxpy_ptr) (const int *N, const float *alpha, const float *X, const int *incX, float *Y, const int *incY) nogil
* ctypedef float (*sdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef double (*dsdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil # <<<<<<<<<<<<<<
* ctypedef double (*snrm2_ptr) (const int *N, const float *X, const int *incX) nogil
* ctypedef void (*sscal_ptr) (const int *N, const float *alpha, const float *X, const int *incX) nogil
*/
typedef double (*__pyx_t_6gensim_6models_14word2vec_inner_dsdot_ptr)(int const *, float const *, int const *, float const *, int const *);
/* "gensim/models/word2vec_inner.pxd":27
* ctypedef float (*sdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef double (*dsdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef double (*snrm2_ptr) (const int *N, const float *X, const int *incX) nogil # <<<<<<<<<<<<<<
* ctypedef void (*sscal_ptr) (const int *N, const float *alpha, const float *X, const int *incX) nogil
*
*/
typedef double (*__pyx_t_6gensim_6models_14word2vec_inner_snrm2_ptr)(int const *, float const *, int const *);
/* "gensim/models/word2vec_inner.pxd":28
* ctypedef double (*dsdot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef double (*snrm2_ptr) (const int *N, const float *X, const int *incX) nogil
* ctypedef void (*sscal_ptr) (const int *N, const float *alpha, const float *X, const int *incX) nogil # <<<<<<<<<<<<<<
*
* cdef scopy_ptr scopy
*/
typedef void (*__pyx_t_6gensim_6models_14word2vec_inner_sscal_ptr)(int const *, float const *, float const *, int const *);
/* "gensim/models/word2vec_inner.pxd":45
*
* # function implementations swapped based on BLAS detected in word2vec_inner.pyx init()
* ctypedef REAL_t (*our_dot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil # <<<<<<<<<<<<<<
* ctypedef void (*our_saxpy_ptr) (const int *N, const float *alpha, const float *X, const int *incX, float *Y, const int *incY) nogil
*
*/
typedef __pyx_t_6gensim_6models_14word2vec_inner_REAL_t (*__pyx_t_6gensim_6models_14word2vec_inner_our_dot_ptr)(int const *, float const *, int const *, float const *, int const *);
/* "gensim/models/word2vec_inner.pxd":46
* # function implementations swapped based on BLAS detected in word2vec_inner.pyx init()
* ctypedef REAL_t (*our_dot_ptr) (const int *N, const float *X, const int *incX, const float *Y, const int *incY) nogil
* ctypedef void (*our_saxpy_ptr) (const int *N, const float *alpha, const float *X, const int *incX, float *Y, const int *incY) nogil # <<<<<<<<<<<<<<
*
* cdef our_dot_ptr our_dot
*/
typedef void (*__pyx_t_6gensim_6models_14word2vec_inner_our_saxpy_ptr)(int const *, float const *, float const *, int const *, float *, int const *);
/* "gensim/models/word2vec_inner.pxd":52
*
*
* cdef struct Word2VecConfig: # <<<<<<<<<<<<<<
* int hs, negative, sample, compute_loss, size, window, cbow_mean, workers
* REAL_t running_training_loss, alpha
*/
struct __pyx_t_6gensim_6models_14word2vec_inner_Word2VecConfig {
int hs;
int negative;
int sample;
int compute_loss;
int size;
int window;
int cbow_mean;
int workers;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t running_training_loss;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t alpha;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *syn0;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *words_lockf;
__pyx_t_5numpy_uint32_t words_lockf_len;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *work;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *neu1;
int codelens[0x2710];
__pyx_t_5numpy_uint32_t indexes[0x2710];
__pyx_t_5numpy_uint32_t reduced_windows[0x2710];
int sentence_idx[(0x2710 + 1)];
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *syn1;
__pyx_t_5numpy_uint32_t *points[0x2710];
__pyx_t_5numpy_uint8_t *codes[0x2710];
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *syn1neg;
__pyx_t_5numpy_uint32_t *cum_table;
unsigned PY_LONG_LONG cum_table_len;
unsigned PY_LONG_LONG next_random;
};
/* "gensim/models/word2vec_inner.pxd":127
*
*
* cdef init_w2v_config(Word2VecConfig *c, model, alpha, compute_loss, _work, _neu1=*) # <<<<<<<<<<<<<<
*/
struct __pyx_opt_args_6gensim_6models_14word2vec_inner_init_w2v_config {
int __pyx_n;
PyObject *_neu1;
};
struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig;
struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config;
/* "gensim/models/doc2vec_inner.pxd":24
*
*
* cdef struct Doc2VecConfig: # <<<<<<<<<<<<<<
* int hs, negative, sample, learn_doctags, learn_words, learn_hidden, train_words, cbow_mean
* int document_len, doctag_len, window, expected_doctag_len, null_word_index, workers, docvecs_count
*/
struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig {
int hs;
int negative;
int sample;
int learn_doctags;
int learn_words;
int learn_hidden;
int train_words;
int cbow_mean;
int document_len;
int doctag_len;
int window;
int expected_doctag_len;
int null_word_index;
int workers;
int docvecs_count;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *word_vectors;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *doctag_vectors;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *words_lockf;
__pyx_t_5numpy_uint32_t words_lockf_len;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *doctags_lockf;
__pyx_t_5numpy_uint32_t doctags_lockf_len;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *work;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *neu1;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t alpha;
int layer1_size;
int vector_size;
int codelens[0x2710];
__pyx_t_5numpy_uint32_t indexes[0x2710];
__pyx_t_5numpy_uint32_t doctag_indexes[0x2710];
__pyx_t_5numpy_uint32_t window_indexes[0x2710];
__pyx_t_5numpy_uint32_t reduced_windows[0x2710];
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *syn1;
__pyx_t_5numpy_uint32_t *points[0x2710];
__pyx_t_5numpy_uint8_t *codes[0x2710];
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *syn1neg;
__pyx_t_5numpy_uint32_t *cum_table;
unsigned PY_LONG_LONG cum_table_len;
unsigned PY_LONG_LONG next_random;
};
/* "gensim/models/doc2vec_inner.pxd":95
*
*
* cdef init_d2v_config(Doc2VecConfig *c, model, alpha, learn_doctags, learn_words, learn_hidden, train_words=*, work=*, # <<<<<<<<<<<<<<
* neu1=*, word_vectors=*, words_lockf=*, doctag_vectors=*, doctags_lockf=*, docvecs_count=*)
*/
struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config {
int __pyx_n;
PyObject *train_words;
PyObject *work;
PyObject *neu1;
PyObject *word_vectors;
PyObject *words_lockf;
PyObject *doctag_vectors;
PyObject *doctags_lockf;
PyObject *docvecs_count;
};
/* --- Runtime support code (head) --- */
/* Refnanny.proto */
#ifndef CYTHON_REFNANNY
#define CYTHON_REFNANNY 0
#endif
#if CYTHON_REFNANNY
typedef struct {
void (*INCREF)(void*, PyObject*, int);
void (*DECREF)(void*, PyObject*, int);
void (*GOTREF)(void*, PyObject*, int);
void (*GIVEREF)(void*, PyObject*, int);
void* (*SetupContext)(const char*, int, const char*);
void (*FinishContext)(void**);
} __Pyx_RefNannyAPIStruct;
static __Pyx_RefNannyAPIStruct *__Pyx_RefNanny = NULL;
static __Pyx_RefNannyAPIStruct *__Pyx_RefNannyImportAPI(const char *modname);
#define __Pyx_RefNannyDeclarations void *__pyx_refnanny = NULL;
#ifdef WITH_THREAD
#define __Pyx_RefNannySetupContext(name, acquire_gil)\
if (acquire_gil) {\
PyGILState_STATE __pyx_gilstate_save = PyGILState_Ensure();\
__pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__);\
PyGILState_Release(__pyx_gilstate_save);\
} else {\
__pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__);\
}
#else
#define __Pyx_RefNannySetupContext(name, acquire_gil)\
__pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__)
#endif
#define __Pyx_RefNannyFinishContext()\
__Pyx_RefNanny->FinishContext(&__pyx_refnanny)
#define __Pyx_INCREF(r) __Pyx_RefNanny->INCREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
#define __Pyx_DECREF(r) __Pyx_RefNanny->DECREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
#define __Pyx_GOTREF(r) __Pyx_RefNanny->GOTREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
#define __Pyx_GIVEREF(r) __Pyx_RefNanny->GIVEREF(__pyx_refnanny, (PyObject *)(r), __LINE__)
#define __Pyx_XINCREF(r) do { if((r) != NULL) {__Pyx_INCREF(r); }} while(0)
#define __Pyx_XDECREF(r) do { if((r) != NULL) {__Pyx_DECREF(r); }} while(0)
#define __Pyx_XGOTREF(r) do { if((r) != NULL) {__Pyx_GOTREF(r); }} while(0)
#define __Pyx_XGIVEREF(r) do { if((r) != NULL) {__Pyx_GIVEREF(r);}} while(0)
#else
#define __Pyx_RefNannyDeclarations
#define __Pyx_RefNannySetupContext(name, acquire_gil)
#define __Pyx_RefNannyFinishContext()
#define __Pyx_INCREF(r) Py_INCREF(r)
#define __Pyx_DECREF(r) Py_DECREF(r)
#define __Pyx_GOTREF(r)
#define __Pyx_GIVEREF(r)
#define __Pyx_XINCREF(r) Py_XINCREF(r)
#define __Pyx_XDECREF(r) Py_XDECREF(r)
#define __Pyx_XGOTREF(r)
#define __Pyx_XGIVEREF(r)
#endif
#define __Pyx_XDECREF_SET(r, v) do {\
PyObject *tmp = (PyObject *) r;\
r = v; __Pyx_XDECREF(tmp);\
} while (0)
#define __Pyx_DECREF_SET(r, v) do {\
PyObject *tmp = (PyObject *) r;\
r = v; __Pyx_DECREF(tmp);\
} while (0)
#define __Pyx_CLEAR(r) do { PyObject* tmp = ((PyObject*)(r)); r = NULL; __Pyx_DECREF(tmp);} while(0)
#define __Pyx_XCLEAR(r) do { if((r) != NULL) {PyObject* tmp = ((PyObject*)(r)); r = NULL; __Pyx_DECREF(tmp);}} while(0)
/* PyObjectGetAttrStr.proto */
#if CYTHON_USE_TYPE_SLOTS
static CYTHON_INLINE PyObject* __Pyx_PyObject_GetAttrStr(PyObject* obj, PyObject* attr_name);
#else
#define __Pyx_PyObject_GetAttrStr(o,n) PyObject_GetAttr(o,n)
#endif
/* GetBuiltinName.proto */
static PyObject *__Pyx_GetBuiltinName(PyObject *name);
/* PyIntCompare.proto */
static CYTHON_INLINE PyObject* __Pyx_PyInt_NeObjC(PyObject *op1, PyObject *op2, long intval, long inplace);
/* PySequenceContains.proto */
static CYTHON_INLINE int __Pyx_PySequence_ContainsTF(PyObject* item, PyObject* seq, int eq) {
int result = PySequence_Contains(seq, item);
return unlikely(result < 0) ? result : (result == (eq == Py_EQ));
}
/* PyCFunctionFastCall.proto */
#if CYTHON_FAST_PYCCALL
static CYTHON_INLINE PyObject *__Pyx_PyCFunction_FastCall(PyObject *func, PyObject **args, Py_ssize_t nargs);
#else
#define __Pyx_PyCFunction_FastCall(func, args, nargs) (assert(0), NULL)
#endif
/* PyFunctionFastCall.proto */
#if CYTHON_FAST_PYCALL
#define __Pyx_PyFunction_FastCall(func, args, nargs)\
__Pyx_PyFunction_FastCallDict((func), (args), (nargs), NULL)
#if 1 || PY_VERSION_HEX < 0x030600B1
static PyObject *__Pyx_PyFunction_FastCallDict(PyObject *func, PyObject **args, Py_ssize_t nargs, PyObject *kwargs);
#else
#define __Pyx_PyFunction_FastCallDict(func, args, nargs, kwargs) _PyFunction_FastCallDict(func, args, nargs, kwargs)
#endif
#define __Pyx_BUILD_ASSERT_EXPR(cond)\
(sizeof(char [1 - 2*!(cond)]) - 1)
#ifndef Py_MEMBER_SIZE
#define Py_MEMBER_SIZE(type, member) sizeof(((type *)0)->member)
#endif
#if CYTHON_FAST_PYCALL
static size_t __pyx_pyframe_localsplus_offset = 0;
#include "frameobject.h"
#if PY_VERSION_HEX >= 0x030b00a6
#ifndef Py_BUILD_CORE
#define Py_BUILD_CORE 1
#endif
#include "internal/pycore_frame.h"
#endif
#define __Pxy_PyFrame_Initialize_Offsets()\
((void)__Pyx_BUILD_ASSERT_EXPR(sizeof(PyFrameObject) == offsetof(PyFrameObject, f_localsplus) + Py_MEMBER_SIZE(PyFrameObject, f_localsplus)),\
(void)(__pyx_pyframe_localsplus_offset = ((size_t)PyFrame_Type.tp_basicsize) - Py_MEMBER_SIZE(PyFrameObject, f_localsplus)))
#define __Pyx_PyFrame_GetLocalsplus(frame)\
(assert(__pyx_pyframe_localsplus_offset), (PyObject **)(((char *)(frame)) + __pyx_pyframe_localsplus_offset))
#endif // CYTHON_FAST_PYCALL
#endif
/* PyObjectCall.proto */
#if CYTHON_COMPILING_IN_CPYTHON
static CYTHON_INLINE PyObject* __Pyx_PyObject_Call(PyObject *func, PyObject *arg, PyObject *kw);
#else
#define __Pyx_PyObject_Call(func, arg, kw) PyObject_Call(func, arg, kw)
#endif
/* PyObjectCall2Args.proto */
static CYTHON_UNUSED PyObject* __Pyx_PyObject_Call2Args(PyObject* function, PyObject* arg1, PyObject* arg2);
/* PyObjectCallMethO.proto */
#if CYTHON_COMPILING_IN_CPYTHON
static CYTHON_INLINE PyObject* __Pyx_PyObject_CallMethO(PyObject *func, PyObject *arg);
#endif
/* PyObjectCallOneArg.proto */
static CYTHON_INLINE PyObject* __Pyx_PyObject_CallOneArg(PyObject *func, PyObject *arg);
/* ExtTypeTest.proto */
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type);
/* PyDictVersioning.proto */
#if CYTHON_USE_DICT_VERSIONS && CYTHON_USE_TYPE_SLOTS
#define __PYX_DICT_VERSION_INIT ((PY_UINT64_T) -1)
#define __PYX_GET_DICT_VERSION(dict) (((PyDictObject*)(dict))->ma_version_tag)
#define __PYX_UPDATE_DICT_CACHE(dict, value, cache_var, version_var)\
(version_var) = __PYX_GET_DICT_VERSION(dict);\
(cache_var) = (value);
#define __PYX_PY_DICT_LOOKUP_IF_MODIFIED(VAR, DICT, LOOKUP) {\
static PY_UINT64_T __pyx_dict_version = 0;\
static PyObject *__pyx_dict_cached_value = NULL;\
if (likely(__PYX_GET_DICT_VERSION(DICT) == __pyx_dict_version)) {\
(VAR) = __pyx_dict_cached_value;\
} else {\
(VAR) = __pyx_dict_cached_value = (LOOKUP);\
__pyx_dict_version = __PYX_GET_DICT_VERSION(DICT);\
}\
}
static CYTHON_INLINE PY_UINT64_T __Pyx_get_tp_dict_version(PyObject *obj);
static CYTHON_INLINE PY_UINT64_T __Pyx_get_object_dict_version(PyObject *obj);
static CYTHON_INLINE int __Pyx_object_dict_version_matches(PyObject* obj, PY_UINT64_T tp_dict_version, PY_UINT64_T obj_dict_version);
#else
#define __PYX_GET_DICT_VERSION(dict) (0)
#define __PYX_UPDATE_DICT_CACHE(dict, value, cache_var, version_var)
#define __PYX_PY_DICT_LOOKUP_IF_MODIFIED(VAR, DICT, LOOKUP) (VAR) = (LOOKUP);
#endif
/* GetModuleGlobalName.proto */
#if CYTHON_USE_DICT_VERSIONS
#define __Pyx_GetModuleGlobalName(var, name) do {\
static PY_UINT64_T __pyx_dict_version = 0;\
static PyObject *__pyx_dict_cached_value = NULL;\
(var) = (likely(__pyx_dict_version == __PYX_GET_DICT_VERSION(__pyx_d))) ?\
(likely(__pyx_dict_cached_value) ? __Pyx_NewRef(__pyx_dict_cached_value) : __Pyx_GetBuiltinName(name)) :\
__Pyx__GetModuleGlobalName(name, &__pyx_dict_version, &__pyx_dict_cached_value);\
} while(0)
#define __Pyx_GetModuleGlobalNameUncached(var, name) do {\
PY_UINT64_T __pyx_dict_version;\
PyObject *__pyx_dict_cached_value;\
(var) = __Pyx__GetModuleGlobalName(name, &__pyx_dict_version, &__pyx_dict_cached_value);\
} while(0)
static PyObject *__Pyx__GetModuleGlobalName(PyObject *name, PY_UINT64_T *dict_version, PyObject **dict_cached_value);
#else
#define __Pyx_GetModuleGlobalName(var, name) (var) = __Pyx__GetModuleGlobalName(name)
#define __Pyx_GetModuleGlobalNameUncached(var, name) (var) = __Pyx__GetModuleGlobalName(name)
static CYTHON_INLINE PyObject *__Pyx__GetModuleGlobalName(PyObject *name);
#endif
/* RaiseArgTupleInvalid.proto */
static void __Pyx_RaiseArgtupleInvalid(const char* func_name, int exact,
Py_ssize_t num_min, Py_ssize_t num_max, Py_ssize_t num_found);
/* RaiseDoubleKeywords.proto */
static void __Pyx_RaiseDoubleKeywordsError(const char* func_name, PyObject* kw_name);
/* ParseKeywords.proto */
static int __Pyx_ParseOptionalKeywords(PyObject *kwds, PyObject **argnames[],\
PyObject *kwds2, PyObject *values[], Py_ssize_t num_pos_args,\
const char* function_name);
/* DictGetItem.proto */
#if PY_MAJOR_VERSION >= 3 && !CYTHON_COMPILING_IN_PYPY
static PyObject *__Pyx_PyDict_GetItem(PyObject *d, PyObject* key);
#define __Pyx_PyObject_Dict_GetItem(obj, name)\
(likely(PyDict_CheckExact(obj)) ?\
__Pyx_PyDict_GetItem(obj, name) : PyObject_GetItem(obj, name))
#else
#define __Pyx_PyDict_GetItem(d, key) PyObject_GetItem(d, key)
#define __Pyx_PyObject_Dict_GetItem(obj, name) PyObject_GetItem(obj, name)
#endif
/* None.proto */
static CYTHON_INLINE void __Pyx_RaiseUnboundLocalError(const char *varname);
/* GetItemInt.proto */
#define __Pyx_GetItemInt(o, i, type, is_signed, to_py_func, is_list, wraparound, boundscheck)\
(__Pyx_fits_Py_ssize_t(i, type, is_signed) ?\
__Pyx_GetItemInt_Fast(o, (Py_ssize_t)i, is_list, wraparound, boundscheck) :\
(is_list ? (PyErr_SetString(PyExc_IndexError, "list index out of range"), (PyObject*)NULL) :\
__Pyx_GetItemInt_Generic(o, to_py_func(i))))
#define __Pyx_GetItemInt_List(o, i, type, is_signed, to_py_func, is_list, wraparound, boundscheck)\
(__Pyx_fits_Py_ssize_t(i, type, is_signed) ?\
__Pyx_GetItemInt_List_Fast(o, (Py_ssize_t)i, wraparound, boundscheck) :\
(PyErr_SetString(PyExc_IndexError, "list index out of range"), (PyObject*)NULL))
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_List_Fast(PyObject *o, Py_ssize_t i,
int wraparound, int boundscheck);
#define __Pyx_GetItemInt_Tuple(o, i, type, is_signed, to_py_func, is_list, wraparound, boundscheck)\
(__Pyx_fits_Py_ssize_t(i, type, is_signed) ?\
__Pyx_GetItemInt_Tuple_Fast(o, (Py_ssize_t)i, wraparound, boundscheck) :\
(PyErr_SetString(PyExc_IndexError, "tuple index out of range"), (PyObject*)NULL))
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Tuple_Fast(PyObject *o, Py_ssize_t i,
int wraparound, int boundscheck);
static PyObject *__Pyx_GetItemInt_Generic(PyObject *o, PyObject* j);
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Fast(PyObject *o, Py_ssize_t i,
int is_list, int wraparound, int boundscheck);
/* ObjectGetItem.proto */
#if CYTHON_USE_TYPE_SLOTS
static CYTHON_INLINE PyObject *__Pyx_PyObject_GetItem(PyObject *obj, PyObject* key);
#else
#define __Pyx_PyObject_GetItem(obj, key) PyObject_GetItem(obj, key)
#endif
/* GetTopmostException.proto */
#if CYTHON_USE_EXC_INFO_STACK
static _PyErr_StackItem * __Pyx_PyErr_GetTopmostException(PyThreadState *tstate);
#endif
/* PyThreadStateGet.proto */
#if CYTHON_FAST_THREAD_STATE
#define __Pyx_PyThreadState_declare PyThreadState *__pyx_tstate;
#define __Pyx_PyThreadState_assign __pyx_tstate = __Pyx_PyThreadState_Current;
#define __Pyx_PyErr_Occurred() __pyx_tstate->curexc_type
#else
#define __Pyx_PyThreadState_declare
#define __Pyx_PyThreadState_assign
#define __Pyx_PyErr_Occurred() PyErr_Occurred()
#endif
/* SaveResetException.proto */
#if CYTHON_FAST_THREAD_STATE
#define __Pyx_ExceptionSave(type, value, tb) __Pyx__ExceptionSave(__pyx_tstate, type, value, tb)
static CYTHON_INLINE void __Pyx__ExceptionSave(PyThreadState *tstate, PyObject **type, PyObject **value, PyObject **tb);
#define __Pyx_ExceptionReset(type, value, tb) __Pyx__ExceptionReset(__pyx_tstate, type, value, tb)
static CYTHON_INLINE void __Pyx__ExceptionReset(PyThreadState *tstate, PyObject *type, PyObject *value, PyObject *tb);
#else
#define __Pyx_ExceptionSave(type, value, tb) PyErr_GetExcInfo(type, value, tb)
#define __Pyx_ExceptionReset(type, value, tb) PyErr_SetExcInfo(type, value, tb)
#endif
/* PyErrExceptionMatches.proto */
#if CYTHON_FAST_THREAD_STATE
#define __Pyx_PyErr_ExceptionMatches(err) __Pyx_PyErr_ExceptionMatchesInState(__pyx_tstate, err)
static CYTHON_INLINE int __Pyx_PyErr_ExceptionMatchesInState(PyThreadState* tstate, PyObject* err);
#else
#define __Pyx_PyErr_ExceptionMatches(err) PyErr_ExceptionMatches(err)
#endif
/* GetException.proto */
#if CYTHON_FAST_THREAD_STATE
#define __Pyx_GetException(type, value, tb) __Pyx__GetException(__pyx_tstate, type, value, tb)
static int __Pyx__GetException(PyThreadState *tstate, PyObject **type, PyObject **value, PyObject **tb);
#else
static int __Pyx_GetException(PyObject **type, PyObject **value, PyObject **tb);
#endif
/* PyErrFetchRestore.proto */
#if CYTHON_FAST_THREAD_STATE
#define __Pyx_PyErr_Clear() __Pyx_ErrRestore(NULL, NULL, NULL)
#define __Pyx_ErrRestoreWithState(type, value, tb) __Pyx_ErrRestoreInState(PyThreadState_GET(), type, value, tb)
#define __Pyx_ErrFetchWithState(type, value, tb) __Pyx_ErrFetchInState(PyThreadState_GET(), type, value, tb)
#define __Pyx_ErrRestore(type, value, tb) __Pyx_ErrRestoreInState(__pyx_tstate, type, value, tb)
#define __Pyx_ErrFetch(type, value, tb) __Pyx_ErrFetchInState(__pyx_tstate, type, value, tb)
static CYTHON_INLINE void __Pyx_ErrRestoreInState(PyThreadState *tstate, PyObject *type, PyObject *value, PyObject *tb);
static CYTHON_INLINE void __Pyx_ErrFetchInState(PyThreadState *tstate, PyObject **type, PyObject **value, PyObject **tb);
#if CYTHON_COMPILING_IN_CPYTHON
#define __Pyx_PyErr_SetNone(exc) (Py_INCREF(exc), __Pyx_ErrRestore((exc), NULL, NULL))
#else
#define __Pyx_PyErr_SetNone(exc) PyErr_SetNone(exc)
#endif
#else
#define __Pyx_PyErr_Clear() PyErr_Clear()
#define __Pyx_PyErr_SetNone(exc) PyErr_SetNone(exc)
#define __Pyx_ErrRestoreWithState(type, value, tb) PyErr_Restore(type, value, tb)
#define __Pyx_ErrFetchWithState(type, value, tb) PyErr_Fetch(type, value, tb)
#define __Pyx_ErrRestoreInState(tstate, type, value, tb) PyErr_Restore(type, value, tb)
#define __Pyx_ErrFetchInState(tstate, type, value, tb) PyErr_Fetch(type, value, tb)
#define __Pyx_ErrRestore(type, value, tb) PyErr_Restore(type, value, tb)
#define __Pyx_ErrFetch(type, value, tb) PyErr_Fetch(type, value, tb)
#endif
/* RaiseException.proto */
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause);
/* TypeImport.proto */
#ifndef __PYX_HAVE_RT_ImportType_proto_0_29_36
#define __PYX_HAVE_RT_ImportType_proto_0_29_36
#if __STDC_VERSION__ >= 201112L
#include
#endif
#if __STDC_VERSION__ >= 201112L || __cplusplus >= 201103L
#define __PYX_GET_STRUCT_ALIGNMENT_0_29_36(s) alignof(s)
#else
#define __PYX_GET_STRUCT_ALIGNMENT_0_29_36(s) sizeof(void*)
#endif
enum __Pyx_ImportType_CheckSize_0_29_36 {
__Pyx_ImportType_CheckSize_Error_0_29_36 = 0,
__Pyx_ImportType_CheckSize_Warn_0_29_36 = 1,
__Pyx_ImportType_CheckSize_Ignore_0_29_36 = 2
};
static PyTypeObject *__Pyx_ImportType_0_29_36(PyObject* module, const char *module_name, const char *class_name, size_t size, size_t alignment, enum __Pyx_ImportType_CheckSize_0_29_36 check_size);
#endif
/* Import.proto */
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, int level);
/* ImportFrom.proto */
static PyObject* __Pyx_ImportFrom(PyObject* module, PyObject* name);
/* CLineInTraceback.proto */
#ifdef CYTHON_CLINE_IN_TRACEBACK
#define __Pyx_CLineForTraceback(tstate, c_line) (((CYTHON_CLINE_IN_TRACEBACK)) ? c_line : 0)
#else
static int __Pyx_CLineForTraceback(PyThreadState *tstate, int c_line);
#endif
/* CodeObjectCache.proto */
typedef struct {
PyCodeObject* code_object;
int code_line;
} __Pyx_CodeObjectCacheEntry;
struct __Pyx_CodeObjectCache {
int count;
int max_count;
__Pyx_CodeObjectCacheEntry* entries;
};
static struct __Pyx_CodeObjectCache __pyx_code_cache = {0,0,NULL};
static int __pyx_bisect_code_objects(__Pyx_CodeObjectCacheEntry* entries, int count, int code_line);
static PyCodeObject *__pyx_find_code_object(int code_line);
static void __pyx_insert_code_object(int code_line, PyCodeObject* code_object);
/* AddTraceback.proto */
static void __Pyx_AddTraceback(const char *funcname, int c_line,
int py_line, const char *filename);
/* GCCDiagnostics.proto */
#if defined(__GNUC__) && (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 6))
#define __Pyx_HAS_GCC_DIAGNOSTIC
#endif
/* RealImag.proto */
#if CYTHON_CCOMPLEX
#ifdef __cplusplus
#define __Pyx_CREAL(z) ((z).real())
#define __Pyx_CIMAG(z) ((z).imag())
#else
#define __Pyx_CREAL(z) (__real__(z))
#define __Pyx_CIMAG(z) (__imag__(z))
#endif
#else
#define __Pyx_CREAL(z) ((z).real)
#define __Pyx_CIMAG(z) ((z).imag)
#endif
#if defined(__cplusplus) && CYTHON_CCOMPLEX\
&& (defined(_WIN32) || defined(__clang__) || (defined(__GNUC__) && (__GNUC__ >= 5 || __GNUC__ == 4 && __GNUC_MINOR__ >= 4 )) || __cplusplus >= 201103)
#define __Pyx_SET_CREAL(z,x) ((z).real(x))
#define __Pyx_SET_CIMAG(z,y) ((z).imag(y))
#else
#define __Pyx_SET_CREAL(z,x) __Pyx_CREAL(z) = (x)
#define __Pyx_SET_CIMAG(z,y) __Pyx_CIMAG(z) = (y)
#endif
/* Arithmetic.proto */
#if CYTHON_CCOMPLEX
#define __Pyx_c_eq_float(a, b) ((a)==(b))
#define __Pyx_c_sum_float(a, b) ((a)+(b))
#define __Pyx_c_diff_float(a, b) ((a)-(b))
#define __Pyx_c_prod_float(a, b) ((a)*(b))
#define __Pyx_c_quot_float(a, b) ((a)/(b))
#define __Pyx_c_neg_float(a) (-(a))
#ifdef __cplusplus
#define __Pyx_c_is_zero_float(z) ((z)==(float)0)
#define __Pyx_c_conj_float(z) (::std::conj(z))
#if 1
#define __Pyx_c_abs_float(z) (::std::abs(z))
#define __Pyx_c_pow_float(a, b) (::std::pow(a, b))
#endif
#else
#define __Pyx_c_is_zero_float(z) ((z)==0)
#define __Pyx_c_conj_float(z) (conjf(z))
#if 1
#define __Pyx_c_abs_float(z) (cabsf(z))
#define __Pyx_c_pow_float(a, b) (cpowf(a, b))
#endif
#endif
#else
static CYTHON_INLINE int __Pyx_c_eq_float(__pyx_t_float_complex, __pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_sum_float(__pyx_t_float_complex, __pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_diff_float(__pyx_t_float_complex, __pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_prod_float(__pyx_t_float_complex, __pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_quot_float(__pyx_t_float_complex, __pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_neg_float(__pyx_t_float_complex);
static CYTHON_INLINE int __Pyx_c_is_zero_float(__pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_conj_float(__pyx_t_float_complex);
#if 1
static CYTHON_INLINE float __Pyx_c_abs_float(__pyx_t_float_complex);
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_pow_float(__pyx_t_float_complex, __pyx_t_float_complex);
#endif
#endif
/* Arithmetic.proto */
#if CYTHON_CCOMPLEX
#define __Pyx_c_eq_double(a, b) ((a)==(b))
#define __Pyx_c_sum_double(a, b) ((a)+(b))
#define __Pyx_c_diff_double(a, b) ((a)-(b))
#define __Pyx_c_prod_double(a, b) ((a)*(b))
#define __Pyx_c_quot_double(a, b) ((a)/(b))
#define __Pyx_c_neg_double(a) (-(a))
#ifdef __cplusplus
#define __Pyx_c_is_zero_double(z) ((z)==(double)0)
#define __Pyx_c_conj_double(z) (::std::conj(z))
#if 1
#define __Pyx_c_abs_double(z) (::std::abs(z))
#define __Pyx_c_pow_double(a, b) (::std::pow(a, b))
#endif
#else
#define __Pyx_c_is_zero_double(z) ((z)==0)
#define __Pyx_c_conj_double(z) (conj(z))
#if 1
#define __Pyx_c_abs_double(z) (cabs(z))
#define __Pyx_c_pow_double(a, b) (cpow(a, b))
#endif
#endif
#else
static CYTHON_INLINE int __Pyx_c_eq_double(__pyx_t_double_complex, __pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_sum_double(__pyx_t_double_complex, __pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_diff_double(__pyx_t_double_complex, __pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_prod_double(__pyx_t_double_complex, __pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_quot_double(__pyx_t_double_complex, __pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_neg_double(__pyx_t_double_complex);
static CYTHON_INLINE int __Pyx_c_is_zero_double(__pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_conj_double(__pyx_t_double_complex);
#if 1
static CYTHON_INLINE double __Pyx_c_abs_double(__pyx_t_double_complex);
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_pow_double(__pyx_t_double_complex, __pyx_t_double_complex);
#endif
#endif
/* CIntToPy.proto */
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_int(int value);
/* CIntFromPy.proto */
static CYTHON_INLINE PY_LONG_LONG __Pyx_PyInt_As_PY_LONG_LONG(PyObject *);
/* CIntToPy.proto */
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_long(long value);
/* CIntFromPy.proto */
static CYTHON_INLINE int __Pyx_PyInt_As_int(PyObject *);
/* CIntFromPy.proto */
static CYTHON_INLINE unsigned PY_LONG_LONG __Pyx_PyInt_As_unsigned_PY_LONG_LONG(PyObject *);
/* CIntFromPy.proto */
static CYTHON_INLINE npy_uint32 __Pyx_PyInt_As_npy_uint32(PyObject *);
/* CIntFromPy.proto */
static CYTHON_INLINE long __Pyx_PyInt_As_long(PyObject *);
/* FastTypeChecks.proto */
#if CYTHON_COMPILING_IN_CPYTHON
#define __Pyx_TypeCheck(obj, type) __Pyx_IsSubtype(Py_TYPE(obj), (PyTypeObject *)type)
static CYTHON_INLINE int __Pyx_IsSubtype(PyTypeObject *a, PyTypeObject *b);
static CYTHON_INLINE int __Pyx_PyErr_GivenExceptionMatches(PyObject *err, PyObject *type);
static CYTHON_INLINE int __Pyx_PyErr_GivenExceptionMatches2(PyObject *err, PyObject *type1, PyObject *type2);
#else
#define __Pyx_TypeCheck(obj, type) PyObject_TypeCheck(obj, (PyTypeObject *)type)
#define __Pyx_PyErr_GivenExceptionMatches(err, type) PyErr_GivenExceptionMatches(err, type)
#define __Pyx_PyErr_GivenExceptionMatches2(err, type1, type2) (PyErr_GivenExceptionMatches(err, type1) || PyErr_GivenExceptionMatches(err, type2))
#endif
#define __Pyx_PyException_Check(obj) __Pyx_TypeCheck(obj, PyExc_Exception)
/* CheckBinaryVersion.proto */
static int __Pyx_check_binary_version(void);
/* FunctionExport.proto */
static int __Pyx_ExportFunction(const char *name, void (*f)(void), const char *sig);
/* VoidPtrImport.proto */
static int __Pyx_ImportVoidPtr_0_29_36(PyObject *module, const char *name, void **p, const char *sig);
/* FunctionImport.proto */
static int __Pyx_ImportFunction_0_29_36(PyObject *module, const char *funcname, void (**f)(void), const char *sig);
/* InitStrings.proto */
static int __Pyx_InitStrings(__Pyx_StringTabEntry *t);
/* Module declarations from 'cpython.buffer' */
/* Module declarations from 'libc.string' */
/* Module declarations from 'libc.stdio' */
/* Module declarations from '__builtin__' */
/* Module declarations from 'cpython.type' */
static PyTypeObject *__pyx_ptype_7cpython_4type_type = 0;
/* Module declarations from 'cpython' */
/* Module declarations from 'cpython.object' */
/* Module declarations from 'cpython.ref' */
/* Module declarations from 'cpython.mem' */
/* Module declarations from 'numpy' */
/* Module declarations from 'numpy' */
static PyTypeObject *__pyx_ptype_5numpy_dtype = 0;
static PyTypeObject *__pyx_ptype_5numpy_flatiter = 0;
static PyTypeObject *__pyx_ptype_5numpy_broadcast = 0;
static PyTypeObject *__pyx_ptype_5numpy_ndarray = 0;
static PyTypeObject *__pyx_ptype_5numpy_generic = 0;
static PyTypeObject *__pyx_ptype_5numpy_number = 0;
static PyTypeObject *__pyx_ptype_5numpy_integer = 0;
static PyTypeObject *__pyx_ptype_5numpy_signedinteger = 0;
static PyTypeObject *__pyx_ptype_5numpy_unsignedinteger = 0;
static PyTypeObject *__pyx_ptype_5numpy_inexact = 0;
static PyTypeObject *__pyx_ptype_5numpy_floating = 0;
static PyTypeObject *__pyx_ptype_5numpy_complexfloating = 0;
static PyTypeObject *__pyx_ptype_5numpy_flexible = 0;
static PyTypeObject *__pyx_ptype_5numpy_character = 0;
static PyTypeObject *__pyx_ptype_5numpy_ufunc = 0;
/* Module declarations from 'gensim.models.word2vec_inner' */
static __pyx_t_6gensim_6models_14word2vec_inner_scopy_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_scopy = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_scopy (*__pyx_vp_6gensim_6models_14word2vec_inner_scopy)
static __pyx_t_6gensim_6models_14word2vec_inner_saxpy_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_saxpy = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_saxpy (*__pyx_vp_6gensim_6models_14word2vec_inner_saxpy)
static __pyx_t_6gensim_6models_14word2vec_inner_sdot_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_sdot = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_sdot (*__pyx_vp_6gensim_6models_14word2vec_inner_sdot)
static __pyx_t_6gensim_6models_14word2vec_inner_dsdot_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_dsdot = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_dsdot (*__pyx_vp_6gensim_6models_14word2vec_inner_dsdot)
static __pyx_t_6gensim_6models_14word2vec_inner_snrm2_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_snrm2 = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_snrm2 (*__pyx_vp_6gensim_6models_14word2vec_inner_snrm2)
static __pyx_t_6gensim_6models_14word2vec_inner_sscal_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_sscal = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_sscal (*__pyx_vp_6gensim_6models_14word2vec_inner_sscal)
static __pyx_t_6gensim_6models_14word2vec_inner_REAL_t (*__pyx_vp_6gensim_6models_14word2vec_inner_EXP_TABLE)[0x3E8] = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE (*__pyx_vp_6gensim_6models_14word2vec_inner_EXP_TABLE)
static __pyx_t_6gensim_6models_14word2vec_inner_our_dot_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_our_dot = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_our_dot (*__pyx_vp_6gensim_6models_14word2vec_inner_our_dot)
static __pyx_t_6gensim_6models_14word2vec_inner_our_saxpy_ptr *__pyx_vp_6gensim_6models_14word2vec_inner_our_saxpy = 0;
#define __pyx_v_6gensim_6models_14word2vec_inner_our_saxpy (*__pyx_vp_6gensim_6models_14word2vec_inner_our_saxpy)
static unsigned PY_LONG_LONG (*__pyx_f_6gensim_6models_14word2vec_inner_bisect_left)(__pyx_t_5numpy_uint32_t *, unsigned PY_LONG_LONG, unsigned PY_LONG_LONG, unsigned PY_LONG_LONG); /*proto*/
static unsigned PY_LONG_LONG (*__pyx_f_6gensim_6models_14word2vec_inner_random_int32)(unsigned PY_LONG_LONG *); /*proto*/
/* Module declarations from 'cython' */
/* Module declarations from 'gensim.models.doc2vec_inner' */
static int __pyx_v_6gensim_6models_13doc2vec_inner_ONE;
static __pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_6gensim_6models_13doc2vec_inner_ONEF;
static void __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_hs(__pyx_t_5numpy_uint32_t const *, __pyx_t_5numpy_uint8_t const *, int const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , __pyx_t_5numpy_uint32_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int, int, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_5numpy_uint32_t const ); /*proto*/
static unsigned PY_LONG_LONG __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_neg(int const , __pyx_t_5numpy_uint32_t *, unsigned PY_LONG_LONG, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , __pyx_t_5numpy_uint32_t const , __pyx_t_5numpy_uint32_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, unsigned PY_LONG_LONG, int, int, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_5numpy_uint32_t const ); /*proto*/
static void __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dm_hs(__pyx_t_5numpy_uint32_t const *, __pyx_t_5numpy_uint8_t const *, int, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , int); /*proto*/
static unsigned PY_LONG_LONG __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dm_neg(int const , __pyx_t_5numpy_uint32_t *, unsigned PY_LONG_LONG, unsigned PY_LONG_LONG, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , int); /*proto*/
static void __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dmc_hs(__pyx_t_5numpy_uint32_t const *, __pyx_t_5numpy_uint8_t const *, int, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , int const , int); /*proto*/
static unsigned PY_LONG_LONG __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dmc_neg(int const , __pyx_t_5numpy_uint32_t *, unsigned PY_LONG_LONG, unsigned PY_LONG_LONG, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const , __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *, int const , int const , int); /*proto*/
static PyObject *__pyx_f_6gensim_6models_13doc2vec_inner_init_d2v_config(struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig *, PyObject *, PyObject *, PyObject *, PyObject *, PyObject *, struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config *__pyx_optional_args); /*proto*/
#define __Pyx_MODULE_NAME "gensim.models.doc2vec_inner"
extern int __pyx_module_is_main_gensim__models__doc2vec_inner;
int __pyx_module_is_main_gensim__models__doc2vec_inner = 0;
/* Implementation of 'gensim.models.doc2vec_inner' */
static PyObject *__pyx_builtin_ImportError;
static PyObject *__pyx_builtin_range;
static PyObject *__pyx_builtin_enumerate;
static const char __pyx_k_[] = "\000";
static const char __pyx_k_c[] = "c";
static const char __pyx_k_i[] = "i";
static const char __pyx_k_j[] = "j";
static const char __pyx_k_k[] = "k";
static const char __pyx_k_m[] = "m";
static const char __pyx_k_n[] = "n";
static const char __pyx_k__5[] = "*";
static const char __pyx_k_dv[] = "dv";
static const char __pyx_k_hs[] = "hs";
static const char __pyx_k_np[] = "np";
static const char __pyx_k_wv[] = "wv";
static const char __pyx_k_get[] = "get";
static const char __pyx_k_REAL[] = "REAL";
static const char __pyx_k_code[] = "code";
static const char __pyx_k_item[] = "item";
static const char __pyx_k_main[] = "__main__";
static const char __pyx_k_name[] = "__name__";
static const char __pyx_k_neu1[] = "neu1";
static const char __pyx_k_syn1[] = "syn1";
static const char __pyx_k_test[] = "__test__";
static const char __pyx_k_work[] = "work";
static const char __pyx_k_alpha[] = "alpha";
static const char __pyx_k_count[] = "count";
static const char __pyx_k_dtype[] = "dtype";
static const char __pyx_k_fblas[] = "fblas";
static const char __pyx_k_model[] = "model";
static const char __pyx_k_numpy[] = "numpy";
static const char __pyx_k_point[] = "point";
static const char __pyx_k_range[] = "range";
static const char __pyx_k_token[] = "token";
static const char __pyx_k_zeros[] = "zeros";
static const char __pyx_k_import[] = "__import__";
static const char __pyx_k_random[] = "random";
static const char __pyx_k_result[] = "result";
static const char __pyx_k_sample[] = "sample";
static const char __pyx_k_window[] = "window";
static const char __pyx_k_float32[] = "float32";
static const char __pyx_k_randint[] = "randint";
static const char __pyx_k_syn1neg[] = "syn1neg";
static const char __pyx_k_vectors[] = "vectors";
static const char __pyx_k_workers[] = "workers";
static const char __pyx_k_expandos[] = "expandos";
static const char __pyx_k_negative[] = "negative";
static const char __pyx_k_cbow_mean[] = "cbow_mean";
static const char __pyx_k_cum_table[] = "cum_table";
static const char __pyx_k_doc_words[] = "doc_words";
static const char __pyx_k_enumerate[] = "enumerate";
static const char __pyx_k_get_index[] = "get_index";
static const char __pyx_k_inv_count[] = "inv_count";
static const char __pyx_k_sample_int[] = "sample_int";
static const char __pyx_k_word_index[] = "word_index";
static const char __pyx_k_ImportError[] = "ImportError";
static const char __pyx_k_layer1_size[] = "layer1_size";
static const char __pyx_k_learn_words[] = "learn_words";
static const char __pyx_k_train_words[] = "train_words";
static const char __pyx_k_vector_size[] = "vector_size";
static const char __pyx_k_vocab_codes[] = "vocab_codes";
static const char __pyx_k_words_lockf[] = "words_lockf";
static const char __pyx_k_dm_tag_count[] = "dm_tag_count";
static const char __pyx_k_key_to_index[] = "key_to_index";
static const char __pyx_k_learn_hidden[] = "learn_hidden";
static const char __pyx_k_vocab_points[] = "vocab_points";
static const char __pyx_k_word_vectors[] = "word_vectors";
static const char __pyx_k_doctags_lockf[] = "doctags_lockf";
static const char __pyx_k_learn_doctags[] = "learn_doctags";
static const char __pyx_k_vectors_lockf[] = "vectors_lockf";
static const char __pyx_k_doctag_indexes[] = "doctag_indexes";
static const char __pyx_k_doctag_vectors[] = "doctag_vectors";
static const char __pyx_k_shrink_windows[] = "shrink_windows";
static const char __pyx_k_scipy_linalg_blas[] = "scipy.linalg.blas";
static const char __pyx_k_train_document_dm[] = "train_document_dm";
static const char __pyx_k_vocab_sample_ints[] = "vocab_sample_ints";
static const char __pyx_k_cline_in_traceback[] = "cline_in_traceback";
static const char __pyx_k_train_document_dbow[] = "train_document_dbow";
static const char __pyx_k_train_document_dm_concat[] = "train_document_dm_concat";
static const char __pyx_k_gensim_models_doc2vec_inner[] = "gensim.models.doc2vec_inner";
static const char __pyx_k_gensim_models_doc2vec_inner_pyx[] = "gensim/models/doc2vec_inner.pyx";
static const char __pyx_k_numpy_core_multiarray_failed_to[] = "numpy.core.multiarray failed to import";
static const char __pyx_k_Optimized_cython_functions_for_t[] = "Optimized cython functions for training :class:`~gensim.models.doc2vec.Doc2Vec` model.";
static const char __pyx_k_numpy_core_umath_failed_to_impor[] = "numpy.core.umath failed to import";
static PyObject *__pyx_kp_u_;
static PyObject *__pyx_n_s_ImportError;
static PyObject *__pyx_n_s_REAL;
static PyObject *__pyx_n_s__5;
static PyObject *__pyx_n_s_alpha;
static PyObject *__pyx_n_s_c;
static PyObject *__pyx_n_s_cbow_mean;
static PyObject *__pyx_n_s_cline_in_traceback;
static PyObject *__pyx_n_u_code;
static PyObject *__pyx_n_s_count;
static PyObject *__pyx_n_s_cum_table;
static PyObject *__pyx_n_s_dm_tag_count;
static PyObject *__pyx_n_s_doc_words;
static PyObject *__pyx_n_s_doctag_indexes;
static PyObject *__pyx_n_s_doctag_vectors;
static PyObject *__pyx_n_s_doctags_lockf;
static PyObject *__pyx_n_s_dtype;
static PyObject *__pyx_n_s_dv;
static PyObject *__pyx_n_s_enumerate;
static PyObject *__pyx_n_s_expandos;
static PyObject *__pyx_n_s_fblas;
static PyObject *__pyx_n_s_float32;
static PyObject *__pyx_n_s_gensim_models_doc2vec_inner;
static PyObject *__pyx_kp_s_gensim_models_doc2vec_inner_pyx;
static PyObject *__pyx_n_s_get;
static PyObject *__pyx_n_s_get_index;
static PyObject *__pyx_n_s_hs;
static PyObject *__pyx_n_s_i;
static PyObject *__pyx_n_s_import;
static PyObject *__pyx_n_s_inv_count;
static PyObject *__pyx_n_s_item;
static PyObject *__pyx_n_s_j;
static PyObject *__pyx_n_s_k;
static PyObject *__pyx_n_s_key_to_index;
static PyObject *__pyx_n_s_layer1_size;
static PyObject *__pyx_n_s_learn_doctags;
static PyObject *__pyx_n_s_learn_hidden;
static PyObject *__pyx_n_s_learn_words;
static PyObject *__pyx_n_s_m;
static PyObject *__pyx_n_s_main;
static PyObject *__pyx_n_s_model;
static PyObject *__pyx_n_s_n;
static PyObject *__pyx_n_s_name;
static PyObject *__pyx_n_s_negative;
static PyObject *__pyx_n_s_neu1;
static PyObject *__pyx_n_s_np;
static PyObject *__pyx_n_s_numpy;
static PyObject *__pyx_kp_u_numpy_core_multiarray_failed_to;
static PyObject *__pyx_kp_u_numpy_core_umath_failed_to_impor;
static PyObject *__pyx_n_u_point;
static PyObject *__pyx_n_s_randint;
static PyObject *__pyx_n_s_random;
static PyObject *__pyx_n_s_range;
static PyObject *__pyx_n_s_result;
static PyObject *__pyx_n_s_sample;
static PyObject *__pyx_n_u_sample_int;
static PyObject *__pyx_n_s_scipy_linalg_blas;
static PyObject *__pyx_n_s_shrink_windows;
static PyObject *__pyx_n_s_syn1;
static PyObject *__pyx_n_s_syn1neg;
static PyObject *__pyx_n_s_test;
static PyObject *__pyx_n_s_token;
static PyObject *__pyx_n_s_train_document_dbow;
static PyObject *__pyx_n_s_train_document_dm;
static PyObject *__pyx_n_s_train_document_dm_concat;
static PyObject *__pyx_n_s_train_words;
static PyObject *__pyx_n_s_vector_size;
static PyObject *__pyx_n_s_vectors;
static PyObject *__pyx_n_s_vectors_lockf;
static PyObject *__pyx_n_s_vocab_codes;
static PyObject *__pyx_n_s_vocab_points;
static PyObject *__pyx_n_s_vocab_sample_ints;
static PyObject *__pyx_n_s_window;
static PyObject *__pyx_n_s_word_index;
static PyObject *__pyx_n_s_word_vectors;
static PyObject *__pyx_n_s_words_lockf;
static PyObject *__pyx_n_s_work;
static PyObject *__pyx_n_s_workers;
static PyObject *__pyx_n_s_wv;
static PyObject *__pyx_n_s_zeros;
static PyObject *__pyx_pf_6gensim_6models_13doc2vec_inner_train_document_dbow(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_model, PyObject *__pyx_v_doc_words, PyObject *__pyx_v_doctag_indexes, PyObject *__pyx_v_alpha, PyObject *__pyx_v_work, PyObject *__pyx_v_train_words, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, PyObject *__pyx_v_word_vectors, PyObject *__pyx_v_words_lockf, PyObject *__pyx_v_doctag_vectors, PyObject *__pyx_v_doctags_lockf); /* proto */
static PyObject *__pyx_pf_6gensim_6models_13doc2vec_inner_2train_document_dm(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_model, PyObject *__pyx_v_doc_words, PyObject *__pyx_v_doctag_indexes, PyObject *__pyx_v_alpha, PyObject *__pyx_v_work, PyObject *__pyx_v_neu1, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, PyObject *__pyx_v_word_vectors, PyObject *__pyx_v_words_lockf, PyObject *__pyx_v_doctag_vectors, PyObject *__pyx_v_doctags_lockf); /* proto */
static PyObject *__pyx_pf_6gensim_6models_13doc2vec_inner_4train_document_dm_concat(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_model, PyObject *__pyx_v_doc_words, PyObject *__pyx_v_doctag_indexes, PyObject *__pyx_v_alpha, PyObject *__pyx_v_work, PyObject *__pyx_v_neu1, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, PyObject *__pyx_v_word_vectors, PyObject *__pyx_v_words_lockf, PyObject *__pyx_v_doctag_vectors, PyObject *__pyx_v_doctags_lockf); /* proto */
static PyObject *__pyx_int_0;
static PyObject *__pyx_int_16777216;
static PyObject *__pyx_tuple__2;
static PyObject *__pyx_tuple__3;
static PyObject *__pyx_tuple__4;
static PyObject *__pyx_tuple__6;
static PyObject *__pyx_tuple__8;
static PyObject *__pyx_tuple__10;
static PyObject *__pyx_codeobj__7;
static PyObject *__pyx_codeobj__9;
static PyObject *__pyx_codeobj__11;
/* Late includes */
/* "gensim/models/doc2vec_inner.pyx":38
* DEF MAX_EXP = 6
*
* cdef void fast_document_dbow_hs( # <<<<<<<<<<<<<<
* const np.uint32_t *word_point, const np.uint8_t *word_code, const int codelen,
* REAL_t *context_vectors, REAL_t *syn1, const int size,
*/
static void __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_hs(__pyx_t_5numpy_uint32_t const *__pyx_v_word_point, __pyx_t_5numpy_uint8_t const *__pyx_v_word_code, int const __pyx_v_codelen, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_context_vectors, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_syn1, int const __pyx_v_size, __pyx_t_5numpy_uint32_t const __pyx_v_context_index, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const __pyx_v_alpha, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_work, int __pyx_v_learn_context, int __pyx_v_learn_hidden, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_contexts_lockf, __pyx_t_5numpy_uint32_t const __pyx_v_contexts_lockf_len) {
PY_LONG_LONG __pyx_v_b;
PY_LONG_LONG __pyx_v_row1;
PY_LONG_LONG __pyx_v_row2;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_f;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_g;
int __pyx_t_1;
int __pyx_t_2;
PY_LONG_LONG __pyx_t_3;
int __pyx_t_4;
int __pyx_t_5;
/* "gensim/models/doc2vec_inner.pyx":45
*
* cdef long long a, b
* cdef long long row1 = context_index * size, row2 # <<<<<<<<<<<<<<
* cdef REAL_t f, g
*
*/
__pyx_v_row1 = (__pyx_v_context_index * __pyx_v_size);
/* "gensim/models/doc2vec_inner.pyx":48
* cdef REAL_t f, g
*
* memset(work, 0, size * cython.sizeof(REAL_t)) # <<<<<<<<<<<<<<
* for b in range(codelen):
* row2 = word_point[b] * size
*/
(void)(memset(__pyx_v_work, 0, (__pyx_v_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
/* "gensim/models/doc2vec_inner.pyx":49
*
* memset(work, 0, size * cython.sizeof(REAL_t))
* for b in range(codelen): # <<<<<<<<<<<<<<
* row2 = word_point[b] * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
*/
__pyx_t_1 = __pyx_v_codelen;
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_b = __pyx_t_3;
/* "gensim/models/doc2vec_inner.pyx":50
* memset(work, 0, size * cython.sizeof(REAL_t))
* for b in range(codelen):
* row2 = word_point[b] * size # <<<<<<<<<<<<<<
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
*/
__pyx_v_row2 = ((__pyx_v_word_point[__pyx_v_b]) * __pyx_v_size);
/* "gensim/models/doc2vec_inner.pyx":51
* for b in range(codelen):
* row2 = word_point[b] * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1[row2], &ONE) # <<<<<<<<<<<<<<
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
*/
__pyx_v_f = __pyx_v_6gensim_6models_14word2vec_inner_our_dot((&__pyx_v_size), (&(__pyx_v_context_vectors[__pyx_v_row1])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":52
* row2 = word_point[b] * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
__pyx_t_5 = ((__pyx_v_f <= -6.0) != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_4 = __pyx_t_5;
goto __pyx_L6_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_f >= 6.0) != 0);
__pyx_t_4 = __pyx_t_5;
__pyx_L6_bool_binop_done:;
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":53
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue # <<<<<<<<<<<<<<
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":52
* row2 = word_point[b] * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
}
/* "gensim/models/doc2vec_inner.pyx":54
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))] # <<<<<<<<<<<<<<
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
*/
__pyx_v_f = (__pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE[((int)((__pyx_v_f + 6.0) * ((0x3E8 / 6) / 2)))]);
/* "gensim/models/doc2vec_inner.pyx":55
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden:
*/
__pyx_v_g = (((1 - (__pyx_v_word_code[__pyx_v_b])) - __pyx_v_f) * __pyx_v_alpha);
/* "gensim/models/doc2vec_inner.pyx":56
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE) # <<<<<<<<<<<<<<
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":57
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if learn_context:
*/
__pyx_t_4 = (__pyx_v_learn_hidden != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":58
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE) # <<<<<<<<<<<<<<
* if learn_context:
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len],
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), (&(__pyx_v_context_vectors[__pyx_v_row1])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":57
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if learn_context:
*/
}
__pyx_L3_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":59
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if learn_context: # <<<<<<<<<<<<<<
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len],
* work, &ONE, &context_vectors[row1], &ONE)
*/
__pyx_t_4 = (__pyx_v_learn_context != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":60
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if learn_context:
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len], # <<<<<<<<<<<<<<
* work, &ONE, &context_vectors[row1], &ONE)
*
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&(__pyx_v_contexts_lockf[(__pyx_v_context_index % __pyx_v_contexts_lockf_len)])), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_context_vectors[__pyx_v_row1])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":59
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1[row2], &ONE)
* if learn_context: # <<<<<<<<<<<<<<
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len],
* work, &ONE, &context_vectors[row1], &ONE)
*/
}
/* "gensim/models/doc2vec_inner.pyx":38
* DEF MAX_EXP = 6
*
* cdef void fast_document_dbow_hs( # <<<<<<<<<<<<<<
* const np.uint32_t *word_point, const np.uint8_t *word_code, const int codelen,
* REAL_t *context_vectors, REAL_t *syn1, const int size,
*/
/* function exit code */
}
/* "gensim/models/doc2vec_inner.pyx":64
*
*
* cdef unsigned long long fast_document_dbow_neg( # <<<<<<<<<<<<<<
* const int negative, np.uint32_t *cum_table, unsigned long long cum_table_len,
* REAL_t *context_vectors, REAL_t *syn1neg, const int size, const np.uint32_t word_index,
*/
static unsigned PY_LONG_LONG __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_neg(int const __pyx_v_negative, __pyx_t_5numpy_uint32_t *__pyx_v_cum_table, unsigned PY_LONG_LONG __pyx_v_cum_table_len, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_context_vectors, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_syn1neg, int const __pyx_v_size, __pyx_t_5numpy_uint32_t const __pyx_v_word_index, __pyx_t_5numpy_uint32_t const __pyx_v_context_index, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const __pyx_v_alpha, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_work, unsigned PY_LONG_LONG __pyx_v_next_random, int __pyx_v_learn_context, int __pyx_v_learn_hidden, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_contexts_lockf, __pyx_t_5numpy_uint32_t const __pyx_v_contexts_lockf_len) {
PY_LONG_LONG __pyx_v_row1;
PY_LONG_LONG __pyx_v_row2;
unsigned PY_LONG_LONG __pyx_v_modulo;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_f;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_g;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_label;
__pyx_t_5numpy_uint32_t __pyx_v_target_index;
int __pyx_v_d;
unsigned PY_LONG_LONG __pyx_r;
long __pyx_t_1;
long __pyx_t_2;
int __pyx_t_3;
int __pyx_t_4;
int __pyx_t_5;
/* "gensim/models/doc2vec_inner.pyx":72
*
* cdef long long a
* cdef long long row1 = context_index * size, row2 # <<<<<<<<<<<<<<
* cdef unsigned long long modulo = 281474976710655ULL
* cdef REAL_t f, g, label
*/
__pyx_v_row1 = (__pyx_v_context_index * __pyx_v_size);
/* "gensim/models/doc2vec_inner.pyx":73
* cdef long long a
* cdef long long row1 = context_index * size, row2
* cdef unsigned long long modulo = 281474976710655ULL # <<<<<<<<<<<<<<
* cdef REAL_t f, g, label
* cdef np.uint32_t target_index
*/
__pyx_v_modulo = 281474976710655ULL;
/* "gensim/models/doc2vec_inner.pyx":78
* cdef int d
*
* memset(work, 0, size * cython.sizeof(REAL_t)) # <<<<<<<<<<<<<<
*
* for d in range(negative+1):
*/
(void)(memset(__pyx_v_work, 0, (__pyx_v_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
/* "gensim/models/doc2vec_inner.pyx":80
* memset(work, 0, size * cython.sizeof(REAL_t))
*
* for d in range(negative+1): # <<<<<<<<<<<<<<
* if d == 0:
* target_index = word_index
*/
__pyx_t_1 = (__pyx_v_negative + 1);
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_d = __pyx_t_3;
/* "gensim/models/doc2vec_inner.pyx":81
*
* for d in range(negative+1):
* if d == 0: # <<<<<<<<<<<<<<
* target_index = word_index
* label = ONEF
*/
__pyx_t_4 = ((__pyx_v_d == 0) != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":82
* for d in range(negative+1):
* if d == 0:
* target_index = word_index # <<<<<<<<<<<<<<
* label = ONEF
* else:
*/
__pyx_v_target_index = __pyx_v_word_index;
/* "gensim/models/doc2vec_inner.pyx":83
* if d == 0:
* target_index = word_index
* label = ONEF # <<<<<<<<<<<<<<
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
*/
__pyx_v_label = __pyx_v_6gensim_6models_13doc2vec_inner_ONEF;
/* "gensim/models/doc2vec_inner.pyx":81
*
* for d in range(negative+1):
* if d == 0: # <<<<<<<<<<<<<<
* target_index = word_index
* label = ONEF
*/
goto __pyx_L5;
}
/* "gensim/models/doc2vec_inner.pyx":85
* label = ONEF
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len) # <<<<<<<<<<<<<<
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == word_index:
*/
/*else*/ {
__pyx_v_target_index = __pyx_f_6gensim_6models_14word2vec_inner_bisect_left(__pyx_v_cum_table, ((__pyx_v_next_random >> 16) % (__pyx_v_cum_table[(__pyx_v_cum_table_len - 1)])), 0, __pyx_v_cum_table_len);
/* "gensim/models/doc2vec_inner.pyx":86
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo # <<<<<<<<<<<<<<
* if target_index == word_index:
* continue
*/
__pyx_v_next_random = (((__pyx_v_next_random * ((unsigned PY_LONG_LONG)25214903917ULL)) + 11) & __pyx_v_modulo);
/* "gensim/models/doc2vec_inner.pyx":87
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == word_index: # <<<<<<<<<<<<<<
* continue
* label = 0.0
*/
__pyx_t_4 = ((__pyx_v_target_index == __pyx_v_word_index) != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":88
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == word_index:
* continue # <<<<<<<<<<<<<<
* label = 0.0
* row2 = target_index * size
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":87
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == word_index: # <<<<<<<<<<<<<<
* continue
* label = 0.0
*/
}
/* "gensim/models/doc2vec_inner.pyx":89
* if target_index == word_index:
* continue
* label = 0.0 # <<<<<<<<<<<<<<
* row2 = target_index * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
*/
__pyx_v_label = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)0.0);
}
__pyx_L5:;
/* "gensim/models/doc2vec_inner.pyx":90
* continue
* label = 0.0
* row2 = target_index * size # <<<<<<<<<<<<<<
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
*/
__pyx_v_row2 = (__pyx_v_target_index * __pyx_v_size);
/* "gensim/models/doc2vec_inner.pyx":91
* label = 0.0
* row2 = target_index * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE) # <<<<<<<<<<<<<<
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
*/
__pyx_v_f = __pyx_v_6gensim_6models_14word2vec_inner_our_dot((&__pyx_v_size), (&(__pyx_v_context_vectors[__pyx_v_row1])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":92
* row2 = target_index * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
__pyx_t_5 = ((__pyx_v_f <= -6.0) != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_4 = __pyx_t_5;
goto __pyx_L8_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_f >= 6.0) != 0);
__pyx_t_4 = __pyx_t_5;
__pyx_L8_bool_binop_done:;
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":93
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue # <<<<<<<<<<<<<<
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":92
* row2 = target_index * size
* f = our_dot(&size, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
}
/* "gensim/models/doc2vec_inner.pyx":94
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))] # <<<<<<<<<<<<<<
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
*/
__pyx_v_f = (__pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE[((int)((__pyx_v_f + 6.0) * ((0x3E8 / 6) / 2)))]);
/* "gensim/models/doc2vec_inner.pyx":95
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden:
*/
__pyx_v_g = ((__pyx_v_label - __pyx_v_f) * __pyx_v_alpha);
/* "gensim/models/doc2vec_inner.pyx":96
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE) # <<<<<<<<<<<<<<
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":97
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if learn_context:
*/
__pyx_t_4 = (__pyx_v_learn_hidden != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":98
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE) # <<<<<<<<<<<<<<
* if learn_context:
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len],
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), (&(__pyx_v_context_vectors[__pyx_v_row1])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":97
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if learn_context:
*/
}
__pyx_L3_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":99
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if learn_context: # <<<<<<<<<<<<<<
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len],
* work, &ONE, &context_vectors[row1], &ONE)
*/
__pyx_t_4 = (__pyx_v_learn_context != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":100
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if learn_context:
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len], # <<<<<<<<<<<<<<
* work, &ONE, &context_vectors[row1], &ONE)
*
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&(__pyx_v_contexts_lockf[(__pyx_v_context_index % __pyx_v_contexts_lockf_len)])), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_context_vectors[__pyx_v_row1])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":99
* if learn_hidden:
* our_saxpy(&size, &g, &context_vectors[row1], &ONE, &syn1neg[row2], &ONE)
* if learn_context: # <<<<<<<<<<<<<<
* our_saxpy(&size, &contexts_lockf[context_index % contexts_lockf_len],
* work, &ONE, &context_vectors[row1], &ONE)
*/
}
/* "gensim/models/doc2vec_inner.pyx":103
* work, &ONE, &context_vectors[row1], &ONE)
*
* return next_random # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = __pyx_v_next_random;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":64
*
*
* cdef unsigned long long fast_document_dbow_neg( # <<<<<<<<<<<<<<
* const int negative, np.uint32_t *cum_table, unsigned long long cum_table_len,
* REAL_t *context_vectors, REAL_t *syn1neg, const int size, const np.uint32_t word_index,
*/
/* function exit code */
__pyx_L0:;
return __pyx_r;
}
/* "gensim/models/doc2vec_inner.pyx":106
*
*
* cdef void fast_document_dm_hs( # <<<<<<<<<<<<<<
* const np.uint32_t *word_point, const np.uint8_t *word_code, int word_code_len,
* REAL_t *neu1, REAL_t *syn1, const REAL_t alpha, REAL_t *work,
*/
static void __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dm_hs(__pyx_t_5numpy_uint32_t const *__pyx_v_word_point, __pyx_t_5numpy_uint8_t const *__pyx_v_word_code, int __pyx_v_word_code_len, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_neu1, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_syn1, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const __pyx_v_alpha, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_work, int const __pyx_v_size, int __pyx_v_learn_hidden) {
PY_LONG_LONG __pyx_v_b;
PY_LONG_LONG __pyx_v_row2;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_f;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_g;
int __pyx_t_1;
int __pyx_t_2;
PY_LONG_LONG __pyx_t_3;
int __pyx_t_4;
int __pyx_t_5;
/* "gensim/models/doc2vec_inner.pyx":117
* # l1 already composed by caller, passed in as neu1
* # work (also passed in) will accumulate l1 error
* for b in range(word_code_len): # <<<<<<<<<<<<<<
* row2 = word_point[b] * size
* f = our_dot(&size, neu1, &ONE, &syn1[row2], &ONE)
*/
__pyx_t_1 = __pyx_v_word_code_len;
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_b = __pyx_t_3;
/* "gensim/models/doc2vec_inner.pyx":118
* # work (also passed in) will accumulate l1 error
* for b in range(word_code_len):
* row2 = word_point[b] * size # <<<<<<<<<<<<<<
* f = our_dot(&size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
*/
__pyx_v_row2 = ((__pyx_v_word_point[__pyx_v_b]) * __pyx_v_size);
/* "gensim/models/doc2vec_inner.pyx":119
* for b in range(word_code_len):
* row2 = word_point[b] * size
* f = our_dot(&size, neu1, &ONE, &syn1[row2], &ONE) # <<<<<<<<<<<<<<
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
*/
__pyx_v_f = __pyx_v_6gensim_6models_14word2vec_inner_our_dot((&__pyx_v_size), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":120
* row2 = word_point[b] * size
* f = our_dot(&size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
__pyx_t_5 = ((__pyx_v_f <= -6.0) != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_4 = __pyx_t_5;
goto __pyx_L6_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_f >= 6.0) != 0);
__pyx_t_4 = __pyx_t_5;
__pyx_L6_bool_binop_done:;
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":121
* f = our_dot(&size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue # <<<<<<<<<<<<<<
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":120
* row2 = word_point[b] * size
* f = our_dot(&size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
}
/* "gensim/models/doc2vec_inner.pyx":122
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))] # <<<<<<<<<<<<<<
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
*/
__pyx_v_f = (__pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE[((int)((__pyx_v_f + 6.0) * ((0x3E8 / 6) / 2)))]);
/* "gensim/models/doc2vec_inner.pyx":123
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden:
*/
__pyx_v_g = (((1 - (__pyx_v_word_code[__pyx_v_b])) - __pyx_v_f) * __pyx_v_alpha);
/* "gensim/models/doc2vec_inner.pyx":124
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE) # <<<<<<<<<<<<<<
* if learn_hidden:
* our_saxpy(&size, &g, neu1, &ONE, &syn1[row2], &ONE)
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":125
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, neu1, &ONE, &syn1[row2], &ONE)
*
*/
__pyx_t_4 = (__pyx_v_learn_hidden != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":126
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden:
* our_saxpy(&size, &g, neu1, &ONE, &syn1[row2], &ONE) # <<<<<<<<<<<<<<
*
*
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":125
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, neu1, &ONE, &syn1[row2], &ONE)
*
*/
}
__pyx_L3_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":106
*
*
* cdef void fast_document_dm_hs( # <<<<<<<<<<<<<<
* const np.uint32_t *word_point, const np.uint8_t *word_code, int word_code_len,
* REAL_t *neu1, REAL_t *syn1, const REAL_t alpha, REAL_t *work,
*/
/* function exit code */
}
/* "gensim/models/doc2vec_inner.pyx":129
*
*
* cdef unsigned long long fast_document_dm_neg( # <<<<<<<<<<<<<<
* const int negative, np.uint32_t *cum_table, unsigned long long cum_table_len, unsigned long long next_random,
* REAL_t *neu1, REAL_t *syn1neg, const int predict_word_index, const REAL_t alpha, REAL_t *work,
*/
static unsigned PY_LONG_LONG __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dm_neg(int const __pyx_v_negative, __pyx_t_5numpy_uint32_t *__pyx_v_cum_table, unsigned PY_LONG_LONG __pyx_v_cum_table_len, unsigned PY_LONG_LONG __pyx_v_next_random, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_neu1, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_syn1neg, int const __pyx_v_predict_word_index, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const __pyx_v_alpha, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_work, int const __pyx_v_size, int __pyx_v_learn_hidden) {
PY_LONG_LONG __pyx_v_row2;
unsigned PY_LONG_LONG __pyx_v_modulo;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_f;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_g;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_label;
__pyx_t_5numpy_uint32_t __pyx_v_target_index;
int __pyx_v_d;
unsigned PY_LONG_LONG __pyx_r;
long __pyx_t_1;
long __pyx_t_2;
int __pyx_t_3;
int __pyx_t_4;
int __pyx_t_5;
/* "gensim/models/doc2vec_inner.pyx":135
*
* cdef long long row2
* cdef unsigned long long modulo = 281474976710655ULL # <<<<<<<<<<<<<<
* cdef REAL_t f, g, label
* cdef np.uint32_t target_index
*/
__pyx_v_modulo = 281474976710655ULL;
/* "gensim/models/doc2vec_inner.pyx":142
* # l1 already composed by caller, passed in as neu1
* # work (also passsed in) will accumulate l1 error for outside application
* for d in range(negative+1): # <<<<<<<<<<<<<<
* if d == 0:
* target_index = predict_word_index
*/
__pyx_t_1 = (__pyx_v_negative + 1);
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_d = __pyx_t_3;
/* "gensim/models/doc2vec_inner.pyx":143
* # work (also passsed in) will accumulate l1 error for outside application
* for d in range(negative+1):
* if d == 0: # <<<<<<<<<<<<<<
* target_index = predict_word_index
* label = ONEF
*/
__pyx_t_4 = ((__pyx_v_d == 0) != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":144
* for d in range(negative+1):
* if d == 0:
* target_index = predict_word_index # <<<<<<<<<<<<<<
* label = ONEF
* else:
*/
__pyx_v_target_index = __pyx_v_predict_word_index;
/* "gensim/models/doc2vec_inner.pyx":145
* if d == 0:
* target_index = predict_word_index
* label = ONEF # <<<<<<<<<<<<<<
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
*/
__pyx_v_label = __pyx_v_6gensim_6models_13doc2vec_inner_ONEF;
/* "gensim/models/doc2vec_inner.pyx":143
* # work (also passsed in) will accumulate l1 error for outside application
* for d in range(negative+1):
* if d == 0: # <<<<<<<<<<<<<<
* target_index = predict_word_index
* label = ONEF
*/
goto __pyx_L5;
}
/* "gensim/models/doc2vec_inner.pyx":147
* label = ONEF
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len) # <<<<<<<<<<<<<<
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index:
*/
/*else*/ {
__pyx_v_target_index = __pyx_f_6gensim_6models_14word2vec_inner_bisect_left(__pyx_v_cum_table, ((__pyx_v_next_random >> 16) % (__pyx_v_cum_table[(__pyx_v_cum_table_len - 1)])), 0, __pyx_v_cum_table_len);
/* "gensim/models/doc2vec_inner.pyx":148
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo # <<<<<<<<<<<<<<
* if target_index == predict_word_index:
* continue
*/
__pyx_v_next_random = (((__pyx_v_next_random * ((unsigned PY_LONG_LONG)25214903917ULL)) + 11) & __pyx_v_modulo);
/* "gensim/models/doc2vec_inner.pyx":149
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index: # <<<<<<<<<<<<<<
* continue
* label = 0.0
*/
__pyx_t_4 = ((__pyx_v_target_index == __pyx_v_predict_word_index) != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":150
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index:
* continue # <<<<<<<<<<<<<<
* label = 0.0
*
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":149
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index: # <<<<<<<<<<<<<<
* continue
* label = 0.0
*/
}
/* "gensim/models/doc2vec_inner.pyx":151
* if target_index == predict_word_index:
* continue
* label = 0.0 # <<<<<<<<<<<<<<
*
* row2 = target_index * size
*/
__pyx_v_label = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)0.0);
}
__pyx_L5:;
/* "gensim/models/doc2vec_inner.pyx":153
* label = 0.0
*
* row2 = target_index * size # <<<<<<<<<<<<<<
* f = our_dot(&size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
*/
__pyx_v_row2 = (__pyx_v_target_index * __pyx_v_size);
/* "gensim/models/doc2vec_inner.pyx":154
*
* row2 = target_index * size
* f = our_dot(&size, neu1, &ONE, &syn1neg[row2], &ONE) # <<<<<<<<<<<<<<
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
*/
__pyx_v_f = __pyx_v_6gensim_6models_14word2vec_inner_our_dot((&__pyx_v_size), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":155
* row2 = target_index * size
* f = our_dot(&size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
__pyx_t_5 = ((__pyx_v_f <= -6.0) != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_4 = __pyx_t_5;
goto __pyx_L8_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_f >= 6.0) != 0);
__pyx_t_4 = __pyx_t_5;
__pyx_L8_bool_binop_done:;
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":156
* f = our_dot(&size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue # <<<<<<<<<<<<<<
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":155
* row2 = target_index * size
* f = our_dot(&size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
}
/* "gensim/models/doc2vec_inner.pyx":157
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))] # <<<<<<<<<<<<<<
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
*/
__pyx_v_f = (__pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE[((int)((__pyx_v_f + 6.0) * ((0x3E8 / 6) / 2)))]);
/* "gensim/models/doc2vec_inner.pyx":158
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden:
*/
__pyx_v_g = ((__pyx_v_label - __pyx_v_f) * __pyx_v_alpha);
/* "gensim/models/doc2vec_inner.pyx":159
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE) # <<<<<<<<<<<<<<
* if learn_hidden:
* our_saxpy(&size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":160
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*
*/
__pyx_t_4 = (__pyx_v_learn_hidden != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":161
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden:
* our_saxpy(&size, &g, neu1, &ONE, &syn1neg[row2], &ONE) # <<<<<<<<<<<<<<
*
* return next_random
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_size), (&__pyx_v_g), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":160
* g = (label - f) * alpha
* our_saxpy(&size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*
*/
}
__pyx_L3_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":163
* our_saxpy(&size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*
* return next_random # <<<<<<<<<<<<<<
*
* cdef void fast_document_dmc_hs(
*/
__pyx_r = __pyx_v_next_random;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":129
*
*
* cdef unsigned long long fast_document_dm_neg( # <<<<<<<<<<<<<<
* const int negative, np.uint32_t *cum_table, unsigned long long cum_table_len, unsigned long long next_random,
* REAL_t *neu1, REAL_t *syn1neg, const int predict_word_index, const REAL_t alpha, REAL_t *work,
*/
/* function exit code */
__pyx_L0:;
return __pyx_r;
}
/* "gensim/models/doc2vec_inner.pyx":165
* return next_random
*
* cdef void fast_document_dmc_hs( # <<<<<<<<<<<<<<
* const np.uint32_t *word_point, const np.uint8_t *word_code, int word_code_len,
* REAL_t *neu1, REAL_t *syn1, const REAL_t alpha, REAL_t *work,
*/
static void __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dmc_hs(__pyx_t_5numpy_uint32_t const *__pyx_v_word_point, __pyx_t_5numpy_uint8_t const *__pyx_v_word_code, int __pyx_v_word_code_len, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_neu1, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_syn1, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const __pyx_v_alpha, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_work, int const __pyx_v_layer1_size, CYTHON_UNUSED int const __pyx_v_vector_size, int __pyx_v_learn_hidden) {
PY_LONG_LONG __pyx_v_b;
PY_LONG_LONG __pyx_v_row2;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_f;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_g;
int __pyx_t_1;
int __pyx_t_2;
PY_LONG_LONG __pyx_t_3;
int __pyx_t_4;
int __pyx_t_5;
/* "gensim/models/doc2vec_inner.pyx":177
* # l1 already composed by caller, passed in as neu1
* # work accumulates net l1 error; eventually applied by caller
* for b in range(word_code_len): # <<<<<<<<<<<<<<
* row2 = word_point[b] * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1[row2], &ONE)
*/
__pyx_t_1 = __pyx_v_word_code_len;
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_b = __pyx_t_3;
/* "gensim/models/doc2vec_inner.pyx":178
* # work accumulates net l1 error; eventually applied by caller
* for b in range(word_code_len):
* row2 = word_point[b] * layer1_size # <<<<<<<<<<<<<<
* f = our_dot(&layer1_size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
*/
__pyx_v_row2 = ((__pyx_v_word_point[__pyx_v_b]) * __pyx_v_layer1_size);
/* "gensim/models/doc2vec_inner.pyx":179
* for b in range(word_code_len):
* row2 = word_point[b] * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1[row2], &ONE) # <<<<<<<<<<<<<<
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
*/
__pyx_v_f = __pyx_v_6gensim_6models_14word2vec_inner_our_dot((&__pyx_v_layer1_size), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":180
* row2 = word_point[b] * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
__pyx_t_5 = ((__pyx_v_f <= -6.0) != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_4 = __pyx_t_5;
goto __pyx_L6_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_f >= 6.0) != 0);
__pyx_t_4 = __pyx_t_5;
__pyx_L6_bool_binop_done:;
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":181
* f = our_dot(&layer1_size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue # <<<<<<<<<<<<<<
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":180
* row2 = word_point[b] * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
}
/* "gensim/models/doc2vec_inner.pyx":182
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))] # <<<<<<<<<<<<<<
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1[row2], &ONE, work, &ONE)
*/
__pyx_v_f = (__pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE[((int)((__pyx_v_f + 6.0) * ((0x3E8 / 6) / 2)))]);
/* "gensim/models/doc2vec_inner.pyx":183
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha # <<<<<<<<<<<<<<
* our_saxpy(&layer1_size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden:
*/
__pyx_v_g = (((1 - (__pyx_v_word_code[__pyx_v_b])) - __pyx_v_f) * __pyx_v_alpha);
/* "gensim/models/doc2vec_inner.pyx":184
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1[row2], &ONE, work, &ONE) # <<<<<<<<<<<<<<
* if learn_hidden:
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1[row2], &ONE)
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_layer1_size), (&__pyx_v_g), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":185
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1[row2], &ONE)
*
*/
__pyx_t_4 = (__pyx_v_learn_hidden != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":186
* our_saxpy(&layer1_size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden:
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1[row2], &ONE) # <<<<<<<<<<<<<<
*
*
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_layer1_size), (&__pyx_v_g), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":185
* g = (1 - word_code[b] - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1[row2], &ONE)
*
*/
}
__pyx_L3_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":165
* return next_random
*
* cdef void fast_document_dmc_hs( # <<<<<<<<<<<<<<
* const np.uint32_t *word_point, const np.uint8_t *word_code, int word_code_len,
* REAL_t *neu1, REAL_t *syn1, const REAL_t alpha, REAL_t *work,
*/
/* function exit code */
}
/* "gensim/models/doc2vec_inner.pyx":189
*
*
* cdef unsigned long long fast_document_dmc_neg( # <<<<<<<<<<<<<<
* const int negative, np.uint32_t *cum_table, unsigned long long cum_table_len, unsigned long long next_random,
* REAL_t *neu1, REAL_t *syn1neg, const int predict_word_index, const REAL_t alpha, REAL_t *work,
*/
static unsigned PY_LONG_LONG __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dmc_neg(int const __pyx_v_negative, __pyx_t_5numpy_uint32_t *__pyx_v_cum_table, unsigned PY_LONG_LONG __pyx_v_cum_table_len, unsigned PY_LONG_LONG __pyx_v_next_random, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_neu1, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_syn1neg, int const __pyx_v_predict_word_index, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t const __pyx_v_alpha, __pyx_t_6gensim_6models_14word2vec_inner_REAL_t *__pyx_v_work, int const __pyx_v_layer1_size, CYTHON_UNUSED int const __pyx_v_vector_size, int __pyx_v_learn_hidden) {
PY_LONG_LONG __pyx_v_row2;
unsigned PY_LONG_LONG __pyx_v_modulo;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_f;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_g;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_label;
__pyx_t_5numpy_uint32_t __pyx_v_target_index;
int __pyx_v_d;
unsigned PY_LONG_LONG __pyx_r;
long __pyx_t_1;
long __pyx_t_2;
int __pyx_t_3;
int __pyx_t_4;
int __pyx_t_5;
/* "gensim/models/doc2vec_inner.pyx":196
* cdef long long a
* cdef long long row2
* cdef unsigned long long modulo = 281474976710655ULL # <<<<<<<<<<<<<<
* cdef REAL_t f, g, label
* cdef np.uint32_t target_index
*/
__pyx_v_modulo = 281474976710655ULL;
/* "gensim/models/doc2vec_inner.pyx":203
* # l1 already composed by caller, passed in as neu1
* # work accumulates net l1 error; eventually applied by caller
* for d in range(negative+1): # <<<<<<<<<<<<<<
* if d == 0:
* target_index = predict_word_index
*/
__pyx_t_1 = (__pyx_v_negative + 1);
__pyx_t_2 = __pyx_t_1;
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
__pyx_v_d = __pyx_t_3;
/* "gensim/models/doc2vec_inner.pyx":204
* # work accumulates net l1 error; eventually applied by caller
* for d in range(negative+1):
* if d == 0: # <<<<<<<<<<<<<<
* target_index = predict_word_index
* label = ONEF
*/
__pyx_t_4 = ((__pyx_v_d == 0) != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":205
* for d in range(negative+1):
* if d == 0:
* target_index = predict_word_index # <<<<<<<<<<<<<<
* label = ONEF
* else:
*/
__pyx_v_target_index = __pyx_v_predict_word_index;
/* "gensim/models/doc2vec_inner.pyx":206
* if d == 0:
* target_index = predict_word_index
* label = ONEF # <<<<<<<<<<<<<<
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
*/
__pyx_v_label = __pyx_v_6gensim_6models_13doc2vec_inner_ONEF;
/* "gensim/models/doc2vec_inner.pyx":204
* # work accumulates net l1 error; eventually applied by caller
* for d in range(negative+1):
* if d == 0: # <<<<<<<<<<<<<<
* target_index = predict_word_index
* label = ONEF
*/
goto __pyx_L5;
}
/* "gensim/models/doc2vec_inner.pyx":208
* label = ONEF
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len) # <<<<<<<<<<<<<<
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index:
*/
/*else*/ {
__pyx_v_target_index = __pyx_f_6gensim_6models_14word2vec_inner_bisect_left(__pyx_v_cum_table, ((__pyx_v_next_random >> 16) % (__pyx_v_cum_table[(__pyx_v_cum_table_len - 1)])), 0, __pyx_v_cum_table_len);
/* "gensim/models/doc2vec_inner.pyx":209
* else:
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo # <<<<<<<<<<<<<<
* if target_index == predict_word_index:
* continue
*/
__pyx_v_next_random = (((__pyx_v_next_random * ((unsigned PY_LONG_LONG)25214903917ULL)) + 11) & __pyx_v_modulo);
/* "gensim/models/doc2vec_inner.pyx":210
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index: # <<<<<<<<<<<<<<
* continue
* label = 0.0
*/
__pyx_t_4 = ((__pyx_v_target_index == __pyx_v_predict_word_index) != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":211
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index:
* continue # <<<<<<<<<<<<<<
* label = 0.0
*
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":210
* target_index = bisect_left(cum_table, (next_random >> 16) % cum_table[cum_table_len-1], 0, cum_table_len)
* next_random = (next_random * 25214903917ULL + 11) & modulo
* if target_index == predict_word_index: # <<<<<<<<<<<<<<
* continue
* label = 0.0
*/
}
/* "gensim/models/doc2vec_inner.pyx":212
* if target_index == predict_word_index:
* continue
* label = 0.0 # <<<<<<<<<<<<<<
*
* row2 = target_index * layer1_size
*/
__pyx_v_label = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)0.0);
}
__pyx_L5:;
/* "gensim/models/doc2vec_inner.pyx":214
* label = 0.0
*
* row2 = target_index * layer1_size # <<<<<<<<<<<<<<
* f = our_dot(&layer1_size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
*/
__pyx_v_row2 = (__pyx_v_target_index * __pyx_v_layer1_size);
/* "gensim/models/doc2vec_inner.pyx":215
*
* row2 = target_index * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1neg[row2], &ONE) # <<<<<<<<<<<<<<
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
*/
__pyx_v_f = __pyx_v_6gensim_6models_14word2vec_inner_our_dot((&__pyx_v_layer1_size), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":216
* row2 = target_index * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
__pyx_t_5 = ((__pyx_v_f <= -6.0) != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_4 = __pyx_t_5;
goto __pyx_L8_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_f >= 6.0) != 0);
__pyx_t_4 = __pyx_t_5;
__pyx_L8_bool_binop_done:;
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":217
* f = our_dot(&layer1_size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue # <<<<<<<<<<<<<<
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha
*/
goto __pyx_L3_continue;
/* "gensim/models/doc2vec_inner.pyx":216
* row2 = target_index * layer1_size
* f = our_dot(&layer1_size, neu1, &ONE, &syn1neg[row2], &ONE)
* if f <= -MAX_EXP or f >= MAX_EXP: # <<<<<<<<<<<<<<
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
*/
}
/* "gensim/models/doc2vec_inner.pyx":218
* if f <= -MAX_EXP or f >= MAX_EXP:
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))] # <<<<<<<<<<<<<<
* g = (label - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1neg[row2], &ONE, work, &ONE)
*/
__pyx_v_f = (__pyx_v_6gensim_6models_14word2vec_inner_EXP_TABLE[((int)((__pyx_v_f + 6.0) * ((0x3E8 / 6) / 2)))]);
/* "gensim/models/doc2vec_inner.pyx":219
* continue
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha # <<<<<<<<<<<<<<
* our_saxpy(&layer1_size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden:
*/
__pyx_v_g = ((__pyx_v_label - __pyx_v_f) * __pyx_v_alpha);
/* "gensim/models/doc2vec_inner.pyx":220
* f = EXP_TABLE[((f + MAX_EXP) * (EXP_TABLE_SIZE / MAX_EXP / 2))]
* g = (label - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1neg[row2], &ONE, work, &ONE) # <<<<<<<<<<<<<<
* if learn_hidden:
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_layer1_size), (&__pyx_v_g), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":221
* g = (label - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*
*/
__pyx_t_4 = (__pyx_v_learn_hidden != 0);
if (__pyx_t_4) {
/* "gensim/models/doc2vec_inner.pyx":222
* our_saxpy(&layer1_size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden:
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1neg[row2], &ONE) # <<<<<<<<<<<<<<
*
* return next_random
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_layer1_size), (&__pyx_v_g), __pyx_v_neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_syn1neg[__pyx_v_row2])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":221
* g = (label - f) * alpha
* our_saxpy(&layer1_size, &g, &syn1neg[row2], &ONE, work, &ONE)
* if learn_hidden: # <<<<<<<<<<<<<<
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*
*/
}
__pyx_L3_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":224
* our_saxpy(&layer1_size, &g, neu1, &ONE, &syn1neg[row2], &ONE)
*
* return next_random # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = __pyx_v_next_random;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":189
*
*
* cdef unsigned long long fast_document_dmc_neg( # <<<<<<<<<<<<<<
* const int negative, np.uint32_t *cum_table, unsigned long long cum_table_len, unsigned long long next_random,
* REAL_t *neu1, REAL_t *syn1neg, const int predict_word_index, const REAL_t alpha, REAL_t *work,
*/
/* function exit code */
__pyx_L0:;
return __pyx_r;
}
/* "gensim/models/doc2vec_inner.pyx":227
*
*
* cdef init_d2v_config(Doc2VecConfig *c, model, alpha, learn_doctags, learn_words, learn_hidden, # <<<<<<<<<<<<<<
* train_words=False, work=None, neu1=None, word_vectors=None, words_lockf=None,
* doctag_vectors=None, doctags_lockf=None, docvecs_count=0):
*/
static PyObject *__pyx_f_6gensim_6models_13doc2vec_inner_init_d2v_config(struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig *__pyx_v_c, PyObject *__pyx_v_model, PyObject *__pyx_v_alpha, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config *__pyx_optional_args) {
/* "gensim/models/doc2vec_inner.pyx":228
*
* cdef init_d2v_config(Doc2VecConfig *c, model, alpha, learn_doctags, learn_words, learn_hidden,
* train_words=False, work=None, neu1=None, word_vectors=None, words_lockf=None, # <<<<<<<<<<<<<<
* doctag_vectors=None, doctags_lockf=None, docvecs_count=0):
* c[0].hs = model.hs
*/
PyObject *__pyx_v_train_words = ((PyObject *)Py_False);
PyObject *__pyx_v_work = ((PyObject *)Py_None);
PyObject *__pyx_v_neu1 = ((PyObject *)Py_None);
PyObject *__pyx_v_word_vectors = ((PyObject *)Py_None);
PyObject *__pyx_v_words_lockf = ((PyObject *)Py_None);
/* "gensim/models/doc2vec_inner.pyx":229
* cdef init_d2v_config(Doc2VecConfig *c, model, alpha, learn_doctags, learn_words, learn_hidden,
* train_words=False, work=None, neu1=None, word_vectors=None, words_lockf=None,
* doctag_vectors=None, doctags_lockf=None, docvecs_count=0): # <<<<<<<<<<<<<<
* c[0].hs = model.hs
* c[0].negative = model.negative
*/
PyObject *__pyx_v_doctag_vectors = ((PyObject *)Py_None);
PyObject *__pyx_v_doctags_lockf = ((PyObject *)Py_None);
PyObject *__pyx_v_docvecs_count = ((PyObject *)__pyx_int_0);
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
int __pyx_t_2;
PyObject *__pyx_t_3 = NULL;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_t_4;
int __pyx_t_5;
int __pyx_t_6;
PyObject *__pyx_t_7 = NULL;
Py_ssize_t __pyx_t_8;
unsigned PY_LONG_LONG __pyx_t_9;
PyObject *__pyx_t_10 = NULL;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("init_d2v_config", 0);
if (__pyx_optional_args) {
if (__pyx_optional_args->__pyx_n > 0) {
__pyx_v_train_words = __pyx_optional_args->train_words;
if (__pyx_optional_args->__pyx_n > 1) {
__pyx_v_work = __pyx_optional_args->work;
if (__pyx_optional_args->__pyx_n > 2) {
__pyx_v_neu1 = __pyx_optional_args->neu1;
if (__pyx_optional_args->__pyx_n > 3) {
__pyx_v_word_vectors = __pyx_optional_args->word_vectors;
if (__pyx_optional_args->__pyx_n > 4) {
__pyx_v_words_lockf = __pyx_optional_args->words_lockf;
if (__pyx_optional_args->__pyx_n > 5) {
__pyx_v_doctag_vectors = __pyx_optional_args->doctag_vectors;
if (__pyx_optional_args->__pyx_n > 6) {
__pyx_v_doctags_lockf = __pyx_optional_args->doctags_lockf;
if (__pyx_optional_args->__pyx_n > 7) {
__pyx_v_docvecs_count = __pyx_optional_args->docvecs_count;
}
}
}
}
}
}
}
}
}
__Pyx_INCREF(__pyx_v_work);
__Pyx_INCREF(__pyx_v_neu1);
__Pyx_INCREF(__pyx_v_word_vectors);
__Pyx_INCREF(__pyx_v_words_lockf);
__Pyx_INCREF(__pyx_v_doctag_vectors);
__Pyx_INCREF(__pyx_v_doctags_lockf);
/* "gensim/models/doc2vec_inner.pyx":230
* train_words=False, work=None, neu1=None, word_vectors=None, words_lockf=None,
* doctag_vectors=None, doctags_lockf=None, docvecs_count=0):
* c[0].hs = model.hs # <<<<<<<<<<<<<<
* c[0].negative = model.negative
* c[0].sample = (model.sample != 0)
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_hs); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 230, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 230, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).hs = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":231
* doctag_vectors=None, doctags_lockf=None, docvecs_count=0):
* c[0].hs = model.hs
* c[0].negative = model.negative # <<<<<<<<<<<<<<
* c[0].sample = (model.sample != 0)
* c[0].cbow_mean = model.cbow_mean
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_negative); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 231, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 231, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).negative = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":232
* c[0].hs = model.hs
* c[0].negative = model.negative
* c[0].sample = (model.sample != 0) # <<<<<<<<<<<<<<
* c[0].cbow_mean = model.cbow_mean
* c[0].train_words = train_words
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_sample); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 232, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_3 = __Pyx_PyInt_NeObjC(__pyx_t_1, __pyx_int_0, 0, 0); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 232, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_3); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 232, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
(__pyx_v_c[0]).sample = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":233
* c[0].negative = model.negative
* c[0].sample = (model.sample != 0)
* c[0].cbow_mean = model.cbow_mean # <<<<<<<<<<<<<<
* c[0].train_words = train_words
* c[0].learn_doctags = learn_doctags
*/
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_cbow_mean); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 233, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_3); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 233, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
(__pyx_v_c[0]).cbow_mean = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":234
* c[0].sample = (model.sample != 0)
* c[0].cbow_mean = model.cbow_mean
* c[0].train_words = train_words # <<<<<<<<<<<<<<
* c[0].learn_doctags = learn_doctags
* c[0].learn_words = learn_words
*/
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_v_train_words); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 234, __pyx_L1_error)
(__pyx_v_c[0]).train_words = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":235
* c[0].cbow_mean = model.cbow_mean
* c[0].train_words = train_words
* c[0].learn_doctags = learn_doctags # <<<<<<<<<<<<<<
* c[0].learn_words = learn_words
* c[0].learn_hidden = learn_hidden
*/
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_v_learn_doctags); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 235, __pyx_L1_error)
(__pyx_v_c[0]).learn_doctags = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":236
* c[0].train_words = train_words
* c[0].learn_doctags = learn_doctags
* c[0].learn_words = learn_words # <<<<<<<<<<<<<<
* c[0].learn_hidden = learn_hidden
* c[0].alpha = alpha
*/
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_v_learn_words); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 236, __pyx_L1_error)
(__pyx_v_c[0]).learn_words = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":237
* c[0].learn_doctags = learn_doctags
* c[0].learn_words = learn_words
* c[0].learn_hidden = learn_hidden # <<<<<<<<<<<<<<
* c[0].alpha = alpha
* c[0].layer1_size = model.layer1_size
*/
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_v_learn_hidden); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 237, __pyx_L1_error)
(__pyx_v_c[0]).learn_hidden = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":238
* c[0].learn_words = learn_words
* c[0].learn_hidden = learn_hidden
* c[0].alpha = alpha # <<<<<<<<<<<<<<
* c[0].layer1_size = model.layer1_size
* c[0].vector_size = model.dv.vector_size
*/
__pyx_t_4 = __pyx_PyFloat_AsFloat(__pyx_v_alpha); if (unlikely((__pyx_t_4 == ((npy_float32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 238, __pyx_L1_error)
(__pyx_v_c[0]).alpha = __pyx_t_4;
/* "gensim/models/doc2vec_inner.pyx":239
* c[0].learn_hidden = learn_hidden
* c[0].alpha = alpha
* c[0].layer1_size = model.layer1_size # <<<<<<<<<<<<<<
* c[0].vector_size = model.dv.vector_size
* c[0].workers = model.workers
*/
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_layer1_size); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 239, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_3); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 239, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
(__pyx_v_c[0]).layer1_size = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":240
* c[0].alpha = alpha
* c[0].layer1_size = model.layer1_size
* c[0].vector_size = model.dv.vector_size # <<<<<<<<<<<<<<
* c[0].workers = model.workers
* c[0].docvecs_count = docvecs_count
*/
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_dv); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 240, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_vector_size); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 240, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 240, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).vector_size = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":241
* c[0].layer1_size = model.layer1_size
* c[0].vector_size = model.dv.vector_size
* c[0].workers = model.workers # <<<<<<<<<<<<<<
* c[0].docvecs_count = docvecs_count
*
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_workers); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 241, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 241, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).workers = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":242
* c[0].vector_size = model.dv.vector_size
* c[0].workers = model.workers
* c[0].docvecs_count = docvecs_count # <<<<<<<<<<<<<<
*
* c[0].window = model.window
*/
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_v_docvecs_count); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 242, __pyx_L1_error)
(__pyx_v_c[0]).docvecs_count = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":244
* c[0].docvecs_count = docvecs_count
*
* c[0].window = model.window # <<<<<<<<<<<<<<
* c[0].expected_doctag_len = model.dm_tag_count
*
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_window); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 244, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 244, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).window = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":245
*
* c[0].window = model.window
* c[0].expected_doctag_len = model.dm_tag_count # <<<<<<<<<<<<<<
*
* if '\0' in model.wv:
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_dm_tag_count); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 245, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 245, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).expected_doctag_len = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":247
* c[0].expected_doctag_len = model.dm_tag_count
*
* if '\0' in model.wv: # <<<<<<<<<<<<<<
* c[0].null_word_index = model.wv.get_index('\0')
*
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 247, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_5 = (__Pyx_PySequence_ContainsTF(__pyx_kp_u_, __pyx_t_1, Py_EQ)); if (unlikely(__pyx_t_5 < 0)) __PYX_ERR(0, 247, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_6 = (__pyx_t_5 != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":248
*
* if '\0' in model.wv:
* c[0].null_word_index = model.wv.get_index('\0') # <<<<<<<<<<<<<<
*
* # default vectors, locks from syn0/doctag_syn0
*/
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 248, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_get_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 248, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_3 = NULL;
if (CYTHON_UNPACK_METHODS && likely(PyMethod_Check(__pyx_t_7))) {
__pyx_t_3 = PyMethod_GET_SELF(__pyx_t_7);
if (likely(__pyx_t_3)) {
PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_7);
__Pyx_INCREF(__pyx_t_3);
__Pyx_INCREF(function);
__Pyx_DECREF_SET(__pyx_t_7, function);
}
}
__pyx_t_1 = (__pyx_t_3) ? __Pyx_PyObject_Call2Args(__pyx_t_7, __pyx_t_3, __pyx_kp_u_) : __Pyx_PyObject_CallOneArg(__pyx_t_7, __pyx_kp_u_);
__Pyx_XDECREF(__pyx_t_3); __pyx_t_3 = 0;
if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 248, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_2 = __Pyx_PyInt_As_int(__pyx_t_1); if (unlikely((__pyx_t_2 == (int)-1) && PyErr_Occurred())) __PYX_ERR(0, 248, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).null_word_index = __pyx_t_2;
/* "gensim/models/doc2vec_inner.pyx":247
* c[0].expected_doctag_len = model.dm_tag_count
*
* if '\0' in model.wv: # <<<<<<<<<<<<<<
* c[0].null_word_index = model.wv.get_index('\0')
*
*/
}
/* "gensim/models/doc2vec_inner.pyx":251
*
* # default vectors, locks from syn0/doctag_syn0
* if word_vectors is None: # <<<<<<<<<<<<<<
* word_vectors = model.wv.vectors
* c[0].word_vectors = (np.PyArray_DATA(word_vectors))
*/
__pyx_t_6 = (__pyx_v_word_vectors == Py_None);
__pyx_t_5 = (__pyx_t_6 != 0);
if (__pyx_t_5) {
/* "gensim/models/doc2vec_inner.pyx":252
* # default vectors, locks from syn0/doctag_syn0
* if word_vectors is None:
* word_vectors = model.wv.vectors # <<<<<<<<<<<<<<
* c[0].word_vectors = (np.PyArray_DATA(word_vectors))
* if doctag_vectors is None:
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 252, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_vectors); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 252, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF_SET(__pyx_v_word_vectors, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":251
*
* # default vectors, locks from syn0/doctag_syn0
* if word_vectors is None: # <<<<<<<<<<<<<<
* word_vectors = model.wv.vectors
* c[0].word_vectors = (np.PyArray_DATA(word_vectors))
*/
}
/* "gensim/models/doc2vec_inner.pyx":253
* if word_vectors is None:
* word_vectors = model.wv.vectors
* c[0].word_vectors = (np.PyArray_DATA(word_vectors)) # <<<<<<<<<<<<<<
* if doctag_vectors is None:
* doctag_vectors = model.dv.vectors
*/
if (!(likely(((__pyx_v_word_vectors) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_word_vectors, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 253, __pyx_L1_error)
(__pyx_v_c[0]).word_vectors = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_v_word_vectors)));
/* "gensim/models/doc2vec_inner.pyx":254
* word_vectors = model.wv.vectors
* c[0].word_vectors = (np.PyArray_DATA(word_vectors))
* if doctag_vectors is None: # <<<<<<<<<<<<<<
* doctag_vectors = model.dv.vectors
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors))
*/
__pyx_t_5 = (__pyx_v_doctag_vectors == Py_None);
__pyx_t_6 = (__pyx_t_5 != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":255
* c[0].word_vectors = (np.PyArray_DATA(word_vectors))
* if doctag_vectors is None:
* doctag_vectors = model.dv.vectors # <<<<<<<<<<<<<<
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors))
* if words_lockf is None:
*/
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_dv); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 255, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_7, __pyx_n_s_vectors); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 255, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF_SET(__pyx_v_doctag_vectors, __pyx_t_1);
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":254
* word_vectors = model.wv.vectors
* c[0].word_vectors = (np.PyArray_DATA(word_vectors))
* if doctag_vectors is None: # <<<<<<<<<<<<<<
* doctag_vectors = model.dv.vectors
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors))
*/
}
/* "gensim/models/doc2vec_inner.pyx":256
* if doctag_vectors is None:
* doctag_vectors = model.dv.vectors
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors)) # <<<<<<<<<<<<<<
* if words_lockf is None:
* words_lockf = model.wv.vectors_lockf
*/
if (!(likely(((__pyx_v_doctag_vectors) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_doctag_vectors, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 256, __pyx_L1_error)
(__pyx_v_c[0]).doctag_vectors = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_v_doctag_vectors)));
/* "gensim/models/doc2vec_inner.pyx":257
* doctag_vectors = model.dv.vectors
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors))
* if words_lockf is None: # <<<<<<<<<<<<<<
* words_lockf = model.wv.vectors_lockf
* c[0].words_lockf = (np.PyArray_DATA(words_lockf))
*/
__pyx_t_6 = (__pyx_v_words_lockf == Py_None);
__pyx_t_5 = (__pyx_t_6 != 0);
if (__pyx_t_5) {
/* "gensim/models/doc2vec_inner.pyx":258
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors))
* if words_lockf is None:
* words_lockf = model.wv.vectors_lockf # <<<<<<<<<<<<<<
* c[0].words_lockf = (np.PyArray_DATA(words_lockf))
* c[0].words_lockf_len = len(words_lockf)
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 258, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_vectors_lockf); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 258, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF_SET(__pyx_v_words_lockf, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":257
* doctag_vectors = model.dv.vectors
* c[0].doctag_vectors = (np.PyArray_DATA(doctag_vectors))
* if words_lockf is None: # <<<<<<<<<<<<<<
* words_lockf = model.wv.vectors_lockf
* c[0].words_lockf = (np.PyArray_DATA(words_lockf))
*/
}
/* "gensim/models/doc2vec_inner.pyx":259
* if words_lockf is None:
* words_lockf = model.wv.vectors_lockf
* c[0].words_lockf = (np.PyArray_DATA(words_lockf)) # <<<<<<<<<<<<<<
* c[0].words_lockf_len = len(words_lockf)
* if doctags_lockf is None:
*/
if (!(likely(((__pyx_v_words_lockf) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_words_lockf, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 259, __pyx_L1_error)
(__pyx_v_c[0]).words_lockf = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_v_words_lockf)));
/* "gensim/models/doc2vec_inner.pyx":260
* words_lockf = model.wv.vectors_lockf
* c[0].words_lockf = (np.PyArray_DATA(words_lockf))
* c[0].words_lockf_len = len(words_lockf) # <<<<<<<<<<<<<<
* if doctags_lockf is None:
* doctags_lockf = model.dv.vectors_lockf
*/
__pyx_t_8 = PyObject_Length(__pyx_v_words_lockf); if (unlikely(__pyx_t_8 == ((Py_ssize_t)-1))) __PYX_ERR(0, 260, __pyx_L1_error)
(__pyx_v_c[0]).words_lockf_len = __pyx_t_8;
/* "gensim/models/doc2vec_inner.pyx":261
* c[0].words_lockf = (np.PyArray_DATA(words_lockf))
* c[0].words_lockf_len = len(words_lockf)
* if doctags_lockf is None: # <<<<<<<<<<<<<<
* doctags_lockf = model.dv.vectors_lockf
* c[0].doctags_lockf = (np.PyArray_DATA(doctags_lockf))
*/
__pyx_t_5 = (__pyx_v_doctags_lockf == Py_None);
__pyx_t_6 = (__pyx_t_5 != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":262
* c[0].words_lockf_len = len(words_lockf)
* if doctags_lockf is None:
* doctags_lockf = model.dv.vectors_lockf # <<<<<<<<<<<<<<
* c[0].doctags_lockf = (np.PyArray_DATA(doctags_lockf))
* c[0].doctags_lockf_len = len(doctags_lockf)
*/
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_dv); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 262, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_7, __pyx_n_s_vectors_lockf); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 262, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF_SET(__pyx_v_doctags_lockf, __pyx_t_1);
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":261
* c[0].words_lockf = (np.PyArray_DATA(words_lockf))
* c[0].words_lockf_len = len(words_lockf)
* if doctags_lockf is None: # <<<<<<<<<<<<<<
* doctags_lockf = model.dv.vectors_lockf
* c[0].doctags_lockf = (np.PyArray_DATA(doctags_lockf))
*/
}
/* "gensim/models/doc2vec_inner.pyx":263
* if doctags_lockf is None:
* doctags_lockf = model.dv.vectors_lockf
* c[0].doctags_lockf = (np.PyArray_DATA(doctags_lockf)) # <<<<<<<<<<<<<<
* c[0].doctags_lockf_len = len(doctags_lockf)
*
*/
if (!(likely(((__pyx_v_doctags_lockf) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_doctags_lockf, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 263, __pyx_L1_error)
(__pyx_v_c[0]).doctags_lockf = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_v_doctags_lockf)));
/* "gensim/models/doc2vec_inner.pyx":264
* doctags_lockf = model.dv.vectors_lockf
* c[0].doctags_lockf = (np.PyArray_DATA(doctags_lockf))
* c[0].doctags_lockf_len = len(doctags_lockf) # <<<<<<<<<<<<<<
*
* if c[0].hs:
*/
__pyx_t_8 = PyObject_Length(__pyx_v_doctags_lockf); if (unlikely(__pyx_t_8 == ((Py_ssize_t)-1))) __PYX_ERR(0, 264, __pyx_L1_error)
(__pyx_v_c[0]).doctags_lockf_len = __pyx_t_8;
/* "gensim/models/doc2vec_inner.pyx":266
* c[0].doctags_lockf_len = len(doctags_lockf)
*
* if c[0].hs: # <<<<<<<<<<<<<<
* c[0].syn1 = (np.PyArray_DATA(model.syn1))
*
*/
__pyx_t_6 = ((__pyx_v_c[0]).hs != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":267
*
* if c[0].hs:
* c[0].syn1 = (np.PyArray_DATA(model.syn1)) # <<<<<<<<<<<<<<
*
* if c[0].negative:
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_syn1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 267, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 267, __pyx_L1_error)
(__pyx_v_c[0]).syn1 = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_1)));
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":266
* c[0].doctags_lockf_len = len(doctags_lockf)
*
* if c[0].hs: # <<<<<<<<<<<<<<
* c[0].syn1 = (np.PyArray_DATA(model.syn1))
*
*/
}
/* "gensim/models/doc2vec_inner.pyx":269
* c[0].syn1 = (np.PyArray_DATA(model.syn1))
*
* if c[0].negative: # <<<<<<<<<<<<<<
* c[0].syn1neg = (np.PyArray_DATA(model.syn1neg))
* c[0].cum_table = (np.PyArray_DATA(model.cum_table))
*/
__pyx_t_6 = ((__pyx_v_c[0]).negative != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":270
*
* if c[0].negative:
* c[0].syn1neg = (np.PyArray_DATA(model.syn1neg)) # <<<<<<<<<<<<<<
* c[0].cum_table = (np.PyArray_DATA(model.cum_table))
* c[0].cum_table_len = len(model.cum_table)
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_syn1neg); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 270, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 270, __pyx_L1_error)
(__pyx_v_c[0]).syn1neg = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_1)));
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":271
* if c[0].negative:
* c[0].syn1neg = (np.PyArray_DATA(model.syn1neg))
* c[0].cum_table = (np.PyArray_DATA(model.cum_table)) # <<<<<<<<<<<<<<
* c[0].cum_table_len = len(model.cum_table)
* if c[0].negative or c[0].sample:
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_cum_table); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 271, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 271, __pyx_L1_error)
(__pyx_v_c[0]).cum_table = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_1)));
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":272
* c[0].syn1neg = (np.PyArray_DATA(model.syn1neg))
* c[0].cum_table = (np.PyArray_DATA(model.cum_table))
* c[0].cum_table_len = len(model.cum_table) # <<<<<<<<<<<<<<
* if c[0].negative or c[0].sample:
* c[0].next_random = (2**24) * model.random.randint(0, 2**24) + model.random.randint(0, 2**24)
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_cum_table); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 272, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_8 = PyObject_Length(__pyx_t_1); if (unlikely(__pyx_t_8 == ((Py_ssize_t)-1))) __PYX_ERR(0, 272, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c[0]).cum_table_len = __pyx_t_8;
/* "gensim/models/doc2vec_inner.pyx":269
* c[0].syn1 = (np.PyArray_DATA(model.syn1))
*
* if c[0].negative: # <<<<<<<<<<<<<<
* c[0].syn1neg = (np.PyArray_DATA(model.syn1neg))
* c[0].cum_table = (np.PyArray_DATA(model.cum_table))
*/
}
/* "gensim/models/doc2vec_inner.pyx":273
* c[0].cum_table = (np.PyArray_DATA(model.cum_table))
* c[0].cum_table_len = len(model.cum_table)
* if c[0].negative or c[0].sample: # <<<<<<<<<<<<<<
* c[0].next_random = (2**24) * model.random.randint(0, 2**24) + model.random.randint(0, 2**24)
*
*/
__pyx_t_5 = ((__pyx_v_c[0]).negative != 0);
if (!__pyx_t_5) {
} else {
__pyx_t_6 = __pyx_t_5;
goto __pyx_L11_bool_binop_done;
}
__pyx_t_5 = ((__pyx_v_c[0]).sample != 0);
__pyx_t_6 = __pyx_t_5;
__pyx_L11_bool_binop_done:;
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":274
* c[0].cum_table_len = len(model.cum_table)
* if c[0].negative or c[0].sample:
* c[0].next_random = (2**24) * model.random.randint(0, 2**24) + model.random.randint(0, 2**24) # <<<<<<<<<<<<<<
*
* # convert Python structures to primitive types, so we can release the GIL
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_random); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_randint); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_7, __pyx_tuple__2, NULL); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = PyNumber_Multiply(__pyx_int_16777216, __pyx_t_1); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_random); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_randint); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_tuple__2, NULL); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_3 = PyNumber_Add(__pyx_t_7, __pyx_t_1); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_9 = __Pyx_PyInt_As_unsigned_PY_LONG_LONG(__pyx_t_3); if (unlikely((__pyx_t_9 == (unsigned PY_LONG_LONG)-1) && PyErr_Occurred())) __PYX_ERR(0, 274, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
(__pyx_v_c[0]).next_random = __pyx_t_9;
/* "gensim/models/doc2vec_inner.pyx":273
* c[0].cum_table = (np.PyArray_DATA(model.cum_table))
* c[0].cum_table_len = len(model.cum_table)
* if c[0].negative or c[0].sample: # <<<<<<<<<<<<<<
* c[0].next_random = (2**24) * model.random.randint(0, 2**24) + model.random.randint(0, 2**24)
*
*/
}
/* "gensim/models/doc2vec_inner.pyx":277
*
* # convert Python structures to primitive types, so we can release the GIL
* if work is None: # <<<<<<<<<<<<<<
* work = zeros(model.layer1_size, dtype=REAL)
* c[0].work = np.PyArray_DATA(work)
*/
__pyx_t_6 = (__pyx_v_work == Py_None);
__pyx_t_5 = (__pyx_t_6 != 0);
if (__pyx_t_5) {
/* "gensim/models/doc2vec_inner.pyx":278
* # convert Python structures to primitive types, so we can release the GIL
* if work is None:
* work = zeros(model.layer1_size, dtype=REAL) # <<<<<<<<<<<<<<
* c[0].work = np.PyArray_DATA(work)
* if neu1 is None:
*/
__Pyx_GetModuleGlobalName(__pyx_t_3, __pyx_n_s_zeros); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_layer1_size); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_GIVEREF(__pyx_t_1);
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_1);
__pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyDict_NewPresized(1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_GetModuleGlobalName(__pyx_t_10, __pyx_n_s_REAL); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_10);
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_10) < 0) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
__pyx_t_10 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_7, __pyx_t_1); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 278, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_10);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF_SET(__pyx_v_work, __pyx_t_10);
__pyx_t_10 = 0;
/* "gensim/models/doc2vec_inner.pyx":277
*
* # convert Python structures to primitive types, so we can release the GIL
* if work is None: # <<<<<<<<<<<<<<
* work = zeros(model.layer1_size, dtype=REAL)
* c[0].work = np.PyArray_DATA(work)
*/
}
/* "gensim/models/doc2vec_inner.pyx":279
* if work is None:
* work = zeros(model.layer1_size, dtype=REAL)
* c[0].work = np.PyArray_DATA(work) # <<<<<<<<<<<<<<
* if neu1 is None:
* neu1 = zeros(model.layer1_size, dtype=REAL)
*/
if (!(likely(((__pyx_v_work) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_work, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 279, __pyx_L1_error)
(__pyx_v_c[0]).work = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_v_work)));
/* "gensim/models/doc2vec_inner.pyx":280
* work = zeros(model.layer1_size, dtype=REAL)
* c[0].work = np.PyArray_DATA(work)
* if neu1 is None: # <<<<<<<<<<<<<<
* neu1 = zeros(model.layer1_size, dtype=REAL)
* c[0].neu1 = np.PyArray_DATA(neu1)
*/
__pyx_t_5 = (__pyx_v_neu1 == Py_None);
__pyx_t_6 = (__pyx_t_5 != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":281
* c[0].work = np.PyArray_DATA(work)
* if neu1 is None:
* neu1 = zeros(model.layer1_size, dtype=REAL) # <<<<<<<<<<<<<<
* c[0].neu1 = np.PyArray_DATA(neu1)
*
*/
__Pyx_GetModuleGlobalName(__pyx_t_10, __pyx_n_s_zeros); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_10);
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_layer1_size); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = PyTuple_New(1); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_GIVEREF(__pyx_t_1);
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_1);
__pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyDict_NewPresized(1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_GetModuleGlobalName(__pyx_t_3, __pyx_n_s_REAL); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_3) < 0) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_10, __pyx_t_7, __pyx_t_1); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 281, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF_SET(__pyx_v_neu1, __pyx_t_3);
__pyx_t_3 = 0;
/* "gensim/models/doc2vec_inner.pyx":280
* work = zeros(model.layer1_size, dtype=REAL)
* c[0].work = np.PyArray_DATA(work)
* if neu1 is None: # <<<<<<<<<<<<<<
* neu1 = zeros(model.layer1_size, dtype=REAL)
* c[0].neu1 = np.PyArray_DATA(neu1)
*/
}
/* "gensim/models/doc2vec_inner.pyx":282
* if neu1 is None:
* neu1 = zeros(model.layer1_size, dtype=REAL)
* c[0].neu1 = np.PyArray_DATA(neu1) # <<<<<<<<<<<<<<
*
*
*/
if (!(likely(((__pyx_v_neu1) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_neu1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 282, __pyx_L1_error)
(__pyx_v_c[0]).neu1 = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t *)PyArray_DATA(((PyArrayObject *)__pyx_v_neu1)));
/* "gensim/models/doc2vec_inner.pyx":227
*
*
* cdef init_d2v_config(Doc2VecConfig *c, model, alpha, learn_doctags, learn_words, learn_hidden, # <<<<<<<<<<<<<<
* train_words=False, work=None, neu1=None, word_vectors=None, words_lockf=None,
* doctag_vectors=None, doctags_lockf=None, docvecs_count=0):
*/
/* function exit code */
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_3);
__Pyx_XDECREF(__pyx_t_7);
__Pyx_XDECREF(__pyx_t_10);
__Pyx_AddTraceback("gensim.models.doc2vec_inner.init_d2v_config", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = 0;
__pyx_L0:;
__Pyx_XDECREF(__pyx_v_work);
__Pyx_XDECREF(__pyx_v_neu1);
__Pyx_XDECREF(__pyx_v_word_vectors);
__Pyx_XDECREF(__pyx_v_words_lockf);
__Pyx_XDECREF(__pyx_v_doctag_vectors);
__Pyx_XDECREF(__pyx_v_doctags_lockf);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "gensim/models/doc2vec_inner.pyx":286
*
*
* def train_document_dbow(model, doc_words, doctag_indexes, alpha, work=None, # <<<<<<<<<<<<<<
* train_words=False, learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* Python wrapper */
static PyObject *__pyx_pw_6gensim_6models_13doc2vec_inner_1train_document_dbow(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_6gensim_6models_13doc2vec_inner_train_document_dbow[] = "train_document_dbow(model, doc_words, doctag_indexes, alpha, work=None, train_words=False, learn_doctags=True, learn_words=True, learn_hidden=True, word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None)\nUpdate distributed bag of words model (\"PV-DBOW\") by training on a single document.\n\n Called internally from :meth:`~gensim.models.doc2vec.Doc2Vec.train` and\n :meth:`~gensim.models.doc2vec.Doc2Vec.infer_vector`.\n\n Parameters\n ----------\n model : :class:`~gensim.models.doc2vec.Doc2Vec`\n The model to train.\n doc_words : list of str\n The input document as a list of words to be used for training. Each word will be looked up in\n the model's vocabulary.\n doctag_indexes : list of int\n Indices into `doctag_vectors` used to obtain the tags of the document.\n alpha : float\n Learning rate.\n work : list of float, optional\n Updates to be performed on each neuron in the hidden layer of the underlying network.\n train_words : bool, optional\n Word vectors will be updated exactly as per Word2Vec skip-gram training only if **both** `learn_words`\n and `train_words` are set to True.\n learn_doctags : bool, optional\n Whether the tag vectors should be updated.\n learn_words : bool, optional\n Word vectors will be updated exactly as per Word2Vec skip-gram training only if **both**\n `learn_words` and `train_words` are set to True.\n learn_hidden : bool, optional\n Whether or not the weights of the hidden layer will be updated.\n word_vectors : numpy.ndarray, optional\n The vector representation for each word in the vocabulary. If None, these will be retrieved from the model.\n words_lockf : numpy.ndarray, optional\n EXPERIMENTAL. A learning lock factor for each word-vector; value 0.0 completely blocks updates, a value\n of 1.0 allows normal updates to word-vectors.\n doctag_vectors : numpy.ndarr""ay, optional\n Vector representations of the tags. If None, these will be retrieved from the model.\n doctags_lockf : numpy.ndarray, optional\n EXPERIMENTAL. The lock factors for each tag, same as `words_lockf`, but for document-vectors.\n\n Returns\n -------\n int\n Number of words in the input document that were actually used for training.\n\n ";
static PyMethodDef __pyx_mdef_6gensim_6models_13doc2vec_inner_1train_document_dbow = {"train_document_dbow", (PyCFunction)(void*)(PyCFunctionWithKeywords)__pyx_pw_6gensim_6models_13doc2vec_inner_1train_document_dbow, METH_VARARGS|METH_KEYWORDS, __pyx_doc_6gensim_6models_13doc2vec_inner_train_document_dbow};
static PyObject *__pyx_pw_6gensim_6models_13doc2vec_inner_1train_document_dbow(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyObject *__pyx_v_model = 0;
PyObject *__pyx_v_doc_words = 0;
PyObject *__pyx_v_doctag_indexes = 0;
PyObject *__pyx_v_alpha = 0;
PyObject *__pyx_v_work = 0;
PyObject *__pyx_v_train_words = 0;
PyObject *__pyx_v_learn_doctags = 0;
PyObject *__pyx_v_learn_words = 0;
PyObject *__pyx_v_learn_hidden = 0;
PyObject *__pyx_v_word_vectors = 0;
PyObject *__pyx_v_words_lockf = 0;
PyObject *__pyx_v_doctag_vectors = 0;
PyObject *__pyx_v_doctags_lockf = 0;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
PyObject *__pyx_r = 0;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("train_document_dbow (wrapper)", 0);
{
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_model,&__pyx_n_s_doc_words,&__pyx_n_s_doctag_indexes,&__pyx_n_s_alpha,&__pyx_n_s_work,&__pyx_n_s_train_words,&__pyx_n_s_learn_doctags,&__pyx_n_s_learn_words,&__pyx_n_s_learn_hidden,&__pyx_n_s_word_vectors,&__pyx_n_s_words_lockf,&__pyx_n_s_doctag_vectors,&__pyx_n_s_doctags_lockf,0};
PyObject* values[13] = {0,0,0,0,0,0,0,0,0,0,0,0,0};
values[4] = ((PyObject *)Py_None);
/* "gensim/models/doc2vec_inner.pyx":287
*
* def train_document_dbow(model, doc_words, doctag_indexes, alpha, work=None,
* train_words=False, learn_doctags=True, learn_words=True, learn_hidden=True, # <<<<<<<<<<<<<<
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
* """Update distributed bag of words model ("PV-DBOW") by training on a single document.
*/
values[5] = ((PyObject *)Py_False);
values[6] = ((PyObject *)Py_True);
values[7] = ((PyObject *)Py_True);
values[8] = ((PyObject *)Py_True);
/* "gensim/models/doc2vec_inner.pyx":288
* def train_document_dbow(model, doc_words, doctag_indexes, alpha, work=None,
* train_words=False, learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None): # <<<<<<<<<<<<<<
* """Update distributed bag of words model ("PV-DBOW") by training on a single document.
*
*/
values[9] = ((PyObject *)Py_None);
values[10] = ((PyObject *)Py_None);
values[11] = ((PyObject *)Py_None);
values[12] = ((PyObject *)Py_None);
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
switch (pos_args) {
case 13: values[12] = PyTuple_GET_ITEM(__pyx_args, 12);
CYTHON_FALLTHROUGH;
case 12: values[11] = PyTuple_GET_ITEM(__pyx_args, 11);
CYTHON_FALLTHROUGH;
case 11: values[10] = PyTuple_GET_ITEM(__pyx_args, 10);
CYTHON_FALLTHROUGH;
case 10: values[9] = PyTuple_GET_ITEM(__pyx_args, 9);
CYTHON_FALLTHROUGH;
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
CYTHON_FALLTHROUGH;
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
CYTHON_FALLTHROUGH;
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
CYTHON_FALLTHROUGH;
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
CYTHON_FALLTHROUGH;
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
CYTHON_FALLTHROUGH;
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
CYTHON_FALLTHROUGH;
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
CYTHON_FALLTHROUGH;
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
CYTHON_FALLTHROUGH;
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
CYTHON_FALLTHROUGH;
case 0: break;
default: goto __pyx_L5_argtuple_error;
}
kw_args = PyDict_Size(__pyx_kwds);
switch (pos_args) {
case 0:
if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_model)) != 0)) kw_args--;
else goto __pyx_L5_argtuple_error;
CYTHON_FALLTHROUGH;
case 1:
if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doc_words)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dbow", 0, 4, 13, 1); __PYX_ERR(0, 286, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 2:
if (likely((values[2] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctag_indexes)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dbow", 0, 4, 13, 2); __PYX_ERR(0, 286, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 3:
if (likely((values[3] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_alpha)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dbow", 0, 4, 13, 3); __PYX_ERR(0, 286, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 4:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_work);
if (value) { values[4] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 5:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_train_words);
if (value) { values[5] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 6:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_doctags);
if (value) { values[6] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 7:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_words);
if (value) { values[7] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 8:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_hidden);
if (value) { values[8] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 9:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_word_vectors);
if (value) { values[9] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 10:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_words_lockf);
if (value) { values[10] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 11:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctag_vectors);
if (value) { values[11] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 12:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctags_lockf);
if (value) { values[12] = value; kw_args--; }
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "train_document_dbow") < 0)) __PYX_ERR(0, 286, __pyx_L3_error)
}
} else {
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 13: values[12] = PyTuple_GET_ITEM(__pyx_args, 12);
CYTHON_FALLTHROUGH;
case 12: values[11] = PyTuple_GET_ITEM(__pyx_args, 11);
CYTHON_FALLTHROUGH;
case 11: values[10] = PyTuple_GET_ITEM(__pyx_args, 10);
CYTHON_FALLTHROUGH;
case 10: values[9] = PyTuple_GET_ITEM(__pyx_args, 9);
CYTHON_FALLTHROUGH;
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
CYTHON_FALLTHROUGH;
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
CYTHON_FALLTHROUGH;
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
CYTHON_FALLTHROUGH;
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
CYTHON_FALLTHROUGH;
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
CYTHON_FALLTHROUGH;
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
break;
default: goto __pyx_L5_argtuple_error;
}
}
__pyx_v_model = values[0];
__pyx_v_doc_words = values[1];
__pyx_v_doctag_indexes = values[2];
__pyx_v_alpha = values[3];
__pyx_v_work = values[4];
__pyx_v_train_words = values[5];
__pyx_v_learn_doctags = values[6];
__pyx_v_learn_words = values[7];
__pyx_v_learn_hidden = values[8];
__pyx_v_word_vectors = values[9];
__pyx_v_words_lockf = values[10];
__pyx_v_doctag_vectors = values[11];
__pyx_v_doctags_lockf = values[12];
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("train_document_dbow", 0, 4, 13, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 286, __pyx_L3_error)
__pyx_L3_error:;
__Pyx_AddTraceback("gensim.models.doc2vec_inner.train_document_dbow", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_r = __pyx_pf_6gensim_6models_13doc2vec_inner_train_document_dbow(__pyx_self, __pyx_v_model, __pyx_v_doc_words, __pyx_v_doctag_indexes, __pyx_v_alpha, __pyx_v_work, __pyx_v_train_words, __pyx_v_learn_doctags, __pyx_v_learn_words, __pyx_v_learn_hidden, __pyx_v_word_vectors, __pyx_v_words_lockf, __pyx_v_doctag_vectors, __pyx_v_doctags_lockf);
/* "gensim/models/doc2vec_inner.pyx":286
*
*
* def train_document_dbow(model, doc_words, doctag_indexes, alpha, work=None, # <<<<<<<<<<<<<<
* train_words=False, learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* function exit code */
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
static PyObject *__pyx_pf_6gensim_6models_13doc2vec_inner_train_document_dbow(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_model, PyObject *__pyx_v_doc_words, PyObject *__pyx_v_doctag_indexes, PyObject *__pyx_v_alpha, PyObject *__pyx_v_work, PyObject *__pyx_v_train_words, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, PyObject *__pyx_v_word_vectors, PyObject *__pyx_v_words_lockf, PyObject *__pyx_v_doctag_vectors, PyObject *__pyx_v_doctags_lockf) {
struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig __pyx_v_c;
int __pyx_v_i;
int __pyx_v_j;
long __pyx_v_result;
__pyx_t_5numpy_uint32_t *__pyx_v_vocab_sample_ints;
PyObject *__pyx_v_vocab_codes = NULL;
PyObject *__pyx_v_vocab_points = NULL;
PyObject *__pyx_v_token = NULL;
PyObject *__pyx_v_word_index = NULL;
PyObject *__pyx_v_item = NULL;
long __pyx_v_k;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config __pyx_t_2;
Py_ssize_t __pyx_t_3;
long __pyx_t_4;
Py_ssize_t __pyx_t_5;
int __pyx_t_6;
PyObject *__pyx_t_7 = NULL;
PyObject *(*__pyx_t_8)(PyObject *);
PyObject *__pyx_t_9 = NULL;
PyObject *__pyx_t_10 = NULL;
int __pyx_t_11;
PyObject *__pyx_t_12 = NULL;
int __pyx_t_13;
__pyx_t_5numpy_uint32_t __pyx_t_14;
int __pyx_t_15;
PyObject *__pyx_t_16 = NULL;
int __pyx_t_17;
int __pyx_t_18;
long __pyx_t_19;
int __pyx_t_20;
int __pyx_t_21;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("train_document_dbow", 0);
/* "gensim/models/doc2vec_inner.pyx":336
*
* cdef int i, j
* cdef long result = 0 # <<<<<<<<<<<<<<
* cdef np.uint32_t *vocab_sample_ints
*
*/
__pyx_v_result = 0;
/* "gensim/models/doc2vec_inner.pyx":339
* cdef np.uint32_t *vocab_sample_ints
*
* init_d2v_config(&c, model, alpha, learn_doctags, learn_words, learn_hidden, train_words=train_words, work=work, # <<<<<<<<<<<<<<
* neu1=None, word_vectors=word_vectors, words_lockf=words_lockf,
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
*/
__pyx_t_2.__pyx_n = 7;
__pyx_t_2.train_words = __pyx_v_train_words;
__pyx_t_2.work = __pyx_v_work;
__pyx_t_2.neu1 = Py_None;
__pyx_t_2.word_vectors = __pyx_v_word_vectors;
__pyx_t_2.words_lockf = __pyx_v_words_lockf;
__pyx_t_2.doctag_vectors = __pyx_v_doctag_vectors;
__pyx_t_2.doctags_lockf = __pyx_v_doctags_lockf;
__pyx_t_1 = __pyx_f_6gensim_6models_13doc2vec_inner_init_d2v_config((&__pyx_v_c), __pyx_v_model, __pyx_v_alpha, __pyx_v_learn_doctags, __pyx_v_learn_words, __pyx_v_learn_hidden, &__pyx_t_2); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 339, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":342
* neu1=None, word_vectors=word_vectors, words_lockf=words_lockf,
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes)) # <<<<<<<<<<<<<<
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
*/
__pyx_t_3 = PyObject_Length(__pyx_v_doctag_indexes); if (unlikely(__pyx_t_3 == ((Py_ssize_t)-1))) __PYX_ERR(0, 342, __pyx_L1_error)
__pyx_t_4 = 0x2710;
if (((__pyx_t_3 < __pyx_t_4) != 0)) {
__pyx_t_5 = __pyx_t_3;
} else {
__pyx_t_5 = __pyx_t_4;
}
__pyx_v_c.doctag_len = ((int)__pyx_t_5);
/* "gensim/models/doc2vec_inner.pyx":343
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample: # <<<<<<<<<<<<<<
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs:
*/
__pyx_t_6 = (__pyx_v_c.sample != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":344
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int']) # <<<<<<<<<<<<<<
* if c.hs:
* vocab_codes = model.wv.expandos['code']
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 344, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 344, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_sample_int); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 344, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 344, __pyx_L1_error)
__pyx_v_vocab_sample_ints = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_1)));
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":343
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample: # <<<<<<<<<<<<<<
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs:
*/
}
/* "gensim/models/doc2vec_inner.pyx":345
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs: # <<<<<<<<<<<<<<
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point']
*/
__pyx_t_6 = (__pyx_v_c.hs != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":346
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs:
* vocab_codes = model.wv.expandos['code'] # <<<<<<<<<<<<<<
* vocab_points = model.wv.expandos['point']
*
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 346, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 346, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_code); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 346, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_vocab_codes = __pyx_t_1;
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":347
* if c.hs:
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point'] # <<<<<<<<<<<<<<
*
* i = 0
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 347, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 347, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_point); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 347, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_vocab_points = __pyx_t_1;
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":345
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs: # <<<<<<<<<<<<<<
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point']
*/
}
/* "gensim/models/doc2vec_inner.pyx":349
* vocab_points = model.wv.expandos['point']
*
* i = 0 # <<<<<<<<<<<<<<
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
*/
__pyx_v_i = 0;
/* "gensim/models/doc2vec_inner.pyx":350
*
* i = 0
* for token in doc_words: # <<<<<<<<<<<<<<
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
*/
if (likely(PyList_CheckExact(__pyx_v_doc_words)) || PyTuple_CheckExact(__pyx_v_doc_words)) {
__pyx_t_1 = __pyx_v_doc_words; __Pyx_INCREF(__pyx_t_1); __pyx_t_5 = 0;
__pyx_t_8 = NULL;
} else {
__pyx_t_5 = -1; __pyx_t_1 = PyObject_GetIter(__pyx_v_doc_words); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 350, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_8 = Py_TYPE(__pyx_t_1)->tp_iternext; if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 350, __pyx_L1_error)
}
for (;;) {
if (likely(!__pyx_t_8)) {
if (likely(PyList_CheckExact(__pyx_t_1))) {
if (__pyx_t_5 >= PyList_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_7 = PyList_GET_ITEM(__pyx_t_1, __pyx_t_5); __Pyx_INCREF(__pyx_t_7); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 350, __pyx_L1_error)
#else
__pyx_t_7 = PySequence_ITEM(__pyx_t_1, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 350, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
#endif
} else {
if (__pyx_t_5 >= PyTuple_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_7 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_5); __Pyx_INCREF(__pyx_t_7); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 350, __pyx_L1_error)
#else
__pyx_t_7 = PySequence_ITEM(__pyx_t_1, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 350, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
#endif
}
} else {
__pyx_t_7 = __pyx_t_8(__pyx_t_1);
if (unlikely(!__pyx_t_7)) {
PyObject* exc_type = PyErr_Occurred();
if (exc_type) {
if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
else __PYX_ERR(0, 350, __pyx_L1_error)
}
break;
}
__Pyx_GOTREF(__pyx_t_7);
}
__Pyx_XDECREF_SET(__pyx_v_token, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":351
* i = 0
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None) # <<<<<<<<<<<<<<
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
*/
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_9, __pyx_n_s_key_to_index); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_10);
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_10, __pyx_n_s_get); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
__pyx_t_10 = NULL;
__pyx_t_11 = 0;
if (CYTHON_UNPACK_METHODS && likely(PyMethod_Check(__pyx_t_9))) {
__pyx_t_10 = PyMethod_GET_SELF(__pyx_t_9);
if (likely(__pyx_t_10)) {
PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_9);
__Pyx_INCREF(__pyx_t_10);
__Pyx_INCREF(function);
__Pyx_DECREF_SET(__pyx_t_9, function);
__pyx_t_11 = 1;
}
}
#if CYTHON_FAST_PYCALL
if (PyFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[3] = {__pyx_t_10, __pyx_v_token, Py_None};
__pyx_t_7 = __Pyx_PyFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_11, 2+__pyx_t_11); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_7);
} else
#endif
#if CYTHON_FAST_PYCCALL
if (__Pyx_PyFastCFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[3] = {__pyx_t_10, __pyx_v_token, Py_None};
__pyx_t_7 = __Pyx_PyCFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_11, 2+__pyx_t_11); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_7);
} else
#endif
{
__pyx_t_12 = PyTuple_New(2+__pyx_t_11); if (unlikely(!__pyx_t_12)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_12);
if (__pyx_t_10) {
__Pyx_GIVEREF(__pyx_t_10); PyTuple_SET_ITEM(__pyx_t_12, 0, __pyx_t_10); __pyx_t_10 = NULL;
}
__Pyx_INCREF(__pyx_v_token);
__Pyx_GIVEREF(__pyx_v_token);
PyTuple_SET_ITEM(__pyx_t_12, 0+__pyx_t_11, __pyx_v_token);
__Pyx_INCREF(Py_None);
__Pyx_GIVEREF(Py_None);
PyTuple_SET_ITEM(__pyx_t_12, 1+__pyx_t_11, Py_None);
__pyx_t_7 = __Pyx_PyObject_Call(__pyx_t_9, __pyx_t_12, NULL); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 351, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
__Pyx_XDECREF_SET(__pyx_v_word_index, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":352
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word # <<<<<<<<<<<<<<
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
*/
__pyx_t_6 = (__pyx_v_word_index == Py_None);
__pyx_t_13 = (__pyx_t_6 != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":353
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged # <<<<<<<<<<<<<<
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue
*/
goto __pyx_L5_continue;
/* "gensim/models/doc2vec_inner.pyx":352
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word # <<<<<<<<<<<<<<
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
*/
}
/* "gensim/models/doc2vec_inner.pyx":354
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random): # <<<<<<<<<<<<<<
* continue
* c.indexes[i] = word_index
*/
__pyx_t_6 = (__pyx_v_c.sample != 0);
if (__pyx_t_6) {
} else {
__pyx_t_13 = __pyx_t_6;
goto __pyx_L9_bool_binop_done;
}
__pyx_t_3 = __Pyx_PyIndex_AsSsize_t(__pyx_v_word_index); if (unlikely((__pyx_t_3 == (Py_ssize_t)-1) && PyErr_Occurred())) __PYX_ERR(0, 354, __pyx_L1_error)
__pyx_t_6 = (((__pyx_v_vocab_sample_ints[__pyx_t_3]) < __pyx_f_6gensim_6models_14word2vec_inner_random_int32((&__pyx_v_c.next_random))) != 0);
__pyx_t_13 = __pyx_t_6;
__pyx_L9_bool_binop_done:;
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":355
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue # <<<<<<<<<<<<<<
* c.indexes[i] = word_index
* if c.hs:
*/
goto __pyx_L5_continue;
/* "gensim/models/doc2vec_inner.pyx":354
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random): # <<<<<<<<<<<<<<
* continue
* c.indexes[i] = word_index
*/
}
/* "gensim/models/doc2vec_inner.pyx":356
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue
* c.indexes[i] = word_index # <<<<<<<<<<<<<<
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index])
*/
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_v_word_index); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 356, __pyx_L1_error)
(__pyx_v_c.indexes[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":357
* continue
* c.indexes[i] = word_index
* if c.hs: # <<<<<<<<<<<<<<
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":358
* c.indexes[i] = word_index
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index]) # <<<<<<<<<<<<<<
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
*/
if (unlikely(!__pyx_v_vocab_codes)) { __Pyx_RaiseUnboundLocalError("vocab_codes"); __PYX_ERR(0, 358, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_codes, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 358, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_3 = PyObject_Length(__pyx_t_7); if (unlikely(__pyx_t_3 == ((Py_ssize_t)-1))) __PYX_ERR(0, 358, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
(__pyx_v_c.codelens[__pyx_v_i]) = ((int)__pyx_t_3);
/* "gensim/models/doc2vec_inner.pyx":359
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index]) # <<<<<<<<<<<<<<
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1
*/
if (unlikely(!__pyx_v_vocab_codes)) { __Pyx_RaiseUnboundLocalError("vocab_codes"); __PYX_ERR(0, 359, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_codes, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 359, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 359, __pyx_L1_error)
(__pyx_v_c.codes[__pyx_v_i]) = ((__pyx_t_5numpy_uint8_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_7)));
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":360
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index]) # <<<<<<<<<<<<<<
* result += 1
* i += 1
*/
if (unlikely(!__pyx_v_vocab_points)) { __Pyx_RaiseUnboundLocalError("vocab_points"); __PYX_ERR(0, 360, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_points, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 360, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 360, __pyx_L1_error)
(__pyx_v_c.points[__pyx_v_i]) = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_7)));
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":357
* continue
* c.indexes[i] = word_index
* if c.hs: # <<<<<<<<<<<<<<
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
*/
}
/* "gensim/models/doc2vec_inner.pyx":361
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1 # <<<<<<<<<<<<<<
* i += 1
* if i == MAX_DOCUMENT_LEN:
*/
__pyx_v_result = (__pyx_v_result + 1);
/* "gensim/models/doc2vec_inner.pyx":362
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1
* i += 1 # <<<<<<<<<<<<<<
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow?
*/
__pyx_v_i = (__pyx_v_i + 1);
/* "gensim/models/doc2vec_inner.pyx":363
* result += 1
* i += 1
* if i == MAX_DOCUMENT_LEN: # <<<<<<<<<<<<<<
* break # TODO: log warning, tally overflow?
* c.document_len = i
*/
__pyx_t_13 = ((__pyx_v_i == 0x2710) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":364
* i += 1
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow? # <<<<<<<<<<<<<<
* c.document_len = i
*
*/
goto __pyx_L6_break;
/* "gensim/models/doc2vec_inner.pyx":363
* result += 1
* i += 1
* if i == MAX_DOCUMENT_LEN: # <<<<<<<<<<<<<<
* break # TODO: log warning, tally overflow?
* c.document_len = i
*/
}
/* "gensim/models/doc2vec_inner.pyx":350
*
* i = 0
* for token in doc_words: # <<<<<<<<<<<<<<
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
*/
__pyx_L5_continue:;
}
__pyx_L6_break:;
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":365
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow?
* c.document_len = i # <<<<<<<<<<<<<<
*
* if c.train_words:
*/
__pyx_v_c.document_len = __pyx_v_i;
/* "gensim/models/doc2vec_inner.pyx":367
* c.document_len = i
*
* if c.train_words: # <<<<<<<<<<<<<<
* # single randint() call avoids a big thread-synchronization slowdown
* if model.shrink_windows:
*/
__pyx_t_13 = (__pyx_v_c.train_words != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":369
* if c.train_words:
* # single randint() call avoids a big thread-synchronization slowdown
* if model.shrink_windows: # <<<<<<<<<<<<<<
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)):
* c.reduced_windows[i] = item
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_shrink_windows); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 369, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_13 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_13 < 0)) __PYX_ERR(0, 369, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":370
* # single randint() call avoids a big thread-synchronization slowdown
* if model.shrink_windows:
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)): # <<<<<<<<<<<<<<
* c.reduced_windows[i] = item
* else:
*/
__pyx_t_11 = 0;
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_random); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_7, __pyx_n_s_randint); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = __Pyx_PyInt_From_int(__pyx_v_c.window); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_12 = __Pyx_PyInt_From_int(__pyx_v_c.document_len); if (unlikely(!__pyx_t_12)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_12);
__pyx_t_10 = NULL;
__pyx_t_15 = 0;
if (CYTHON_UNPACK_METHODS && likely(PyMethod_Check(__pyx_t_9))) {
__pyx_t_10 = PyMethod_GET_SELF(__pyx_t_9);
if (likely(__pyx_t_10)) {
PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_9);
__Pyx_INCREF(__pyx_t_10);
__Pyx_INCREF(function);
__Pyx_DECREF_SET(__pyx_t_9, function);
__pyx_t_15 = 1;
}
}
#if CYTHON_FAST_PYCALL
if (PyFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[4] = {__pyx_t_10, __pyx_int_0, __pyx_t_7, __pyx_t_12};
__pyx_t_1 = __Pyx_PyFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_15, 3+__pyx_t_15); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
} else
#endif
#if CYTHON_FAST_PYCCALL
if (__Pyx_PyFastCFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[4] = {__pyx_t_10, __pyx_int_0, __pyx_t_7, __pyx_t_12};
__pyx_t_1 = __Pyx_PyCFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_15, 3+__pyx_t_15); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
} else
#endif
{
__pyx_t_16 = PyTuple_New(3+__pyx_t_15); if (unlikely(!__pyx_t_16)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_16);
if (__pyx_t_10) {
__Pyx_GIVEREF(__pyx_t_10); PyTuple_SET_ITEM(__pyx_t_16, 0, __pyx_t_10); __pyx_t_10 = NULL;
}
__Pyx_INCREF(__pyx_int_0);
__Pyx_GIVEREF(__pyx_int_0);
PyTuple_SET_ITEM(__pyx_t_16, 0+__pyx_t_15, __pyx_int_0);
__Pyx_GIVEREF(__pyx_t_7);
PyTuple_SET_ITEM(__pyx_t_16, 1+__pyx_t_15, __pyx_t_7);
__Pyx_GIVEREF(__pyx_t_12);
PyTuple_SET_ITEM(__pyx_t_16, 2+__pyx_t_15, __pyx_t_12);
__pyx_t_7 = 0;
__pyx_t_12 = 0;
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_9, __pyx_t_16, NULL); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_16); __pyx_t_16 = 0;
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
if (likely(PyList_CheckExact(__pyx_t_1)) || PyTuple_CheckExact(__pyx_t_1)) {
__pyx_t_9 = __pyx_t_1; __Pyx_INCREF(__pyx_t_9); __pyx_t_5 = 0;
__pyx_t_8 = NULL;
} else {
__pyx_t_5 = -1; __pyx_t_9 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_8 = Py_TYPE(__pyx_t_9)->tp_iternext; if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 370, __pyx_L1_error)
}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
for (;;) {
if (likely(!__pyx_t_8)) {
if (likely(PyList_CheckExact(__pyx_t_9))) {
if (__pyx_t_5 >= PyList_GET_SIZE(__pyx_t_9)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_1 = PyList_GET_ITEM(__pyx_t_9, __pyx_t_5); __Pyx_INCREF(__pyx_t_1); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 370, __pyx_L1_error)
#else
__pyx_t_1 = PySequence_ITEM(__pyx_t_9, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
#endif
} else {
if (__pyx_t_5 >= PyTuple_GET_SIZE(__pyx_t_9)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_1 = PyTuple_GET_ITEM(__pyx_t_9, __pyx_t_5); __Pyx_INCREF(__pyx_t_1); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 370, __pyx_L1_error)
#else
__pyx_t_1 = PySequence_ITEM(__pyx_t_9, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 370, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
#endif
}
} else {
__pyx_t_1 = __pyx_t_8(__pyx_t_9);
if (unlikely(!__pyx_t_1)) {
PyObject* exc_type = PyErr_Occurred();
if (exc_type) {
if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
else __PYX_ERR(0, 370, __pyx_L1_error)
}
break;
}
__Pyx_GOTREF(__pyx_t_1);
}
__Pyx_XDECREF_SET(__pyx_v_item, __pyx_t_1);
__pyx_t_1 = 0;
__pyx_v_i = __pyx_t_11;
__pyx_t_11 = (__pyx_t_11 + 1);
/* "gensim/models/doc2vec_inner.pyx":371
* if model.shrink_windows:
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)):
* c.reduced_windows[i] = item # <<<<<<<<<<<<<<
* else:
* for i in range(c.document_len):
*/
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_v_item); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 371, __pyx_L1_error)
(__pyx_v_c.reduced_windows[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":370
* # single randint() call avoids a big thread-synchronization slowdown
* if model.shrink_windows:
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)): # <<<<<<<<<<<<<<
* c.reduced_windows[i] = item
* else:
*/
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
/* "gensim/models/doc2vec_inner.pyx":369
* if c.train_words:
* # single randint() call avoids a big thread-synchronization slowdown
* if model.shrink_windows: # <<<<<<<<<<<<<<
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)):
* c.reduced_windows[i] = item
*/
goto __pyx_L14;
}
/* "gensim/models/doc2vec_inner.pyx":373
* c.reduced_windows[i] = item
* else:
* for i in range(c.document_len): # <<<<<<<<<<<<<<
* c.reduced_windows[i] = 0
*
*/
/*else*/ {
__pyx_t_11 = __pyx_v_c.document_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_15; __pyx_t_17+=1) {
__pyx_v_i = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":374
* else:
* for i in range(c.document_len):
* c.reduced_windows[i] = 0 # <<<<<<<<<<<<<<
*
* for i in range(c.doctag_len):
*/
(__pyx_v_c.reduced_windows[__pyx_v_i]) = 0;
}
}
__pyx_L14:;
/* "gensim/models/doc2vec_inner.pyx":367
* c.document_len = i
*
* if c.train_words: # <<<<<<<<<<<<<<
* # single randint() call avoids a big thread-synchronization slowdown
* if model.shrink_windows:
*/
}
/* "gensim/models/doc2vec_inner.pyx":376
* c.reduced_windows[i] = 0
*
* for i in range(c.doctag_len): # <<<<<<<<<<<<<<
* c.doctag_indexes[i] = doctag_indexes[i]
* result += 1
*/
__pyx_t_11 = __pyx_v_c.doctag_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_15; __pyx_t_17+=1) {
__pyx_v_i = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":377
*
* for i in range(c.doctag_len):
* c.doctag_indexes[i] = doctag_indexes[i] # <<<<<<<<<<<<<<
* result += 1
*
*/
__pyx_t_9 = __Pyx_GetItemInt(__pyx_v_doctag_indexes, __pyx_v_i, int, 1, __Pyx_PyInt_From_int, 0, 0, 0); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 377, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_t_9); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 377, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
(__pyx_v_c.doctag_indexes[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":378
* for i in range(c.doctag_len):
* c.doctag_indexes[i] = doctag_indexes[i]
* result += 1 # <<<<<<<<<<<<<<
*
* # release GIL & train on the document
*/
__pyx_v_result = (__pyx_v_result + 1);
}
/* "gensim/models/doc2vec_inner.pyx":381
*
* # release GIL & train on the document
* with nogil: # <<<<<<<<<<<<<<
* for i in range(c.document_len):
* if c.train_words: # simultaneous skip-gram wordvec-training
*/
{
#ifdef WITH_THREAD
PyThreadState *_save;
Py_UNBLOCK_THREADS
__Pyx_FastGIL_Remember();
#endif
/*try:*/ {
/* "gensim/models/doc2vec_inner.pyx":382
* # release GIL & train on the document
* with nogil:
* for i in range(c.document_len): # <<<<<<<<<<<<<<
* if c.train_words: # simultaneous skip-gram wordvec-training
* j = i - c.window + c.reduced_windows[i]
*/
__pyx_t_11 = __pyx_v_c.document_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_15; __pyx_t_17+=1) {
__pyx_v_i = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":383
* with nogil:
* for i in range(c.document_len):
* if c.train_words: # simultaneous skip-gram wordvec-training # <<<<<<<<<<<<<<
* j = i - c.window + c.reduced_windows[i]
* if j < 0:
*/
__pyx_t_13 = (__pyx_v_c.train_words != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":384
* for i in range(c.document_len):
* if c.train_words: # simultaneous skip-gram wordvec-training
* j = i - c.window + c.reduced_windows[i] # <<<<<<<<<<<<<<
* if j < 0:
* j = 0
*/
__pyx_v_j = ((__pyx_v_i - __pyx_v_c.window) + (__pyx_v_c.reduced_windows[__pyx_v_i]));
/* "gensim/models/doc2vec_inner.pyx":385
* if c.train_words: # simultaneous skip-gram wordvec-training
* j = i - c.window + c.reduced_windows[i]
* if j < 0: # <<<<<<<<<<<<<<
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
*/
__pyx_t_13 = ((__pyx_v_j < 0) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":386
* j = i - c.window + c.reduced_windows[i]
* if j < 0:
* j = 0 # <<<<<<<<<<<<<<
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len:
*/
__pyx_v_j = 0;
/* "gensim/models/doc2vec_inner.pyx":385
* if c.train_words: # simultaneous skip-gram wordvec-training
* j = i - c.window + c.reduced_windows[i]
* if j < 0: # <<<<<<<<<<<<<<
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
*/
}
/* "gensim/models/doc2vec_inner.pyx":387
* if j < 0:
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i] # <<<<<<<<<<<<<<
* if k > c.document_len:
* k = c.document_len
*/
__pyx_v_k = (((__pyx_v_i + __pyx_v_c.window) + 1) - (__pyx_v_c.reduced_windows[__pyx_v_i]));
/* "gensim/models/doc2vec_inner.pyx":388
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len: # <<<<<<<<<<<<<<
* k = c.document_len
* for j in range(j, k):
*/
__pyx_t_13 = ((__pyx_v_k > __pyx_v_c.document_len) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":389
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len:
* k = c.document_len # <<<<<<<<<<<<<<
* for j in range(j, k):
* if j == i:
*/
__pyx_t_18 = __pyx_v_c.document_len;
__pyx_v_k = __pyx_t_18;
/* "gensim/models/doc2vec_inner.pyx":388
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len: # <<<<<<<<<<<<<<
* k = c.document_len
* for j in range(j, k):
*/
}
/* "gensim/models/doc2vec_inner.pyx":390
* if k > c.document_len:
* k = c.document_len
* for j in range(j, k): # <<<<<<<<<<<<<<
* if j == i:
* continue
*/
__pyx_t_4 = __pyx_v_k;
__pyx_t_19 = __pyx_t_4;
for (__pyx_t_18 = __pyx_v_j; __pyx_t_18 < __pyx_t_19; __pyx_t_18+=1) {
__pyx_v_j = __pyx_t_18;
/* "gensim/models/doc2vec_inner.pyx":391
* k = c.document_len
* for j in range(j, k):
* if j == i: # <<<<<<<<<<<<<<
* continue
* if c.hs:
*/
__pyx_t_13 = ((__pyx_v_j == __pyx_v_i) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":392
* for j in range(j, k):
* if j == i:
* continue # <<<<<<<<<<<<<<
* if c.hs:
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
*/
goto __pyx_L29_continue;
/* "gensim/models/doc2vec_inner.pyx":391
* k = c.document_len
* for j in range(j, k):
* if j == i: # <<<<<<<<<<<<<<
* continue
* if c.hs:
*/
}
/* "gensim/models/doc2vec_inner.pyx":393
* if j == i:
* continue
* if c.hs: # <<<<<<<<<<<<<<
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.word_vectors, c.syn1, c.layer1_size,
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":395
* if c.hs:
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.word_vectors, c.syn1, c.layer1_size, # <<<<<<<<<<<<<<
* c.indexes[j], c.alpha, c.work, c.learn_words, c.learn_hidden, c.words_lockf,
* c.words_lockf_len)
*/
__pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_hs((__pyx_v_c.points[__pyx_v_i]), (__pyx_v_c.codes[__pyx_v_i]), (__pyx_v_c.codelens[__pyx_v_i]), __pyx_v_c.word_vectors, __pyx_v_c.syn1, __pyx_v_c.layer1_size, (__pyx_v_c.indexes[__pyx_v_j]), __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.learn_words, __pyx_v_c.learn_hidden, __pyx_v_c.words_lockf, __pyx_v_c.words_lockf_len);
/* "gensim/models/doc2vec_inner.pyx":393
* if j == i:
* continue
* if c.hs: # <<<<<<<<<<<<<<
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.word_vectors, c.syn1, c.layer1_size,
*/
}
/* "gensim/models/doc2vec_inner.pyx":398
* c.indexes[j], c.alpha, c.work, c.learn_words, c.learn_hidden, c.words_lockf,
* c.words_lockf_len)
* if c.negative: # <<<<<<<<<<<<<<
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
* c.next_random = fast_document_dbow_neg(c.negative, c.cum_table, c.cum_table_len, c.word_vectors,
*/
__pyx_t_13 = (__pyx_v_c.negative != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":400
* if c.negative:
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
* c.next_random = fast_document_dbow_neg(c.negative, c.cum_table, c.cum_table_len, c.word_vectors, # <<<<<<<<<<<<<<
* c.syn1neg, c.layer1_size, c.indexes[i], c.indexes[j],
* c.alpha, c.work, c.next_random, c.learn_words,
*/
__pyx_v_c.next_random = __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_neg(__pyx_v_c.negative, __pyx_v_c.cum_table, __pyx_v_c.cum_table_len, __pyx_v_c.word_vectors, __pyx_v_c.syn1neg, __pyx_v_c.layer1_size, (__pyx_v_c.indexes[__pyx_v_i]), (__pyx_v_c.indexes[__pyx_v_j]), __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.next_random, __pyx_v_c.learn_words, __pyx_v_c.learn_hidden, __pyx_v_c.words_lockf, __pyx_v_c.words_lockf_len);
/* "gensim/models/doc2vec_inner.pyx":398
* c.indexes[j], c.alpha, c.work, c.learn_words, c.learn_hidden, c.words_lockf,
* c.words_lockf_len)
* if c.negative: # <<<<<<<<<<<<<<
* # we reuse the DBOW function, as it is equivalent to skip-gram for this purpose
* c.next_random = fast_document_dbow_neg(c.negative, c.cum_table, c.cum_table_len, c.word_vectors,
*/
}
__pyx_L29_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":383
* with nogil:
* for i in range(c.document_len):
* if c.train_words: # simultaneous skip-gram wordvec-training # <<<<<<<<<<<<<<
* j = i - c.window + c.reduced_windows[i]
* if j < 0:
*/
}
/* "gensim/models/doc2vec_inner.pyx":406
*
* # docvec-training
* for j in range(c.doctag_len): # <<<<<<<<<<<<<<
* if c.hs:
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.doctag_vectors, c.syn1, c.layer1_size,
*/
__pyx_t_18 = __pyx_v_c.doctag_len;
__pyx_t_20 = __pyx_t_18;
for (__pyx_t_21 = 0; __pyx_t_21 < __pyx_t_20; __pyx_t_21+=1) {
__pyx_v_j = __pyx_t_21;
/* "gensim/models/doc2vec_inner.pyx":407
* # docvec-training
* for j in range(c.doctag_len):
* if c.hs: # <<<<<<<<<<<<<<
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.doctag_vectors, c.syn1, c.layer1_size,
* c.doctag_indexes[j], c.alpha, c.work, c.learn_doctags, c.learn_hidden, c.doctags_lockf,
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":408
* for j in range(c.doctag_len):
* if c.hs:
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.doctag_vectors, c.syn1, c.layer1_size, # <<<<<<<<<<<<<<
* c.doctag_indexes[j], c.alpha, c.work, c.learn_doctags, c.learn_hidden, c.doctags_lockf,
* c.doctags_lockf_len)
*/
__pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_hs((__pyx_v_c.points[__pyx_v_i]), (__pyx_v_c.codes[__pyx_v_i]), (__pyx_v_c.codelens[__pyx_v_i]), __pyx_v_c.doctag_vectors, __pyx_v_c.syn1, __pyx_v_c.layer1_size, (__pyx_v_c.doctag_indexes[__pyx_v_j]), __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.learn_doctags, __pyx_v_c.learn_hidden, __pyx_v_c.doctags_lockf, __pyx_v_c.doctags_lockf_len);
/* "gensim/models/doc2vec_inner.pyx":407
* # docvec-training
* for j in range(c.doctag_len):
* if c.hs: # <<<<<<<<<<<<<<
* fast_document_dbow_hs(c.points[i], c.codes[i], c.codelens[i], c.doctag_vectors, c.syn1, c.layer1_size,
* c.doctag_indexes[j], c.alpha, c.work, c.learn_doctags, c.learn_hidden, c.doctags_lockf,
*/
}
/* "gensim/models/doc2vec_inner.pyx":411
* c.doctag_indexes[j], c.alpha, c.work, c.learn_doctags, c.learn_hidden, c.doctags_lockf,
* c.doctags_lockf_len)
* if c.negative: # <<<<<<<<<<<<<<
* c.next_random = fast_document_dbow_neg(c.negative, c.cum_table, c.cum_table_len, c.doctag_vectors,
* c.syn1neg, c.layer1_size, c.indexes[i], c.doctag_indexes[j],
*/
__pyx_t_13 = (__pyx_v_c.negative != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":412
* c.doctags_lockf_len)
* if c.negative:
* c.next_random = fast_document_dbow_neg(c.negative, c.cum_table, c.cum_table_len, c.doctag_vectors, # <<<<<<<<<<<<<<
* c.syn1neg, c.layer1_size, c.indexes[i], c.doctag_indexes[j],
* c.alpha, c.work, c.next_random, c.learn_doctags,
*/
__pyx_v_c.next_random = __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dbow_neg(__pyx_v_c.negative, __pyx_v_c.cum_table, __pyx_v_c.cum_table_len, __pyx_v_c.doctag_vectors, __pyx_v_c.syn1neg, __pyx_v_c.layer1_size, (__pyx_v_c.indexes[__pyx_v_i]), (__pyx_v_c.doctag_indexes[__pyx_v_j]), __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.next_random, __pyx_v_c.learn_doctags, __pyx_v_c.learn_hidden, __pyx_v_c.doctags_lockf, __pyx_v_c.doctags_lockf_len);
/* "gensim/models/doc2vec_inner.pyx":411
* c.doctag_indexes[j], c.alpha, c.work, c.learn_doctags, c.learn_hidden, c.doctags_lockf,
* c.doctags_lockf_len)
* if c.negative: # <<<<<<<<<<<<<<
* c.next_random = fast_document_dbow_neg(c.negative, c.cum_table, c.cum_table_len, c.doctag_vectors,
* c.syn1neg, c.layer1_size, c.indexes[i], c.doctag_indexes[j],
*/
}
}
}
}
/* "gensim/models/doc2vec_inner.pyx":381
*
* # release GIL & train on the document
* with nogil: # <<<<<<<<<<<<<<
* for i in range(c.document_len):
* if c.train_words: # simultaneous skip-gram wordvec-training
*/
/*finally:*/ {
/*normal exit:*/{
#ifdef WITH_THREAD
__Pyx_FastGIL_Forget();
Py_BLOCK_THREADS
#endif
goto __pyx_L23;
}
__pyx_L23:;
}
}
/* "gensim/models/doc2vec_inner.pyx":417
* c.learn_hidden, c.doctags_lockf, c.doctags_lockf_len)
*
* return result # <<<<<<<<<<<<<<
*
*
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_9 = __Pyx_PyInt_From_long(__pyx_v_result); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 417, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_r = __pyx_t_9;
__pyx_t_9 = 0;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":286
*
*
* def train_document_dbow(model, doc_words, doctag_indexes, alpha, work=None, # <<<<<<<<<<<<<<
* train_words=False, learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_7);
__Pyx_XDECREF(__pyx_t_9);
__Pyx_XDECREF(__pyx_t_10);
__Pyx_XDECREF(__pyx_t_12);
__Pyx_XDECREF(__pyx_t_16);
__Pyx_AddTraceback("gensim.models.doc2vec_inner.train_document_dbow", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XDECREF(__pyx_v_vocab_codes);
__Pyx_XDECREF(__pyx_v_vocab_points);
__Pyx_XDECREF(__pyx_v_token);
__Pyx_XDECREF(__pyx_v_word_index);
__Pyx_XDECREF(__pyx_v_item);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "gensim/models/doc2vec_inner.pyx":420
*
*
* def train_document_dm(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, # <<<<<<<<<<<<<<
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* Python wrapper */
static PyObject *__pyx_pw_6gensim_6models_13doc2vec_inner_3train_document_dm(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_6gensim_6models_13doc2vec_inner_2train_document_dm[] = "train_document_dm(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, learn_doctags=True, learn_words=True, learn_hidden=True, word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None)\nUpdate distributed memory model (\"PV-DM\") by training on a single document.\n This method implements the DM model with a projection (input) layer that is either the sum or mean of the context\n vectors, depending on the model's `dm_mean` configuration field.\n\n Called internally from :meth:`~gensim.models.doc2vec.Doc2Vec.train` and\n :meth:`~gensim.models.doc2vec.Doc2Vec.infer_vector`.\n\n Parameters\n ----------\n model : :class:`~gensim.models.doc2vec.Doc2Vec`\n The model to train.\n doc_words : list of str\n The input document as a list of words to be used for training. Each word will be looked up in\n the model's vocabulary.\n doctag_indexes : list of int\n Indices into `doctag_vectors` used to obtain the tags of the document.\n alpha : float\n Learning rate.\n work : np.ndarray, optional\n Private working memory for each worker.\n neu1 : np.ndarray, optional\n Private working memory for each worker.\n learn_doctags : bool, optional\n Whether the tag vectors should be updated.\n learn_words : bool, optional\n Word vectors will be updated exactly as per Word2Vec skip-gram training only if **both**\n `learn_words` and `train_words` are set to True.\n learn_hidden : bool, optional\n Whether or not the weights of the hidden layer will be updated.\n word_vectors : numpy.ndarray, optional\n The vector representation for each word in the vocabulary. If None, these will be retrieved from the model.\n words_lockf : numpy.ndarray, optional\n EXPERIMENTAL. A learning lock factor for each word-vector; value 0.0 completely blocks updates, a value\n of 1.0 allows normal updates to word-vectors.\n doctag_vectors"" : numpy.ndarray, optional\n Vector representations of the tags. If None, these will be retrieved from the model.\n doctags_lockf : numpy.ndarray, optional\n EXPERIMENTAL. The lock factors for each tag, same as `words_lockf`, but for document-vectors.\n\n Returns\n -------\n int\n Number of words in the input document that were actually used for training.\n\n ";
static PyMethodDef __pyx_mdef_6gensim_6models_13doc2vec_inner_3train_document_dm = {"train_document_dm", (PyCFunction)(void*)(PyCFunctionWithKeywords)__pyx_pw_6gensim_6models_13doc2vec_inner_3train_document_dm, METH_VARARGS|METH_KEYWORDS, __pyx_doc_6gensim_6models_13doc2vec_inner_2train_document_dm};
static PyObject *__pyx_pw_6gensim_6models_13doc2vec_inner_3train_document_dm(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyObject *__pyx_v_model = 0;
PyObject *__pyx_v_doc_words = 0;
PyObject *__pyx_v_doctag_indexes = 0;
PyObject *__pyx_v_alpha = 0;
PyObject *__pyx_v_work = 0;
PyObject *__pyx_v_neu1 = 0;
PyObject *__pyx_v_learn_doctags = 0;
PyObject *__pyx_v_learn_words = 0;
PyObject *__pyx_v_learn_hidden = 0;
PyObject *__pyx_v_word_vectors = 0;
PyObject *__pyx_v_words_lockf = 0;
PyObject *__pyx_v_doctag_vectors = 0;
PyObject *__pyx_v_doctags_lockf = 0;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
PyObject *__pyx_r = 0;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("train_document_dm (wrapper)", 0);
{
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_model,&__pyx_n_s_doc_words,&__pyx_n_s_doctag_indexes,&__pyx_n_s_alpha,&__pyx_n_s_work,&__pyx_n_s_neu1,&__pyx_n_s_learn_doctags,&__pyx_n_s_learn_words,&__pyx_n_s_learn_hidden,&__pyx_n_s_word_vectors,&__pyx_n_s_words_lockf,&__pyx_n_s_doctag_vectors,&__pyx_n_s_doctags_lockf,0};
PyObject* values[13] = {0,0,0,0,0,0,0,0,0,0,0,0,0};
values[4] = ((PyObject *)Py_None);
values[5] = ((PyObject *)Py_None);
/* "gensim/models/doc2vec_inner.pyx":421
*
* def train_document_dm(model, doc_words, doctag_indexes, alpha, work=None, neu1=None,
* learn_doctags=True, learn_words=True, learn_hidden=True, # <<<<<<<<<<<<<<
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
* """Update distributed memory model ("PV-DM") by training on a single document.
*/
values[6] = ((PyObject *)Py_True);
values[7] = ((PyObject *)Py_True);
values[8] = ((PyObject *)Py_True);
/* "gensim/models/doc2vec_inner.pyx":422
* def train_document_dm(model, doc_words, doctag_indexes, alpha, work=None, neu1=None,
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None): # <<<<<<<<<<<<<<
* """Update distributed memory model ("PV-DM") by training on a single document.
* This method implements the DM model with a projection (input) layer that is either the sum or mean of the context
*/
values[9] = ((PyObject *)Py_None);
values[10] = ((PyObject *)Py_None);
values[11] = ((PyObject *)Py_None);
values[12] = ((PyObject *)Py_None);
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
switch (pos_args) {
case 13: values[12] = PyTuple_GET_ITEM(__pyx_args, 12);
CYTHON_FALLTHROUGH;
case 12: values[11] = PyTuple_GET_ITEM(__pyx_args, 11);
CYTHON_FALLTHROUGH;
case 11: values[10] = PyTuple_GET_ITEM(__pyx_args, 10);
CYTHON_FALLTHROUGH;
case 10: values[9] = PyTuple_GET_ITEM(__pyx_args, 9);
CYTHON_FALLTHROUGH;
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
CYTHON_FALLTHROUGH;
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
CYTHON_FALLTHROUGH;
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
CYTHON_FALLTHROUGH;
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
CYTHON_FALLTHROUGH;
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
CYTHON_FALLTHROUGH;
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
CYTHON_FALLTHROUGH;
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
CYTHON_FALLTHROUGH;
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
CYTHON_FALLTHROUGH;
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
CYTHON_FALLTHROUGH;
case 0: break;
default: goto __pyx_L5_argtuple_error;
}
kw_args = PyDict_Size(__pyx_kwds);
switch (pos_args) {
case 0:
if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_model)) != 0)) kw_args--;
else goto __pyx_L5_argtuple_error;
CYTHON_FALLTHROUGH;
case 1:
if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doc_words)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dm", 0, 4, 13, 1); __PYX_ERR(0, 420, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 2:
if (likely((values[2] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctag_indexes)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dm", 0, 4, 13, 2); __PYX_ERR(0, 420, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 3:
if (likely((values[3] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_alpha)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dm", 0, 4, 13, 3); __PYX_ERR(0, 420, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 4:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_work);
if (value) { values[4] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 5:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_neu1);
if (value) { values[5] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 6:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_doctags);
if (value) { values[6] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 7:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_words);
if (value) { values[7] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 8:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_hidden);
if (value) { values[8] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 9:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_word_vectors);
if (value) { values[9] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 10:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_words_lockf);
if (value) { values[10] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 11:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctag_vectors);
if (value) { values[11] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 12:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctags_lockf);
if (value) { values[12] = value; kw_args--; }
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "train_document_dm") < 0)) __PYX_ERR(0, 420, __pyx_L3_error)
}
} else {
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 13: values[12] = PyTuple_GET_ITEM(__pyx_args, 12);
CYTHON_FALLTHROUGH;
case 12: values[11] = PyTuple_GET_ITEM(__pyx_args, 11);
CYTHON_FALLTHROUGH;
case 11: values[10] = PyTuple_GET_ITEM(__pyx_args, 10);
CYTHON_FALLTHROUGH;
case 10: values[9] = PyTuple_GET_ITEM(__pyx_args, 9);
CYTHON_FALLTHROUGH;
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
CYTHON_FALLTHROUGH;
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
CYTHON_FALLTHROUGH;
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
CYTHON_FALLTHROUGH;
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
CYTHON_FALLTHROUGH;
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
CYTHON_FALLTHROUGH;
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
break;
default: goto __pyx_L5_argtuple_error;
}
}
__pyx_v_model = values[0];
__pyx_v_doc_words = values[1];
__pyx_v_doctag_indexes = values[2];
__pyx_v_alpha = values[3];
__pyx_v_work = values[4];
__pyx_v_neu1 = values[5];
__pyx_v_learn_doctags = values[6];
__pyx_v_learn_words = values[7];
__pyx_v_learn_hidden = values[8];
__pyx_v_word_vectors = values[9];
__pyx_v_words_lockf = values[10];
__pyx_v_doctag_vectors = values[11];
__pyx_v_doctags_lockf = values[12];
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("train_document_dm", 0, 4, 13, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 420, __pyx_L3_error)
__pyx_L3_error:;
__Pyx_AddTraceback("gensim.models.doc2vec_inner.train_document_dm", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_r = __pyx_pf_6gensim_6models_13doc2vec_inner_2train_document_dm(__pyx_self, __pyx_v_model, __pyx_v_doc_words, __pyx_v_doctag_indexes, __pyx_v_alpha, __pyx_v_work, __pyx_v_neu1, __pyx_v_learn_doctags, __pyx_v_learn_words, __pyx_v_learn_hidden, __pyx_v_word_vectors, __pyx_v_words_lockf, __pyx_v_doctag_vectors, __pyx_v_doctags_lockf);
/* "gensim/models/doc2vec_inner.pyx":420
*
*
* def train_document_dm(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, # <<<<<<<<<<<<<<
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* function exit code */
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
static PyObject *__pyx_pf_6gensim_6models_13doc2vec_inner_2train_document_dm(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_model, PyObject *__pyx_v_doc_words, PyObject *__pyx_v_doctag_indexes, PyObject *__pyx_v_alpha, PyObject *__pyx_v_work, PyObject *__pyx_v_neu1, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, PyObject *__pyx_v_word_vectors, PyObject *__pyx_v_words_lockf, PyObject *__pyx_v_doctag_vectors, PyObject *__pyx_v_doctags_lockf) {
struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig __pyx_v_c;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_count;
__pyx_t_6gensim_6models_14word2vec_inner_REAL_t __pyx_v_inv_count;
int __pyx_v_i;
int __pyx_v_j;
int __pyx_v_k;
int __pyx_v_m;
long __pyx_v_result;
__pyx_t_5numpy_uint32_t *__pyx_v_vocab_sample_ints;
PyObject *__pyx_v_vocab_codes = NULL;
PyObject *__pyx_v_vocab_points = NULL;
PyObject *__pyx_v_token = NULL;
PyObject *__pyx_v_word_index = NULL;
PyObject *__pyx_v_item = NULL;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config __pyx_t_2;
Py_ssize_t __pyx_t_3;
long __pyx_t_4;
Py_ssize_t __pyx_t_5;
int __pyx_t_6;
PyObject *__pyx_t_7 = NULL;
PyObject *(*__pyx_t_8)(PyObject *);
PyObject *__pyx_t_9 = NULL;
PyObject *__pyx_t_10 = NULL;
int __pyx_t_11;
PyObject *__pyx_t_12 = NULL;
int __pyx_t_13;
__pyx_t_5numpy_uint32_t __pyx_t_14;
int __pyx_t_15;
PyObject *__pyx_t_16 = NULL;
int __pyx_t_17;
int __pyx_t_18;
int __pyx_t_19;
int __pyx_t_20;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("train_document_dm", 0);
/* "gensim/models/doc2vec_inner.pyx":470
* cdef Doc2VecConfig c
*
* cdef REAL_t count, inv_count = 1.0 # <<<<<<<<<<<<<<
* cdef int i, j, k, m
* cdef long result = 0
*/
__pyx_v_inv_count = 1.0;
/* "gensim/models/doc2vec_inner.pyx":472
* cdef REAL_t count, inv_count = 1.0
* cdef int i, j, k, m
* cdef long result = 0 # <<<<<<<<<<<<<<
* cdef np.uint32_t *vocab_sample_ints
*
*/
__pyx_v_result = 0;
/* "gensim/models/doc2vec_inner.pyx":475
* cdef np.uint32_t *vocab_sample_ints
*
* init_d2v_config(&c, model, alpha, learn_doctags, learn_words, learn_hidden, train_words=False, # <<<<<<<<<<<<<<
* work=work, neu1=neu1, word_vectors=word_vectors, words_lockf=words_lockf,
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
*/
__pyx_t_2.__pyx_n = 7;
__pyx_t_2.train_words = Py_False;
__pyx_t_2.work = __pyx_v_work;
__pyx_t_2.neu1 = __pyx_v_neu1;
__pyx_t_2.word_vectors = __pyx_v_word_vectors;
__pyx_t_2.words_lockf = __pyx_v_words_lockf;
__pyx_t_2.doctag_vectors = __pyx_v_doctag_vectors;
__pyx_t_2.doctags_lockf = __pyx_v_doctags_lockf;
__pyx_t_1 = __pyx_f_6gensim_6models_13doc2vec_inner_init_d2v_config((&__pyx_v_c), __pyx_v_model, __pyx_v_alpha, __pyx_v_learn_doctags, __pyx_v_learn_words, __pyx_v_learn_hidden, &__pyx_t_2); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 475, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":478
* work=work, neu1=neu1, word_vectors=word_vectors, words_lockf=words_lockf,
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes)) # <<<<<<<<<<<<<<
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
*/
__pyx_t_3 = PyObject_Length(__pyx_v_doctag_indexes); if (unlikely(__pyx_t_3 == ((Py_ssize_t)-1))) __PYX_ERR(0, 478, __pyx_L1_error)
__pyx_t_4 = 0x2710;
if (((__pyx_t_3 < __pyx_t_4) != 0)) {
__pyx_t_5 = __pyx_t_3;
} else {
__pyx_t_5 = __pyx_t_4;
}
__pyx_v_c.doctag_len = ((int)__pyx_t_5);
/* "gensim/models/doc2vec_inner.pyx":479
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample: # <<<<<<<<<<<<<<
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* # vocab_sample_ints = model.wv.expandos['sample_int'] # this variant noticeably slower
*/
__pyx_t_6 = (__pyx_v_c.sample != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":480
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int']) # <<<<<<<<<<<<<<
* # vocab_sample_ints = model.wv.expandos['sample_int'] # this variant noticeably slower
* if c.hs:
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 480, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 480, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_sample_int); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 480, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 480, __pyx_L1_error)
__pyx_v_vocab_sample_ints = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_1)));
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":479
* doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample: # <<<<<<<<<<<<<<
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* # vocab_sample_ints = model.wv.expandos['sample_int'] # this variant noticeably slower
*/
}
/* "gensim/models/doc2vec_inner.pyx":482
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* # vocab_sample_ints = model.wv.expandos['sample_int'] # this variant noticeably slower
* if c.hs: # <<<<<<<<<<<<<<
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point']
*/
__pyx_t_6 = (__pyx_v_c.hs != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":483
* # vocab_sample_ints = model.wv.expandos['sample_int'] # this variant noticeably slower
* if c.hs:
* vocab_codes = model.wv.expandos['code'] # <<<<<<<<<<<<<<
* vocab_points = model.wv.expandos['point']
*
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 483, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 483, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_code); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 483, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_vocab_codes = __pyx_t_1;
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":484
* if c.hs:
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point'] # <<<<<<<<<<<<<<
*
* i = 0
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 484, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 484, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_point); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 484, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_vocab_points = __pyx_t_1;
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":482
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* # vocab_sample_ints = model.wv.expandos['sample_int'] # this variant noticeably slower
* if c.hs: # <<<<<<<<<<<<<<
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point']
*/
}
/* "gensim/models/doc2vec_inner.pyx":486
* vocab_points = model.wv.expandos['point']
*
* i = 0 # <<<<<<<<<<<<<<
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
*/
__pyx_v_i = 0;
/* "gensim/models/doc2vec_inner.pyx":487
*
* i = 0
* for token in doc_words: # <<<<<<<<<<<<<<
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
*/
if (likely(PyList_CheckExact(__pyx_v_doc_words)) || PyTuple_CheckExact(__pyx_v_doc_words)) {
__pyx_t_1 = __pyx_v_doc_words; __Pyx_INCREF(__pyx_t_1); __pyx_t_5 = 0;
__pyx_t_8 = NULL;
} else {
__pyx_t_5 = -1; __pyx_t_1 = PyObject_GetIter(__pyx_v_doc_words); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 487, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_8 = Py_TYPE(__pyx_t_1)->tp_iternext; if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 487, __pyx_L1_error)
}
for (;;) {
if (likely(!__pyx_t_8)) {
if (likely(PyList_CheckExact(__pyx_t_1))) {
if (__pyx_t_5 >= PyList_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_7 = PyList_GET_ITEM(__pyx_t_1, __pyx_t_5); __Pyx_INCREF(__pyx_t_7); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 487, __pyx_L1_error)
#else
__pyx_t_7 = PySequence_ITEM(__pyx_t_1, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 487, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
#endif
} else {
if (__pyx_t_5 >= PyTuple_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_7 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_5); __Pyx_INCREF(__pyx_t_7); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 487, __pyx_L1_error)
#else
__pyx_t_7 = PySequence_ITEM(__pyx_t_1, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 487, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
#endif
}
} else {
__pyx_t_7 = __pyx_t_8(__pyx_t_1);
if (unlikely(!__pyx_t_7)) {
PyObject* exc_type = PyErr_Occurred();
if (exc_type) {
if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
else __PYX_ERR(0, 487, __pyx_L1_error)
}
break;
}
__Pyx_GOTREF(__pyx_t_7);
}
__Pyx_XDECREF_SET(__pyx_v_token, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":488
* i = 0
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None) # <<<<<<<<<<<<<<
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
*/
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_9, __pyx_n_s_key_to_index); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_10);
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_10, __pyx_n_s_get); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
__pyx_t_10 = NULL;
__pyx_t_11 = 0;
if (CYTHON_UNPACK_METHODS && likely(PyMethod_Check(__pyx_t_9))) {
__pyx_t_10 = PyMethod_GET_SELF(__pyx_t_9);
if (likely(__pyx_t_10)) {
PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_9);
__Pyx_INCREF(__pyx_t_10);
__Pyx_INCREF(function);
__Pyx_DECREF_SET(__pyx_t_9, function);
__pyx_t_11 = 1;
}
}
#if CYTHON_FAST_PYCALL
if (PyFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[3] = {__pyx_t_10, __pyx_v_token, Py_None};
__pyx_t_7 = __Pyx_PyFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_11, 2+__pyx_t_11); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_7);
} else
#endif
#if CYTHON_FAST_PYCCALL
if (__Pyx_PyFastCFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[3] = {__pyx_t_10, __pyx_v_token, Py_None};
__pyx_t_7 = __Pyx_PyCFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_11, 2+__pyx_t_11); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_7);
} else
#endif
{
__pyx_t_12 = PyTuple_New(2+__pyx_t_11); if (unlikely(!__pyx_t_12)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_12);
if (__pyx_t_10) {
__Pyx_GIVEREF(__pyx_t_10); PyTuple_SET_ITEM(__pyx_t_12, 0, __pyx_t_10); __pyx_t_10 = NULL;
}
__Pyx_INCREF(__pyx_v_token);
__Pyx_GIVEREF(__pyx_v_token);
PyTuple_SET_ITEM(__pyx_t_12, 0+__pyx_t_11, __pyx_v_token);
__Pyx_INCREF(Py_None);
__Pyx_GIVEREF(Py_None);
PyTuple_SET_ITEM(__pyx_t_12, 1+__pyx_t_11, Py_None);
__pyx_t_7 = __Pyx_PyObject_Call(__pyx_t_9, __pyx_t_12, NULL); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 488, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
__Pyx_XDECREF_SET(__pyx_v_word_index, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":489
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word # <<<<<<<<<<<<<<
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
*/
__pyx_t_6 = (__pyx_v_word_index == Py_None);
__pyx_t_13 = (__pyx_t_6 != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":490
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged # <<<<<<<<<<<<<<
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue
*/
goto __pyx_L5_continue;
/* "gensim/models/doc2vec_inner.pyx":489
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word # <<<<<<<<<<<<<<
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
*/
}
/* "gensim/models/doc2vec_inner.pyx":491
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random): # <<<<<<<<<<<<<<
* continue
* c.indexes[i] = word_index
*/
__pyx_t_6 = (__pyx_v_c.sample != 0);
if (__pyx_t_6) {
} else {
__pyx_t_13 = __pyx_t_6;
goto __pyx_L9_bool_binop_done;
}
__pyx_t_3 = __Pyx_PyIndex_AsSsize_t(__pyx_v_word_index); if (unlikely((__pyx_t_3 == (Py_ssize_t)-1) && PyErr_Occurred())) __PYX_ERR(0, 491, __pyx_L1_error)
__pyx_t_6 = (((__pyx_v_vocab_sample_ints[__pyx_t_3]) < __pyx_f_6gensim_6models_14word2vec_inner_random_int32((&__pyx_v_c.next_random))) != 0);
__pyx_t_13 = __pyx_t_6;
__pyx_L9_bool_binop_done:;
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":492
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue # <<<<<<<<<<<<<<
* c.indexes[i] = word_index
* if c.hs:
*/
goto __pyx_L5_continue;
/* "gensim/models/doc2vec_inner.pyx":491
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random): # <<<<<<<<<<<<<<
* continue
* c.indexes[i] = word_index
*/
}
/* "gensim/models/doc2vec_inner.pyx":493
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue
* c.indexes[i] = word_index # <<<<<<<<<<<<<<
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index])
*/
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_v_word_index); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 493, __pyx_L1_error)
(__pyx_v_c.indexes[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":494
* continue
* c.indexes[i] = word_index
* if c.hs: # <<<<<<<<<<<<<<
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":495
* c.indexes[i] = word_index
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index]) # <<<<<<<<<<<<<<
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
*/
if (unlikely(!__pyx_v_vocab_codes)) { __Pyx_RaiseUnboundLocalError("vocab_codes"); __PYX_ERR(0, 495, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_codes, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 495, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_3 = PyObject_Length(__pyx_t_7); if (unlikely(__pyx_t_3 == ((Py_ssize_t)-1))) __PYX_ERR(0, 495, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
(__pyx_v_c.codelens[__pyx_v_i]) = ((int)__pyx_t_3);
/* "gensim/models/doc2vec_inner.pyx":496
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index]) # <<<<<<<<<<<<<<
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1
*/
if (unlikely(!__pyx_v_vocab_codes)) { __Pyx_RaiseUnboundLocalError("vocab_codes"); __PYX_ERR(0, 496, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_codes, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 496, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 496, __pyx_L1_error)
(__pyx_v_c.codes[__pyx_v_i]) = ((__pyx_t_5numpy_uint8_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_7)));
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":497
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index]) # <<<<<<<<<<<<<<
* result += 1
* i += 1
*/
if (unlikely(!__pyx_v_vocab_points)) { __Pyx_RaiseUnboundLocalError("vocab_points"); __PYX_ERR(0, 497, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_points, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 497, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 497, __pyx_L1_error)
(__pyx_v_c.points[__pyx_v_i]) = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_7)));
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":494
* continue
* c.indexes[i] = word_index
* if c.hs: # <<<<<<<<<<<<<<
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
*/
}
/* "gensim/models/doc2vec_inner.pyx":498
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1 # <<<<<<<<<<<<<<
* i += 1
* if i == MAX_DOCUMENT_LEN:
*/
__pyx_v_result = (__pyx_v_result + 1);
/* "gensim/models/doc2vec_inner.pyx":499
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1
* i += 1 # <<<<<<<<<<<<<<
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow?
*/
__pyx_v_i = (__pyx_v_i + 1);
/* "gensim/models/doc2vec_inner.pyx":500
* result += 1
* i += 1
* if i == MAX_DOCUMENT_LEN: # <<<<<<<<<<<<<<
* break # TODO: log warning, tally overflow?
* c.document_len = i
*/
__pyx_t_13 = ((__pyx_v_i == 0x2710) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":501
* i += 1
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow? # <<<<<<<<<<<<<<
* c.document_len = i
*
*/
goto __pyx_L6_break;
/* "gensim/models/doc2vec_inner.pyx":500
* result += 1
* i += 1
* if i == MAX_DOCUMENT_LEN: # <<<<<<<<<<<<<<
* break # TODO: log warning, tally overflow?
* c.document_len = i
*/
}
/* "gensim/models/doc2vec_inner.pyx":487
*
* i = 0
* for token in doc_words: # <<<<<<<<<<<<<<
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
*/
__pyx_L5_continue:;
}
__pyx_L6_break:;
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":502
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow?
* c.document_len = i # <<<<<<<<<<<<<<
*
* # single randint() call avoids a big thread-sync slowdown
*/
__pyx_v_c.document_len = __pyx_v_i;
/* "gensim/models/doc2vec_inner.pyx":505
*
* # single randint() call avoids a big thread-sync slowdown
* if model.shrink_windows: # <<<<<<<<<<<<<<
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)):
* c.reduced_windows[i] = item
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_shrink_windows); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 505, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_13 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_13 < 0)) __PYX_ERR(0, 505, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":506
* # single randint() call avoids a big thread-sync slowdown
* if model.shrink_windows:
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)): # <<<<<<<<<<<<<<
* c.reduced_windows[i] = item
* else:
*/
__pyx_t_11 = 0;
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_random); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_7, __pyx_n_s_randint); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = __Pyx_PyInt_From_int(__pyx_v_c.window); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_12 = __Pyx_PyInt_From_int(__pyx_v_c.document_len); if (unlikely(!__pyx_t_12)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_12);
__pyx_t_10 = NULL;
__pyx_t_15 = 0;
if (CYTHON_UNPACK_METHODS && likely(PyMethod_Check(__pyx_t_9))) {
__pyx_t_10 = PyMethod_GET_SELF(__pyx_t_9);
if (likely(__pyx_t_10)) {
PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_9);
__Pyx_INCREF(__pyx_t_10);
__Pyx_INCREF(function);
__Pyx_DECREF_SET(__pyx_t_9, function);
__pyx_t_15 = 1;
}
}
#if CYTHON_FAST_PYCALL
if (PyFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[4] = {__pyx_t_10, __pyx_int_0, __pyx_t_7, __pyx_t_12};
__pyx_t_1 = __Pyx_PyFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_15, 3+__pyx_t_15); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
} else
#endif
#if CYTHON_FAST_PYCCALL
if (__Pyx_PyFastCFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[4] = {__pyx_t_10, __pyx_int_0, __pyx_t_7, __pyx_t_12};
__pyx_t_1 = __Pyx_PyCFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_15, 3+__pyx_t_15); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
} else
#endif
{
__pyx_t_16 = PyTuple_New(3+__pyx_t_15); if (unlikely(!__pyx_t_16)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_16);
if (__pyx_t_10) {
__Pyx_GIVEREF(__pyx_t_10); PyTuple_SET_ITEM(__pyx_t_16, 0, __pyx_t_10); __pyx_t_10 = NULL;
}
__Pyx_INCREF(__pyx_int_0);
__Pyx_GIVEREF(__pyx_int_0);
PyTuple_SET_ITEM(__pyx_t_16, 0+__pyx_t_15, __pyx_int_0);
__Pyx_GIVEREF(__pyx_t_7);
PyTuple_SET_ITEM(__pyx_t_16, 1+__pyx_t_15, __pyx_t_7);
__Pyx_GIVEREF(__pyx_t_12);
PyTuple_SET_ITEM(__pyx_t_16, 2+__pyx_t_15, __pyx_t_12);
__pyx_t_7 = 0;
__pyx_t_12 = 0;
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_9, __pyx_t_16, NULL); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_16); __pyx_t_16 = 0;
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
if (likely(PyList_CheckExact(__pyx_t_1)) || PyTuple_CheckExact(__pyx_t_1)) {
__pyx_t_9 = __pyx_t_1; __Pyx_INCREF(__pyx_t_9); __pyx_t_5 = 0;
__pyx_t_8 = NULL;
} else {
__pyx_t_5 = -1; __pyx_t_9 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_8 = Py_TYPE(__pyx_t_9)->tp_iternext; if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 506, __pyx_L1_error)
}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
for (;;) {
if (likely(!__pyx_t_8)) {
if (likely(PyList_CheckExact(__pyx_t_9))) {
if (__pyx_t_5 >= PyList_GET_SIZE(__pyx_t_9)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_1 = PyList_GET_ITEM(__pyx_t_9, __pyx_t_5); __Pyx_INCREF(__pyx_t_1); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 506, __pyx_L1_error)
#else
__pyx_t_1 = PySequence_ITEM(__pyx_t_9, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
#endif
} else {
if (__pyx_t_5 >= PyTuple_GET_SIZE(__pyx_t_9)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_1 = PyTuple_GET_ITEM(__pyx_t_9, __pyx_t_5); __Pyx_INCREF(__pyx_t_1); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 506, __pyx_L1_error)
#else
__pyx_t_1 = PySequence_ITEM(__pyx_t_9, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 506, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
#endif
}
} else {
__pyx_t_1 = __pyx_t_8(__pyx_t_9);
if (unlikely(!__pyx_t_1)) {
PyObject* exc_type = PyErr_Occurred();
if (exc_type) {
if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
else __PYX_ERR(0, 506, __pyx_L1_error)
}
break;
}
__Pyx_GOTREF(__pyx_t_1);
}
__Pyx_XDECREF_SET(__pyx_v_item, __pyx_t_1);
__pyx_t_1 = 0;
__pyx_v_i = __pyx_t_11;
__pyx_t_11 = (__pyx_t_11 + 1);
/* "gensim/models/doc2vec_inner.pyx":507
* if model.shrink_windows:
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)):
* c.reduced_windows[i] = item # <<<<<<<<<<<<<<
* else:
* for i in range(c.document_len):
*/
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_v_item); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 507, __pyx_L1_error)
(__pyx_v_c.reduced_windows[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":506
* # single randint() call avoids a big thread-sync slowdown
* if model.shrink_windows:
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)): # <<<<<<<<<<<<<<
* c.reduced_windows[i] = item
* else:
*/
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
/* "gensim/models/doc2vec_inner.pyx":505
*
* # single randint() call avoids a big thread-sync slowdown
* if model.shrink_windows: # <<<<<<<<<<<<<<
* for i, item in enumerate(model.random.randint(0, c.window, c.document_len)):
* c.reduced_windows[i] = item
*/
goto __pyx_L13;
}
/* "gensim/models/doc2vec_inner.pyx":509
* c.reduced_windows[i] = item
* else:
* for i in range(c.document_len): # <<<<<<<<<<<<<<
* c.reduced_windows[i] = 0
*
*/
/*else*/ {
__pyx_t_11 = __pyx_v_c.document_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_15; __pyx_t_17+=1) {
__pyx_v_i = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":510
* else:
* for i in range(c.document_len):
* c.reduced_windows[i] = 0 # <<<<<<<<<<<<<<
*
* for i in range(c.doctag_len):
*/
(__pyx_v_c.reduced_windows[__pyx_v_i]) = 0;
}
}
__pyx_L13:;
/* "gensim/models/doc2vec_inner.pyx":512
* c.reduced_windows[i] = 0
*
* for i in range(c.doctag_len): # <<<<<<<<<<<<<<
* c.doctag_indexes[i] = doctag_indexes[i]
* result += 1
*/
__pyx_t_11 = __pyx_v_c.doctag_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_15; __pyx_t_17+=1) {
__pyx_v_i = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":513
*
* for i in range(c.doctag_len):
* c.doctag_indexes[i] = doctag_indexes[i] # <<<<<<<<<<<<<<
* result += 1
*
*/
__pyx_t_9 = __Pyx_GetItemInt(__pyx_v_doctag_indexes, __pyx_v_i, int, 1, __Pyx_PyInt_From_int, 0, 0, 0); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 513, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_t_9); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 513, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
(__pyx_v_c.doctag_indexes[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":514
* for i in range(c.doctag_len):
* c.doctag_indexes[i] = doctag_indexes[i]
* result += 1 # <<<<<<<<<<<<<<
*
* # release GIL & train on the document
*/
__pyx_v_result = (__pyx_v_result + 1);
}
/* "gensim/models/doc2vec_inner.pyx":517
*
* # release GIL & train on the document
* with nogil: # <<<<<<<<<<<<<<
* for i in range(c.document_len):
* j = i - c.window + c.reduced_windows[i]
*/
{
#ifdef WITH_THREAD
PyThreadState *_save;
Py_UNBLOCK_THREADS
__Pyx_FastGIL_Remember();
#endif
/*try:*/ {
/* "gensim/models/doc2vec_inner.pyx":518
* # release GIL & train on the document
* with nogil:
* for i in range(c.document_len): # <<<<<<<<<<<<<<
* j = i - c.window + c.reduced_windows[i]
* if j < 0:
*/
__pyx_t_11 = __pyx_v_c.document_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_15; __pyx_t_17+=1) {
__pyx_v_i = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":519
* with nogil:
* for i in range(c.document_len):
* j = i - c.window + c.reduced_windows[i] # <<<<<<<<<<<<<<
* if j < 0:
* j = 0
*/
__pyx_v_j = ((__pyx_v_i - __pyx_v_c.window) + (__pyx_v_c.reduced_windows[__pyx_v_i]));
/* "gensim/models/doc2vec_inner.pyx":520
* for i in range(c.document_len):
* j = i - c.window + c.reduced_windows[i]
* if j < 0: # <<<<<<<<<<<<<<
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
*/
__pyx_t_13 = ((__pyx_v_j < 0) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":521
* j = i - c.window + c.reduced_windows[i]
* if j < 0:
* j = 0 # <<<<<<<<<<<<<<
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len:
*/
__pyx_v_j = 0;
/* "gensim/models/doc2vec_inner.pyx":520
* for i in range(c.document_len):
* j = i - c.window + c.reduced_windows[i]
* if j < 0: # <<<<<<<<<<<<<<
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
*/
}
/* "gensim/models/doc2vec_inner.pyx":522
* if j < 0:
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i] # <<<<<<<<<<<<<<
* if k > c.document_len:
* k = c.document_len
*/
__pyx_v_k = (((__pyx_v_i + __pyx_v_c.window) + 1) - (__pyx_v_c.reduced_windows[__pyx_v_i]));
/* "gensim/models/doc2vec_inner.pyx":523
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len: # <<<<<<<<<<<<<<
* k = c.document_len
*
*/
__pyx_t_13 = ((__pyx_v_k > __pyx_v_c.document_len) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":524
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len:
* k = c.document_len # <<<<<<<<<<<<<<
*
* # compose l1 (in _neu1) & clear _work
*/
__pyx_t_18 = __pyx_v_c.document_len;
__pyx_v_k = __pyx_t_18;
/* "gensim/models/doc2vec_inner.pyx":523
* j = 0
* k = i + c.window + 1 - c.reduced_windows[i]
* if k > c.document_len: # <<<<<<<<<<<<<<
* k = c.document_len
*
*/
}
/* "gensim/models/doc2vec_inner.pyx":527
*
* # compose l1 (in _neu1) & clear _work
* memset(c.neu1, 0, c.layer1_size * cython.sizeof(REAL_t)) # <<<<<<<<<<<<<<
* count = 0.0
* for m in range(j, k):
*/
(void)(memset(__pyx_v_c.neu1, 0, (__pyx_v_c.layer1_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
/* "gensim/models/doc2vec_inner.pyx":528
* # compose l1 (in _neu1) & clear _work
* memset(c.neu1, 0, c.layer1_size * cython.sizeof(REAL_t))
* count = 0.0 # <<<<<<<<<<<<<<
* for m in range(j, k):
* if m == i:
*/
__pyx_v_count = ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)0.0);
/* "gensim/models/doc2vec_inner.pyx":529
* memset(c.neu1, 0, c.layer1_size * cython.sizeof(REAL_t))
* count = 0.0
* for m in range(j, k): # <<<<<<<<<<<<<<
* if m == i:
* continue
*/
__pyx_t_18 = __pyx_v_k;
__pyx_t_19 = __pyx_t_18;
for (__pyx_t_20 = __pyx_v_j; __pyx_t_20 < __pyx_t_19; __pyx_t_20+=1) {
__pyx_v_m = __pyx_t_20;
/* "gensim/models/doc2vec_inner.pyx":530
* count = 0.0
* for m in range(j, k):
* if m == i: # <<<<<<<<<<<<<<
* continue
* else:
*/
__pyx_t_13 = ((__pyx_v_m == __pyx_v_i) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":531
* for m in range(j, k):
* if m == i:
* continue # <<<<<<<<<<<<<<
* else:
* count += ONEF
*/
goto __pyx_L27_continue;
/* "gensim/models/doc2vec_inner.pyx":530
* count = 0.0
* for m in range(j, k):
* if m == i: # <<<<<<<<<<<<<<
* continue
* else:
*/
}
/* "gensim/models/doc2vec_inner.pyx":533
* continue
* else:
* count += ONEF # <<<<<<<<<<<<<<
* our_saxpy(&c.layer1_size, &ONEF, &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* for m in range(c.doctag_len):
*/
/*else*/ {
__pyx_v_count = (__pyx_v_count + __pyx_v_6gensim_6models_13doc2vec_inner_ONEF);
/* "gensim/models/doc2vec_inner.pyx":534
* else:
* count += ONEF
* our_saxpy(&c.layer1_size, &ONEF, &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE) # <<<<<<<<<<<<<<
* for m in range(c.doctag_len):
* count += ONEF
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_c.layer1_size), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONEF), (&(__pyx_v_c.word_vectors[((__pyx_v_c.indexes[__pyx_v_m]) * __pyx_v_c.layer1_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_c.neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
}
__pyx_L27_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":535
* count += ONEF
* our_saxpy(&c.layer1_size, &ONEF, &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* for m in range(c.doctag_len): # <<<<<<<<<<<<<<
* count += ONEF
* our_saxpy(&c.layer1_size, &ONEF, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
*/
__pyx_t_18 = __pyx_v_c.doctag_len;
__pyx_t_19 = __pyx_t_18;
for (__pyx_t_20 = 0; __pyx_t_20 < __pyx_t_19; __pyx_t_20+=1) {
__pyx_v_m = __pyx_t_20;
/* "gensim/models/doc2vec_inner.pyx":536
* our_saxpy(&c.layer1_size, &ONEF, &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* for m in range(c.doctag_len):
* count += ONEF # <<<<<<<<<<<<<<
* our_saxpy(&c.layer1_size, &ONEF, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* if count > (0.5):
*/
__pyx_v_count = (__pyx_v_count + __pyx_v_6gensim_6models_13doc2vec_inner_ONEF);
/* "gensim/models/doc2vec_inner.pyx":537
* for m in range(c.doctag_len):
* count += ONEF
* our_saxpy(&c.layer1_size, &ONEF, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE) # <<<<<<<<<<<<<<
* if count > (0.5):
* inv_count = ONEF/count
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_c.layer1_size), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONEF), (&(__pyx_v_c.doctag_vectors[((__pyx_v_c.doctag_indexes[__pyx_v_m]) * __pyx_v_c.layer1_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), __pyx_v_c.neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
}
/* "gensim/models/doc2vec_inner.pyx":538
* count += ONEF
* our_saxpy(&c.layer1_size, &ONEF, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* if count > (0.5): # <<<<<<<<<<<<<<
* inv_count = ONEF/count
* if c.cbow_mean:
*/
__pyx_t_13 = ((__pyx_v_count > ((__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)0.5)) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":539
* our_saxpy(&c.layer1_size, &ONEF, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* if count > (0.5):
* inv_count = ONEF/count # <<<<<<<<<<<<<<
* if c.cbow_mean:
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?)
*/
__pyx_v_inv_count = (__pyx_v_6gensim_6models_13doc2vec_inner_ONEF / __pyx_v_count);
/* "gensim/models/doc2vec_inner.pyx":538
* count += ONEF
* our_saxpy(&c.layer1_size, &ONEF, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE, c.neu1, &ONE)
* if count > (0.5): # <<<<<<<<<<<<<<
* inv_count = ONEF/count
* if c.cbow_mean:
*/
}
/* "gensim/models/doc2vec_inner.pyx":540
* if count > (0.5):
* inv_count = ONEF/count
* if c.cbow_mean: # <<<<<<<<<<<<<<
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?)
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
*/
__pyx_t_13 = (__pyx_v_c.cbow_mean != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":541
* inv_count = ONEF/count
* if c.cbow_mean:
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?) # <<<<<<<<<<<<<<
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
* if c.hs:
*/
__pyx_v_6gensim_6models_14word2vec_inner_sscal((&__pyx_v_c.layer1_size), (&__pyx_v_inv_count), __pyx_v_c.neu1, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":540
* if count > (0.5):
* inv_count = ONEF/count
* if c.cbow_mean: # <<<<<<<<<<<<<<
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?)
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
*/
}
/* "gensim/models/doc2vec_inner.pyx":542
* if c.cbow_mean:
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?)
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error # <<<<<<<<<<<<<<
* if c.hs:
* fast_document_dm_hs(c.points[i], c.codes[i], c.codelens[i], c.neu1, c.syn1, c.alpha, c.work,
*/
(void)(memset(__pyx_v_c.work, 0, (__pyx_v_c.layer1_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
/* "gensim/models/doc2vec_inner.pyx":543
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?)
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
* if c.hs: # <<<<<<<<<<<<<<
* fast_document_dm_hs(c.points[i], c.codes[i], c.codelens[i], c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.learn_hidden)
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":544
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
* if c.hs:
* fast_document_dm_hs(c.points[i], c.codes[i], c.codelens[i], c.neu1, c.syn1, c.alpha, c.work, # <<<<<<<<<<<<<<
* c.layer1_size, c.learn_hidden)
* if c.negative:
*/
__pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dm_hs((__pyx_v_c.points[__pyx_v_i]), (__pyx_v_c.codes[__pyx_v_i]), (__pyx_v_c.codelens[__pyx_v_i]), __pyx_v_c.neu1, __pyx_v_c.syn1, __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.layer1_size, __pyx_v_c.learn_hidden);
/* "gensim/models/doc2vec_inner.pyx":543
* sscal(&c.layer1_size, &inv_count, c.neu1, &ONE) # (does this need BLAS-variants like saxpy?)
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
* if c.hs: # <<<<<<<<<<<<<<
* fast_document_dm_hs(c.points[i], c.codes[i], c.codelens[i], c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.learn_hidden)
*/
}
/* "gensim/models/doc2vec_inner.pyx":546
* fast_document_dm_hs(c.points[i], c.codes[i], c.codelens[i], c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.learn_hidden)
* if c.negative: # <<<<<<<<<<<<<<
* c.next_random = fast_document_dm_neg(c.negative, c.cum_table, c.cum_table_len, c.next_random,
* c.neu1, c.syn1neg, c.indexes[i], c.alpha, c.work, c.layer1_size,
*/
__pyx_t_13 = (__pyx_v_c.negative != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":547
* c.layer1_size, c.learn_hidden)
* if c.negative:
* c.next_random = fast_document_dm_neg(c.negative, c.cum_table, c.cum_table_len, c.next_random, # <<<<<<<<<<<<<<
* c.neu1, c.syn1neg, c.indexes[i], c.alpha, c.work, c.layer1_size,
* c.learn_hidden)
*/
__pyx_v_c.next_random = __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dm_neg(__pyx_v_c.negative, __pyx_v_c.cum_table, __pyx_v_c.cum_table_len, __pyx_v_c.next_random, __pyx_v_c.neu1, __pyx_v_c.syn1neg, (__pyx_v_c.indexes[__pyx_v_i]), __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.layer1_size, __pyx_v_c.learn_hidden);
/* "gensim/models/doc2vec_inner.pyx":546
* fast_document_dm_hs(c.points[i], c.codes[i], c.codelens[i], c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.learn_hidden)
* if c.negative: # <<<<<<<<<<<<<<
* c.next_random = fast_document_dm_neg(c.negative, c.cum_table, c.cum_table_len, c.next_random,
* c.neu1, c.syn1neg, c.indexes[i], c.alpha, c.work, c.layer1_size,
*/
}
/* "gensim/models/doc2vec_inner.pyx":551
* c.learn_hidden)
*
* if not c.cbow_mean: # <<<<<<<<<<<<<<
* sscal(&c.layer1_size, &inv_count, c.work, &ONE) # (does this need BLAS-variants like saxpy?)
* # apply accumulated error in work
*/
__pyx_t_13 = ((!(__pyx_v_c.cbow_mean != 0)) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":552
*
* if not c.cbow_mean:
* sscal(&c.layer1_size, &inv_count, c.work, &ONE) # (does this need BLAS-variants like saxpy?) # <<<<<<<<<<<<<<
* # apply accumulated error in work
* if c.learn_doctags:
*/
__pyx_v_6gensim_6models_14word2vec_inner_sscal((&__pyx_v_c.layer1_size), (&__pyx_v_inv_count), __pyx_v_c.work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
/* "gensim/models/doc2vec_inner.pyx":551
* c.learn_hidden)
*
* if not c.cbow_mean: # <<<<<<<<<<<<<<
* sscal(&c.layer1_size, &inv_count, c.work, &ONE) # (does this need BLAS-variants like saxpy?)
* # apply accumulated error in work
*/
}
/* "gensim/models/doc2vec_inner.pyx":554
* sscal(&c.layer1_size, &inv_count, c.work, &ONE) # (does this need BLAS-variants like saxpy?)
* # apply accumulated error in work
* if c.learn_doctags: # <<<<<<<<<<<<<<
* for m in range(c.doctag_len):
* our_saxpy(&c.layer1_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], c.work,
*/
__pyx_t_13 = (__pyx_v_c.learn_doctags != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":555
* # apply accumulated error in work
* if c.learn_doctags:
* for m in range(c.doctag_len): # <<<<<<<<<<<<<<
* our_saxpy(&c.layer1_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], c.work,
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE)
*/
__pyx_t_18 = __pyx_v_c.doctag_len;
__pyx_t_19 = __pyx_t_18;
for (__pyx_t_20 = 0; __pyx_t_20 < __pyx_t_19; __pyx_t_20+=1) {
__pyx_v_m = __pyx_t_20;
/* "gensim/models/doc2vec_inner.pyx":556
* if c.learn_doctags:
* for m in range(c.doctag_len):
* our_saxpy(&c.layer1_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], c.work, # <<<<<<<<<<<<<<
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE)
* if c.learn_words:
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_c.layer1_size), (&(__pyx_v_c.doctags_lockf[((__pyx_v_c.doctag_indexes[__pyx_v_m]) % __pyx_v_c.doctags_lockf_len)])), __pyx_v_c.work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_c.doctag_vectors[((__pyx_v_c.doctag_indexes[__pyx_v_m]) * __pyx_v_c.layer1_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
}
/* "gensim/models/doc2vec_inner.pyx":554
* sscal(&c.layer1_size, &inv_count, c.work, &ONE) # (does this need BLAS-variants like saxpy?)
* # apply accumulated error in work
* if c.learn_doctags: # <<<<<<<<<<<<<<
* for m in range(c.doctag_len):
* our_saxpy(&c.layer1_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], c.work,
*/
}
/* "gensim/models/doc2vec_inner.pyx":558
* our_saxpy(&c.layer1_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], c.work,
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE)
* if c.learn_words: # <<<<<<<<<<<<<<
* for m in range(j, k):
* if m == i:
*/
__pyx_t_13 = (__pyx_v_c.learn_words != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":559
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE)
* if c.learn_words:
* for m in range(j, k): # <<<<<<<<<<<<<<
* if m == i:
* continue
*/
__pyx_t_18 = __pyx_v_k;
__pyx_t_19 = __pyx_t_18;
for (__pyx_t_20 = __pyx_v_j; __pyx_t_20 < __pyx_t_19; __pyx_t_20+=1) {
__pyx_v_m = __pyx_t_20;
/* "gensim/models/doc2vec_inner.pyx":560
* if c.learn_words:
* for m in range(j, k):
* if m == i: # <<<<<<<<<<<<<<
* continue
* else:
*/
__pyx_t_13 = ((__pyx_v_m == __pyx_v_i) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":561
* for m in range(j, k):
* if m == i:
* continue # <<<<<<<<<<<<<<
* else:
* our_saxpy(&c.layer1_size, &c.words_lockf[c.indexes[m] % c.doctags_lockf_len], c.work, &ONE,
*/
goto __pyx_L41_continue;
/* "gensim/models/doc2vec_inner.pyx":560
* if c.learn_words:
* for m in range(j, k):
* if m == i: # <<<<<<<<<<<<<<
* continue
* else:
*/
}
/* "gensim/models/doc2vec_inner.pyx":563
* continue
* else:
* our_saxpy(&c.layer1_size, &c.words_lockf[c.indexes[m] % c.doctags_lockf_len], c.work, &ONE, # <<<<<<<<<<<<<<
* &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE)
*
*/
/*else*/ {
/* "gensim/models/doc2vec_inner.pyx":564
* else:
* our_saxpy(&c.layer1_size, &c.words_lockf[c.indexes[m] % c.doctags_lockf_len], c.work, &ONE,
* &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE) # <<<<<<<<<<<<<<
*
* return result
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_c.layer1_size), (&(__pyx_v_c.words_lockf[((__pyx_v_c.indexes[__pyx_v_m]) % __pyx_v_c.doctags_lockf_len)])), __pyx_v_c.work, (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_c.word_vectors[((__pyx_v_c.indexes[__pyx_v_m]) * __pyx_v_c.layer1_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
}
__pyx_L41_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":558
* our_saxpy(&c.layer1_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], c.work,
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.layer1_size], &ONE)
* if c.learn_words: # <<<<<<<<<<<<<<
* for m in range(j, k):
* if m == i:
*/
}
}
}
/* "gensim/models/doc2vec_inner.pyx":517
*
* # release GIL & train on the document
* with nogil: # <<<<<<<<<<<<<<
* for i in range(c.document_len):
* j = i - c.window + c.reduced_windows[i]
*/
/*finally:*/ {
/*normal exit:*/{
#ifdef WITH_THREAD
__Pyx_FastGIL_Forget();
Py_BLOCK_THREADS
#endif
goto __pyx_L22;
}
__pyx_L22:;
}
}
/* "gensim/models/doc2vec_inner.pyx":566
* &c.word_vectors[c.indexes[m] * c.layer1_size], &ONE)
*
* return result # <<<<<<<<<<<<<<
*
*
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_9 = __Pyx_PyInt_From_long(__pyx_v_result); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 566, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_r = __pyx_t_9;
__pyx_t_9 = 0;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":420
*
*
* def train_document_dm(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, # <<<<<<<<<<<<<<
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_7);
__Pyx_XDECREF(__pyx_t_9);
__Pyx_XDECREF(__pyx_t_10);
__Pyx_XDECREF(__pyx_t_12);
__Pyx_XDECREF(__pyx_t_16);
__Pyx_AddTraceback("gensim.models.doc2vec_inner.train_document_dm", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XDECREF(__pyx_v_vocab_codes);
__Pyx_XDECREF(__pyx_v_vocab_points);
__Pyx_XDECREF(__pyx_v_token);
__Pyx_XDECREF(__pyx_v_word_index);
__Pyx_XDECREF(__pyx_v_item);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "gensim/models/doc2vec_inner.pyx":569
*
*
* def train_document_dm_concat(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, # <<<<<<<<<<<<<<
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* Python wrapper */
static PyObject *__pyx_pw_6gensim_6models_13doc2vec_inner_5train_document_dm_concat(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_6gensim_6models_13doc2vec_inner_4train_document_dm_concat[] = "train_document_dm_concat(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, learn_doctags=True, learn_words=True, learn_hidden=True, word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None)\nUpdate distributed memory model (\"PV-DM\") by training on a single document, using a concatenation of the\n context window word vectors (rather than a sum or average).\n This will be slower since the input at each batch will be significantly larger.\n\n Called internally from :meth:`~gensim.models.doc2vec.Doc2Vec.train` and\n :meth:`~gensim.models.doc2vec.Doc2Vec.infer_vector`.\n\n Parameters\n ----------\n model : :class:`~gensim.models.doc2vec.Doc2Vec`\n The model to train.\n doc_words : list of str\n The input document as a list of words to be used for training. Each word will be looked up in\n the model's vocabulary.\n doctag_indexes : list of int\n Indices into `doctag_vectors` used to obtain the tags of the document.\n alpha : float, optional\n Learning rate.\n work : np.ndarray, optional\n Private working memory for each worker.\n neu1 : np.ndarray, optional\n Private working memory for each worker.\n learn_doctags : bool, optional\n Whether the tag vectors should be updated.\n learn_words : bool, optional\n Word vectors will be updated exactly as per Word2Vec skip-gram training only if **both**\n `learn_words` and `train_words` are set to True.\n learn_hidden : bool, optional\n Whether or not the weights of the hidden layer will be updated.\n word_vectors : numpy.ndarray, optional\n The vector representation for each word in the vocabulary. If None, these will be retrieved from the model.\n words_lockf : numpy.ndarray, optional\n EXPERIMENTAL. A learning lock factor for each word-vector, value 0.0 completely blocks updates, a value\n of 1.0 allows normal updates to word-vectors.\n docta""g_vectors : numpy.ndarray, optional\n Vector representations of the tags. If None, these will be retrieved from the model.\n doctags_lockf : numpy.ndarray, optional\n EXPERIMENTAL. The lock factors for each tag, same as `words_lockf`, but for document-vectors.\n\n Returns\n -------\n int\n Number of words in the input document that were actually used for training.\n\n ";
static PyMethodDef __pyx_mdef_6gensim_6models_13doc2vec_inner_5train_document_dm_concat = {"train_document_dm_concat", (PyCFunction)(void*)(PyCFunctionWithKeywords)__pyx_pw_6gensim_6models_13doc2vec_inner_5train_document_dm_concat, METH_VARARGS|METH_KEYWORDS, __pyx_doc_6gensim_6models_13doc2vec_inner_4train_document_dm_concat};
static PyObject *__pyx_pw_6gensim_6models_13doc2vec_inner_5train_document_dm_concat(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyObject *__pyx_v_model = 0;
PyObject *__pyx_v_doc_words = 0;
PyObject *__pyx_v_doctag_indexes = 0;
PyObject *__pyx_v_alpha = 0;
PyObject *__pyx_v_work = 0;
PyObject *__pyx_v_neu1 = 0;
PyObject *__pyx_v_learn_doctags = 0;
PyObject *__pyx_v_learn_words = 0;
PyObject *__pyx_v_learn_hidden = 0;
PyObject *__pyx_v_word_vectors = 0;
PyObject *__pyx_v_words_lockf = 0;
PyObject *__pyx_v_doctag_vectors = 0;
PyObject *__pyx_v_doctags_lockf = 0;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
PyObject *__pyx_r = 0;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("train_document_dm_concat (wrapper)", 0);
{
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_model,&__pyx_n_s_doc_words,&__pyx_n_s_doctag_indexes,&__pyx_n_s_alpha,&__pyx_n_s_work,&__pyx_n_s_neu1,&__pyx_n_s_learn_doctags,&__pyx_n_s_learn_words,&__pyx_n_s_learn_hidden,&__pyx_n_s_word_vectors,&__pyx_n_s_words_lockf,&__pyx_n_s_doctag_vectors,&__pyx_n_s_doctags_lockf,0};
PyObject* values[13] = {0,0,0,0,0,0,0,0,0,0,0,0,0};
values[4] = ((PyObject *)Py_None);
values[5] = ((PyObject *)Py_None);
/* "gensim/models/doc2vec_inner.pyx":570
*
* def train_document_dm_concat(model, doc_words, doctag_indexes, alpha, work=None, neu1=None,
* learn_doctags=True, learn_words=True, learn_hidden=True, # <<<<<<<<<<<<<<
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
* """Update distributed memory model ("PV-DM") by training on a single document, using a concatenation of the
*/
values[6] = ((PyObject *)Py_True);
values[7] = ((PyObject *)Py_True);
values[8] = ((PyObject *)Py_True);
/* "gensim/models/doc2vec_inner.pyx":571
* def train_document_dm_concat(model, doc_words, doctag_indexes, alpha, work=None, neu1=None,
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None): # <<<<<<<<<<<<<<
* """Update distributed memory model ("PV-DM") by training on a single document, using a concatenation of the
* context window word vectors (rather than a sum or average).
*/
values[9] = ((PyObject *)Py_None);
values[10] = ((PyObject *)Py_None);
values[11] = ((PyObject *)Py_None);
values[12] = ((PyObject *)Py_None);
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
switch (pos_args) {
case 13: values[12] = PyTuple_GET_ITEM(__pyx_args, 12);
CYTHON_FALLTHROUGH;
case 12: values[11] = PyTuple_GET_ITEM(__pyx_args, 11);
CYTHON_FALLTHROUGH;
case 11: values[10] = PyTuple_GET_ITEM(__pyx_args, 10);
CYTHON_FALLTHROUGH;
case 10: values[9] = PyTuple_GET_ITEM(__pyx_args, 9);
CYTHON_FALLTHROUGH;
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
CYTHON_FALLTHROUGH;
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
CYTHON_FALLTHROUGH;
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
CYTHON_FALLTHROUGH;
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
CYTHON_FALLTHROUGH;
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
CYTHON_FALLTHROUGH;
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
CYTHON_FALLTHROUGH;
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
CYTHON_FALLTHROUGH;
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
CYTHON_FALLTHROUGH;
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
CYTHON_FALLTHROUGH;
case 0: break;
default: goto __pyx_L5_argtuple_error;
}
kw_args = PyDict_Size(__pyx_kwds);
switch (pos_args) {
case 0:
if (likely((values[0] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_model)) != 0)) kw_args--;
else goto __pyx_L5_argtuple_error;
CYTHON_FALLTHROUGH;
case 1:
if (likely((values[1] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doc_words)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dm_concat", 0, 4, 13, 1); __PYX_ERR(0, 569, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 2:
if (likely((values[2] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctag_indexes)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dm_concat", 0, 4, 13, 2); __PYX_ERR(0, 569, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 3:
if (likely((values[3] = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_alpha)) != 0)) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("train_document_dm_concat", 0, 4, 13, 3); __PYX_ERR(0, 569, __pyx_L3_error)
}
CYTHON_FALLTHROUGH;
case 4:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_work);
if (value) { values[4] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 5:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_neu1);
if (value) { values[5] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 6:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_doctags);
if (value) { values[6] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 7:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_words);
if (value) { values[7] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 8:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_learn_hidden);
if (value) { values[8] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 9:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_word_vectors);
if (value) { values[9] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 10:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_words_lockf);
if (value) { values[10] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 11:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctag_vectors);
if (value) { values[11] = value; kw_args--; }
}
CYTHON_FALLTHROUGH;
case 12:
if (kw_args > 0) {
PyObject* value = __Pyx_PyDict_GetItemStr(__pyx_kwds, __pyx_n_s_doctags_lockf);
if (value) { values[12] = value; kw_args--; }
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "train_document_dm_concat") < 0)) __PYX_ERR(0, 569, __pyx_L3_error)
}
} else {
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 13: values[12] = PyTuple_GET_ITEM(__pyx_args, 12);
CYTHON_FALLTHROUGH;
case 12: values[11] = PyTuple_GET_ITEM(__pyx_args, 11);
CYTHON_FALLTHROUGH;
case 11: values[10] = PyTuple_GET_ITEM(__pyx_args, 10);
CYTHON_FALLTHROUGH;
case 10: values[9] = PyTuple_GET_ITEM(__pyx_args, 9);
CYTHON_FALLTHROUGH;
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
CYTHON_FALLTHROUGH;
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
CYTHON_FALLTHROUGH;
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
CYTHON_FALLTHROUGH;
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
CYTHON_FALLTHROUGH;
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
CYTHON_FALLTHROUGH;
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
break;
default: goto __pyx_L5_argtuple_error;
}
}
__pyx_v_model = values[0];
__pyx_v_doc_words = values[1];
__pyx_v_doctag_indexes = values[2];
__pyx_v_alpha = values[3];
__pyx_v_work = values[4];
__pyx_v_neu1 = values[5];
__pyx_v_learn_doctags = values[6];
__pyx_v_learn_words = values[7];
__pyx_v_learn_hidden = values[8];
__pyx_v_word_vectors = values[9];
__pyx_v_words_lockf = values[10];
__pyx_v_doctag_vectors = values[11];
__pyx_v_doctags_lockf = values[12];
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("train_document_dm_concat", 0, 4, 13, PyTuple_GET_SIZE(__pyx_args)); __PYX_ERR(0, 569, __pyx_L3_error)
__pyx_L3_error:;
__Pyx_AddTraceback("gensim.models.doc2vec_inner.train_document_dm_concat", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_r = __pyx_pf_6gensim_6models_13doc2vec_inner_4train_document_dm_concat(__pyx_self, __pyx_v_model, __pyx_v_doc_words, __pyx_v_doctag_indexes, __pyx_v_alpha, __pyx_v_work, __pyx_v_neu1, __pyx_v_learn_doctags, __pyx_v_learn_words, __pyx_v_learn_hidden, __pyx_v_word_vectors, __pyx_v_words_lockf, __pyx_v_doctag_vectors, __pyx_v_doctags_lockf);
/* "gensim/models/doc2vec_inner.pyx":569
*
*
* def train_document_dm_concat(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, # <<<<<<<<<<<<<<
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* function exit code */
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
static PyObject *__pyx_pf_6gensim_6models_13doc2vec_inner_4train_document_dm_concat(CYTHON_UNUSED PyObject *__pyx_self, PyObject *__pyx_v_model, PyObject *__pyx_v_doc_words, PyObject *__pyx_v_doctag_indexes, PyObject *__pyx_v_alpha, PyObject *__pyx_v_work, PyObject *__pyx_v_neu1, PyObject *__pyx_v_learn_doctags, PyObject *__pyx_v_learn_words, PyObject *__pyx_v_learn_hidden, PyObject *__pyx_v_word_vectors, PyObject *__pyx_v_words_lockf, PyObject *__pyx_v_doctag_vectors, PyObject *__pyx_v_doctags_lockf) {
struct __pyx_t_6gensim_6models_13doc2vec_inner_Doc2VecConfig __pyx_v_c;
int __pyx_v_i;
int __pyx_v_j;
int __pyx_v_k;
int __pyx_v_m;
int __pyx_v_n;
long __pyx_v_result;
__pyx_t_5numpy_uint32_t *__pyx_v_vocab_sample_ints;
PyObject *__pyx_v_vocab_codes = NULL;
PyObject *__pyx_v_vocab_points = NULL;
PyObject *__pyx_v_token = NULL;
PyObject *__pyx_v_word_index = NULL;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
struct __pyx_opt_args_6gensim_6models_13doc2vec_inner_init_d2v_config __pyx_t_2;
Py_ssize_t __pyx_t_3;
long __pyx_t_4;
Py_ssize_t __pyx_t_5;
int __pyx_t_6;
PyObject *__pyx_t_7 = NULL;
PyObject *(*__pyx_t_8)(PyObject *);
PyObject *__pyx_t_9 = NULL;
PyObject *__pyx_t_10 = NULL;
int __pyx_t_11;
PyObject *__pyx_t_12 = NULL;
int __pyx_t_13;
__pyx_t_5numpy_uint32_t __pyx_t_14;
int __pyx_t_15;
int __pyx_t_16;
int __pyx_t_17;
int __pyx_t_18;
int __pyx_t_19;
int __pyx_t_20;
long __pyx_t_21;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("train_document_dm_concat", 0);
/* "gensim/models/doc2vec_inner.pyx":620
*
* cdef int i, j, k, m, n
* cdef long result = 0 # <<<<<<<<<<<<<<
* cdef np.uint32_t *vocab_sample_ints
*
*/
__pyx_v_result = 0;
/* "gensim/models/doc2vec_inner.pyx":623
* cdef np.uint32_t *vocab_sample_ints
*
* init_d2v_config(&c, model, alpha, learn_doctags, learn_words, learn_hidden, train_words=False, work=work, neu1=neu1, # <<<<<<<<<<<<<<
* word_vectors=word_vectors, words_lockf=words_lockf, doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
*/
__pyx_t_2.__pyx_n = 7;
__pyx_t_2.train_words = Py_False;
__pyx_t_2.work = __pyx_v_work;
__pyx_t_2.neu1 = __pyx_v_neu1;
__pyx_t_2.word_vectors = __pyx_v_word_vectors;
__pyx_t_2.words_lockf = __pyx_v_words_lockf;
__pyx_t_2.doctag_vectors = __pyx_v_doctag_vectors;
__pyx_t_2.doctags_lockf = __pyx_v_doctags_lockf;
__pyx_t_1 = __pyx_f_6gensim_6models_13doc2vec_inner_init_d2v_config((&__pyx_v_c), __pyx_v_model, __pyx_v_alpha, __pyx_v_learn_doctags, __pyx_v_learn_words, __pyx_v_learn_hidden, &__pyx_t_2); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 623, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":625
* init_d2v_config(&c, model, alpha, learn_doctags, learn_words, learn_hidden, train_words=False, work=work, neu1=neu1,
* word_vectors=word_vectors, words_lockf=words_lockf, doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes)) # <<<<<<<<<<<<<<
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
*/
__pyx_t_3 = PyObject_Length(__pyx_v_doctag_indexes); if (unlikely(__pyx_t_3 == ((Py_ssize_t)-1))) __PYX_ERR(0, 625, __pyx_L1_error)
__pyx_t_4 = 0x2710;
if (((__pyx_t_3 < __pyx_t_4) != 0)) {
__pyx_t_5 = __pyx_t_3;
} else {
__pyx_t_5 = __pyx_t_4;
}
__pyx_v_c.doctag_len = ((int)__pyx_t_5);
/* "gensim/models/doc2vec_inner.pyx":626
* word_vectors=word_vectors, words_lockf=words_lockf, doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample: # <<<<<<<<<<<<<<
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs:
*/
__pyx_t_6 = (__pyx_v_c.sample != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":627
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int']) # <<<<<<<<<<<<<<
* if c.hs:
* vocab_codes = model.wv.expandos['code']
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 627, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 627, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_sample_int); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 627, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 627, __pyx_L1_error)
__pyx_v_vocab_sample_ints = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_1)));
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":626
* word_vectors=word_vectors, words_lockf=words_lockf, doctag_vectors=doctag_vectors, doctags_lockf=doctags_lockf)
* c.doctag_len = min(MAX_DOCUMENT_LEN, len(doctag_indexes))
* if c.sample: # <<<<<<<<<<<<<<
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs:
*/
}
/* "gensim/models/doc2vec_inner.pyx":628
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs: # <<<<<<<<<<<<<<
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point']
*/
__pyx_t_6 = (__pyx_v_c.hs != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":629
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs:
* vocab_codes = model.wv.expandos['code'] # <<<<<<<<<<<<<<
* vocab_points = model.wv.expandos['point']
*
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 629, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 629, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_code); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 629, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_vocab_codes = __pyx_t_1;
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":630
* if c.hs:
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point'] # <<<<<<<<<<<<<<
*
* if c.doctag_len != c.expected_doctag_len:
*/
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 630, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_expandos); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 630, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_7, __pyx_n_u_point); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 630, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_vocab_points = __pyx_t_1;
__pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":628
* if c.sample:
* vocab_sample_ints = np.PyArray_DATA(model.wv.expandos['sample_int'])
* if c.hs: # <<<<<<<<<<<<<<
* vocab_codes = model.wv.expandos['code']
* vocab_points = model.wv.expandos['point']
*/
}
/* "gensim/models/doc2vec_inner.pyx":632
* vocab_points = model.wv.expandos['point']
*
* if c.doctag_len != c.expected_doctag_len: # <<<<<<<<<<<<<<
* return 0 # skip doc without expected number of tags
*
*/
__pyx_t_6 = ((__pyx_v_c.doctag_len != __pyx_v_c.expected_doctag_len) != 0);
if (__pyx_t_6) {
/* "gensim/models/doc2vec_inner.pyx":633
*
* if c.doctag_len != c.expected_doctag_len:
* return 0 # skip doc without expected number of tags # <<<<<<<<<<<<<<
*
* i = 0
*/
__Pyx_XDECREF(__pyx_r);
__Pyx_INCREF(__pyx_int_0);
__pyx_r = __pyx_int_0;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":632
* vocab_points = model.wv.expandos['point']
*
* if c.doctag_len != c.expected_doctag_len: # <<<<<<<<<<<<<<
* return 0 # skip doc without expected number of tags
*
*/
}
/* "gensim/models/doc2vec_inner.pyx":635
* return 0 # skip doc without expected number of tags
*
* i = 0 # <<<<<<<<<<<<<<
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
*/
__pyx_v_i = 0;
/* "gensim/models/doc2vec_inner.pyx":636
*
* i = 0
* for token in doc_words: # <<<<<<<<<<<<<<
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
*/
if (likely(PyList_CheckExact(__pyx_v_doc_words)) || PyTuple_CheckExact(__pyx_v_doc_words)) {
__pyx_t_1 = __pyx_v_doc_words; __Pyx_INCREF(__pyx_t_1); __pyx_t_5 = 0;
__pyx_t_8 = NULL;
} else {
__pyx_t_5 = -1; __pyx_t_1 = PyObject_GetIter(__pyx_v_doc_words); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 636, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_8 = Py_TYPE(__pyx_t_1)->tp_iternext; if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 636, __pyx_L1_error)
}
for (;;) {
if (likely(!__pyx_t_8)) {
if (likely(PyList_CheckExact(__pyx_t_1))) {
if (__pyx_t_5 >= PyList_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_7 = PyList_GET_ITEM(__pyx_t_1, __pyx_t_5); __Pyx_INCREF(__pyx_t_7); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 636, __pyx_L1_error)
#else
__pyx_t_7 = PySequence_ITEM(__pyx_t_1, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 636, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
#endif
} else {
if (__pyx_t_5 >= PyTuple_GET_SIZE(__pyx_t_1)) break;
#if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
__pyx_t_7 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_5); __Pyx_INCREF(__pyx_t_7); __pyx_t_5++; if (unlikely(0 < 0)) __PYX_ERR(0, 636, __pyx_L1_error)
#else
__pyx_t_7 = PySequence_ITEM(__pyx_t_1, __pyx_t_5); __pyx_t_5++; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 636, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
#endif
}
} else {
__pyx_t_7 = __pyx_t_8(__pyx_t_1);
if (unlikely(!__pyx_t_7)) {
PyObject* exc_type = PyErr_Occurred();
if (exc_type) {
if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
else __PYX_ERR(0, 636, __pyx_L1_error)
}
break;
}
__Pyx_GOTREF(__pyx_t_7);
}
__Pyx_XDECREF_SET(__pyx_v_token, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":637
* i = 0
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None) # <<<<<<<<<<<<<<
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
*/
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_v_model, __pyx_n_s_wv); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_9, __pyx_n_s_key_to_index); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_10);
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_10, __pyx_n_s_get); if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_9);
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
__pyx_t_10 = NULL;
__pyx_t_11 = 0;
if (CYTHON_UNPACK_METHODS && likely(PyMethod_Check(__pyx_t_9))) {
__pyx_t_10 = PyMethod_GET_SELF(__pyx_t_9);
if (likely(__pyx_t_10)) {
PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_9);
__Pyx_INCREF(__pyx_t_10);
__Pyx_INCREF(function);
__Pyx_DECREF_SET(__pyx_t_9, function);
__pyx_t_11 = 1;
}
}
#if CYTHON_FAST_PYCALL
if (PyFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[3] = {__pyx_t_10, __pyx_v_token, Py_None};
__pyx_t_7 = __Pyx_PyFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_11, 2+__pyx_t_11); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_7);
} else
#endif
#if CYTHON_FAST_PYCCALL
if (__Pyx_PyFastCFunction_Check(__pyx_t_9)) {
PyObject *__pyx_temp[3] = {__pyx_t_10, __pyx_v_token, Py_None};
__pyx_t_7 = __Pyx_PyCFunction_FastCall(__pyx_t_9, __pyx_temp+1-__pyx_t_11, 2+__pyx_t_11); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
__Pyx_GOTREF(__pyx_t_7);
} else
#endif
{
__pyx_t_12 = PyTuple_New(2+__pyx_t_11); if (unlikely(!__pyx_t_12)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_12);
if (__pyx_t_10) {
__Pyx_GIVEREF(__pyx_t_10); PyTuple_SET_ITEM(__pyx_t_12, 0, __pyx_t_10); __pyx_t_10 = NULL;
}
__Pyx_INCREF(__pyx_v_token);
__Pyx_GIVEREF(__pyx_v_token);
PyTuple_SET_ITEM(__pyx_t_12, 0+__pyx_t_11, __pyx_v_token);
__Pyx_INCREF(Py_None);
__Pyx_GIVEREF(Py_None);
PyTuple_SET_ITEM(__pyx_t_12, 1+__pyx_t_11, Py_None);
__pyx_t_7 = __Pyx_PyObject_Call(__pyx_t_9, __pyx_t_12, NULL); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 637, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_12); __pyx_t_12 = 0;
}
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
__Pyx_XDECREF_SET(__pyx_v_word_index, __pyx_t_7);
__pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":638
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word # <<<<<<<<<<<<<<
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
*/
__pyx_t_6 = (__pyx_v_word_index == Py_None);
__pyx_t_13 = (__pyx_t_6 != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":639
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged # <<<<<<<<<<<<<<
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue
*/
goto __pyx_L6_continue;
/* "gensim/models/doc2vec_inner.pyx":638
* for token in doc_words:
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word # <<<<<<<<<<<<<<
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
*/
}
/* "gensim/models/doc2vec_inner.pyx":640
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random): # <<<<<<<<<<<<<<
* continue
* c.indexes[i] = word_index
*/
__pyx_t_6 = (__pyx_v_c.sample != 0);
if (__pyx_t_6) {
} else {
__pyx_t_13 = __pyx_t_6;
goto __pyx_L10_bool_binop_done;
}
__pyx_t_3 = __Pyx_PyIndex_AsSsize_t(__pyx_v_word_index); if (unlikely((__pyx_t_3 == (Py_ssize_t)-1) && PyErr_Occurred())) __PYX_ERR(0, 640, __pyx_L1_error)
__pyx_t_6 = (((__pyx_v_vocab_sample_ints[__pyx_t_3]) < __pyx_f_6gensim_6models_14word2vec_inner_random_int32((&__pyx_v_c.next_random))) != 0);
__pyx_t_13 = __pyx_t_6;
__pyx_L10_bool_binop_done:;
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":641
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue # <<<<<<<<<<<<<<
* c.indexes[i] = word_index
* if c.hs:
*/
goto __pyx_L6_continue;
/* "gensim/models/doc2vec_inner.pyx":640
* if word_index is None: # shrink document to leave out word
* continue # leaving i unchanged
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random): # <<<<<<<<<<<<<<
* continue
* c.indexes[i] = word_index
*/
}
/* "gensim/models/doc2vec_inner.pyx":642
* if c.sample and vocab_sample_ints[word_index] < random_int32(&c.next_random):
* continue
* c.indexes[i] = word_index # <<<<<<<<<<<<<<
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index])
*/
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_v_word_index); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 642, __pyx_L1_error)
(__pyx_v_c.indexes[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":643
* continue
* c.indexes[i] = word_index
* if c.hs: # <<<<<<<<<<<<<<
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":644
* c.indexes[i] = word_index
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index]) # <<<<<<<<<<<<<<
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
*/
if (unlikely(!__pyx_v_vocab_codes)) { __Pyx_RaiseUnboundLocalError("vocab_codes"); __PYX_ERR(0, 644, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_codes, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 644, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_3 = PyObject_Length(__pyx_t_7); if (unlikely(__pyx_t_3 == ((Py_ssize_t)-1))) __PYX_ERR(0, 644, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
(__pyx_v_c.codelens[__pyx_v_i]) = ((int)__pyx_t_3);
/* "gensim/models/doc2vec_inner.pyx":645
* if c.hs:
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index]) # <<<<<<<<<<<<<<
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1
*/
if (unlikely(!__pyx_v_vocab_codes)) { __Pyx_RaiseUnboundLocalError("vocab_codes"); __PYX_ERR(0, 645, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_codes, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 645, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 645, __pyx_L1_error)
(__pyx_v_c.codes[__pyx_v_i]) = ((__pyx_t_5numpy_uint8_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_7)));
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":646
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index]) # <<<<<<<<<<<<<<
* result += 1
* i += 1
*/
if (unlikely(!__pyx_v_vocab_points)) { __Pyx_RaiseUnboundLocalError("vocab_points"); __PYX_ERR(0, 646, __pyx_L1_error) }
__pyx_t_7 = __Pyx_PyObject_GetItem(__pyx_v_vocab_points, __pyx_v_word_index); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 646, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_7);
if (!(likely(((__pyx_t_7) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_7, __pyx_ptype_5numpy_ndarray))))) __PYX_ERR(0, 646, __pyx_L1_error)
(__pyx_v_c.points[__pyx_v_i]) = ((__pyx_t_5numpy_uint32_t *)PyArray_DATA(((PyArrayObject *)__pyx_t_7)));
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
/* "gensim/models/doc2vec_inner.pyx":643
* continue
* c.indexes[i] = word_index
* if c.hs: # <<<<<<<<<<<<<<
* c.codelens[i] = len(vocab_codes[word_index])
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
*/
}
/* "gensim/models/doc2vec_inner.pyx":647
* c.codes[i] = np.PyArray_DATA(vocab_codes[word_index])
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1 # <<<<<<<<<<<<<<
* i += 1
* if i == MAX_DOCUMENT_LEN:
*/
__pyx_v_result = (__pyx_v_result + 1);
/* "gensim/models/doc2vec_inner.pyx":648
* c.points[i] = np.PyArray_DATA(vocab_points[word_index])
* result += 1
* i += 1 # <<<<<<<<<<<<<<
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow?
*/
__pyx_v_i = (__pyx_v_i + 1);
/* "gensim/models/doc2vec_inner.pyx":649
* result += 1
* i += 1
* if i == MAX_DOCUMENT_LEN: # <<<<<<<<<<<<<<
* break # TODO: log warning, tally overflow?
* c.document_len = i
*/
__pyx_t_13 = ((__pyx_v_i == 0x2710) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":650
* i += 1
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow? # <<<<<<<<<<<<<<
* c.document_len = i
*
*/
goto __pyx_L7_break;
/* "gensim/models/doc2vec_inner.pyx":649
* result += 1
* i += 1
* if i == MAX_DOCUMENT_LEN: # <<<<<<<<<<<<<<
* break # TODO: log warning, tally overflow?
* c.document_len = i
*/
}
/* "gensim/models/doc2vec_inner.pyx":636
*
* i = 0
* for token in doc_words: # <<<<<<<<<<<<<<
* word_index = model.wv.key_to_index.get(token, None)
* if word_index is None: # shrink document to leave out word
*/
__pyx_L6_continue:;
}
__pyx_L7_break:;
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "gensim/models/doc2vec_inner.pyx":651
* if i == MAX_DOCUMENT_LEN:
* break # TODO: log warning, tally overflow?
* c.document_len = i # <<<<<<<<<<<<<<
*
* for i in range(c.doctag_len):
*/
__pyx_v_c.document_len = __pyx_v_i;
/* "gensim/models/doc2vec_inner.pyx":653
* c.document_len = i
*
* for i in range(c.doctag_len): # <<<<<<<<<<<<<<
* c.doctag_indexes[i] = doctag_indexes[i]
* result += 1
*/
__pyx_t_11 = __pyx_v_c.doctag_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_16 = 0; __pyx_t_16 < __pyx_t_15; __pyx_t_16+=1) {
__pyx_v_i = __pyx_t_16;
/* "gensim/models/doc2vec_inner.pyx":654
*
* for i in range(c.doctag_len):
* c.doctag_indexes[i] = doctag_indexes[i] # <<<<<<<<<<<<<<
* result += 1
*
*/
__pyx_t_1 = __Pyx_GetItemInt(__pyx_v_doctag_indexes, __pyx_v_i, int, 1, __Pyx_PyInt_From_int, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 654, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_14 = __Pyx_PyInt_As_npy_uint32(__pyx_t_1); if (unlikely((__pyx_t_14 == ((npy_uint32)-1)) && PyErr_Occurred())) __PYX_ERR(0, 654, __pyx_L1_error)
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
(__pyx_v_c.doctag_indexes[__pyx_v_i]) = __pyx_t_14;
/* "gensim/models/doc2vec_inner.pyx":655
* for i in range(c.doctag_len):
* c.doctag_indexes[i] = doctag_indexes[i]
* result += 1 # <<<<<<<<<<<<<<
*
* # release GIL & train on the document
*/
__pyx_v_result = (__pyx_v_result + 1);
}
/* "gensim/models/doc2vec_inner.pyx":658
*
* # release GIL & train on the document
* with nogil: # <<<<<<<<<<<<<<
* for i in range(c.document_len):
* j = i - c.window # negative OK: will pad with null word
*/
{
#ifdef WITH_THREAD
PyThreadState *_save;
Py_UNBLOCK_THREADS
__Pyx_FastGIL_Remember();
#endif
/*try:*/ {
/* "gensim/models/doc2vec_inner.pyx":659
* # release GIL & train on the document
* with nogil:
* for i in range(c.document_len): # <<<<<<<<<<<<<<
* j = i - c.window # negative OK: will pad with null word
* k = i + c.window + 1 # past document end OK: will pad with null word
*/
__pyx_t_11 = __pyx_v_c.document_len;
__pyx_t_15 = __pyx_t_11;
for (__pyx_t_16 = 0; __pyx_t_16 < __pyx_t_15; __pyx_t_16+=1) {
__pyx_v_i = __pyx_t_16;
/* "gensim/models/doc2vec_inner.pyx":660
* with nogil:
* for i in range(c.document_len):
* j = i - c.window # negative OK: will pad with null word # <<<<<<<<<<<<<<
* k = i + c.window + 1 # past document end OK: will pad with null word
*
*/
__pyx_v_j = (__pyx_v_i - __pyx_v_c.window);
/* "gensim/models/doc2vec_inner.pyx":661
* for i in range(c.document_len):
* j = i - c.window # negative OK: will pad with null word
* k = i + c.window + 1 # past document end OK: will pad with null word # <<<<<<<<<<<<<<
*
* # compose l1 & clear work
*/
__pyx_v_k = ((__pyx_v_i + __pyx_v_c.window) + 1);
/* "gensim/models/doc2vec_inner.pyx":664
*
* # compose l1 & clear work
* for m in range(c.doctag_len): # <<<<<<<<<<<<<<
* # doc vector(s)
* memcpy(&c.neu1[m * c.vector_size], &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size],
*/
__pyx_t_17 = __pyx_v_c.doctag_len;
__pyx_t_18 = __pyx_t_17;
for (__pyx_t_19 = 0; __pyx_t_19 < __pyx_t_18; __pyx_t_19+=1) {
__pyx_v_m = __pyx_t_19;
/* "gensim/models/doc2vec_inner.pyx":666
* for m in range(c.doctag_len):
* # doc vector(s)
* memcpy(&c.neu1[m * c.vector_size], &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size], # <<<<<<<<<<<<<<
* c.vector_size * cython.sizeof(REAL_t))
* n = 0
*/
(void)(memcpy((&(__pyx_v_c.neu1[(__pyx_v_m * __pyx_v_c.vector_size)])), (&(__pyx_v_c.doctag_vectors[((__pyx_v_c.doctag_indexes[__pyx_v_m]) * __pyx_v_c.vector_size)])), (__pyx_v_c.vector_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
}
/* "gensim/models/doc2vec_inner.pyx":668
* memcpy(&c.neu1[m * c.vector_size], &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size],
* c.vector_size * cython.sizeof(REAL_t))
* n = 0 # <<<<<<<<<<<<<<
* for m in range(j, k):
* # word vectors in window
*/
__pyx_v_n = 0;
/* "gensim/models/doc2vec_inner.pyx":669
* c.vector_size * cython.sizeof(REAL_t))
* n = 0
* for m in range(j, k): # <<<<<<<<<<<<<<
* # word vectors in window
* if m == i:
*/
__pyx_t_17 = __pyx_v_k;
__pyx_t_18 = __pyx_t_17;
for (__pyx_t_19 = __pyx_v_j; __pyx_t_19 < __pyx_t_18; __pyx_t_19+=1) {
__pyx_v_m = __pyx_t_19;
/* "gensim/models/doc2vec_inner.pyx":671
* for m in range(j, k):
* # word vectors in window
* if m == i: # <<<<<<<<<<<<<<
* continue
* if m < 0 or m >= c.document_len:
*/
__pyx_t_13 = ((__pyx_v_m == __pyx_v_i) != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":672
* # word vectors in window
* if m == i:
* continue # <<<<<<<<<<<<<<
* if m < 0 or m >= c.document_len:
* c.window_indexes[n] = c.null_word_index
*/
goto __pyx_L23_continue;
/* "gensim/models/doc2vec_inner.pyx":671
* for m in range(j, k):
* # word vectors in window
* if m == i: # <<<<<<<<<<<<<<
* continue
* if m < 0 or m >= c.document_len:
*/
}
/* "gensim/models/doc2vec_inner.pyx":673
* if m == i:
* continue
* if m < 0 or m >= c.document_len: # <<<<<<<<<<<<<<
* c.window_indexes[n] = c.null_word_index
* else:
*/
__pyx_t_6 = ((__pyx_v_m < 0) != 0);
if (!__pyx_t_6) {
} else {
__pyx_t_13 = __pyx_t_6;
goto __pyx_L27_bool_binop_done;
}
__pyx_t_6 = ((__pyx_v_m >= __pyx_v_c.document_len) != 0);
__pyx_t_13 = __pyx_t_6;
__pyx_L27_bool_binop_done:;
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":674
* continue
* if m < 0 or m >= c.document_len:
* c.window_indexes[n] = c.null_word_index # <<<<<<<<<<<<<<
* else:
* c.window_indexes[n] = c.indexes[m]
*/
__pyx_t_20 = __pyx_v_c.null_word_index;
(__pyx_v_c.window_indexes[__pyx_v_n]) = __pyx_t_20;
/* "gensim/models/doc2vec_inner.pyx":673
* if m == i:
* continue
* if m < 0 or m >= c.document_len: # <<<<<<<<<<<<<<
* c.window_indexes[n] = c.null_word_index
* else:
*/
goto __pyx_L26;
}
/* "gensim/models/doc2vec_inner.pyx":676
* c.window_indexes[n] = c.null_word_index
* else:
* c.window_indexes[n] = c.indexes[m] # <<<<<<<<<<<<<<
* n += 1
* for m in range(2 * c.window):
*/
/*else*/ {
(__pyx_v_c.window_indexes[__pyx_v_n]) = (__pyx_v_c.indexes[__pyx_v_m]);
}
__pyx_L26:;
/* "gensim/models/doc2vec_inner.pyx":677
* else:
* c.window_indexes[n] = c.indexes[m]
* n += 1 # <<<<<<<<<<<<<<
* for m in range(2 * c.window):
* memcpy(&c.neu1[(c.doctag_len + m) * c.vector_size], &c.word_vectors[c.window_indexes[m] * c.vector_size],
*/
__pyx_v_n = (__pyx_v_n + 1);
__pyx_L23_continue:;
}
/* "gensim/models/doc2vec_inner.pyx":678
* c.window_indexes[n] = c.indexes[m]
* n += 1
* for m in range(2 * c.window): # <<<<<<<<<<<<<<
* memcpy(&c.neu1[(c.doctag_len + m) * c.vector_size], &c.word_vectors[c.window_indexes[m] * c.vector_size],
* c.vector_size * cython.sizeof(REAL_t))
*/
__pyx_t_4 = (2 * __pyx_v_c.window);
__pyx_t_21 = __pyx_t_4;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_21; __pyx_t_17+=1) {
__pyx_v_m = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":679
* n += 1
* for m in range(2 * c.window):
* memcpy(&c.neu1[(c.doctag_len + m) * c.vector_size], &c.word_vectors[c.window_indexes[m] * c.vector_size], # <<<<<<<<<<<<<<
* c.vector_size * cython.sizeof(REAL_t))
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
*/
(void)(memcpy((&(__pyx_v_c.neu1[((__pyx_v_c.doctag_len + __pyx_v_m) * __pyx_v_c.vector_size)])), (&(__pyx_v_c.word_vectors[((__pyx_v_c.window_indexes[__pyx_v_m]) * __pyx_v_c.vector_size)])), (__pyx_v_c.vector_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
}
/* "gensim/models/doc2vec_inner.pyx":681
* memcpy(&c.neu1[(c.doctag_len + m) * c.vector_size], &c.word_vectors[c.window_indexes[m] * c.vector_size],
* c.vector_size * cython.sizeof(REAL_t))
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error # <<<<<<<<<<<<<<
*
* if c.hs:
*/
(void)(memset(__pyx_v_c.work, 0, (__pyx_v_c.layer1_size * (sizeof(__pyx_t_6gensim_6models_14word2vec_inner_REAL_t)))));
/* "gensim/models/doc2vec_inner.pyx":683
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
*
* if c.hs: # <<<<<<<<<<<<<<
* fast_document_dmc_hs(c.points[i], c.codes[i], c.codelens[i],
* c.neu1, c.syn1, c.alpha, c.work,
*/
__pyx_t_13 = (__pyx_v_c.hs != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":684
*
* if c.hs:
* fast_document_dmc_hs(c.points[i], c.codes[i], c.codelens[i], # <<<<<<<<<<<<<<
* c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.vector_size, c.learn_hidden)
*/
__pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dmc_hs((__pyx_v_c.points[__pyx_v_i]), (__pyx_v_c.codes[__pyx_v_i]), (__pyx_v_c.codelens[__pyx_v_i]), __pyx_v_c.neu1, __pyx_v_c.syn1, __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.layer1_size, __pyx_v_c.vector_size, __pyx_v_c.learn_hidden);
/* "gensim/models/doc2vec_inner.pyx":683
* memset(c.work, 0, c.layer1_size * cython.sizeof(REAL_t)) # work to accumulate l1 error
*
* if c.hs: # <<<<<<<<<<<<<<
* fast_document_dmc_hs(c.points[i], c.codes[i], c.codelens[i],
* c.neu1, c.syn1, c.alpha, c.work,
*/
}
/* "gensim/models/doc2vec_inner.pyx":687
* c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.vector_size, c.learn_hidden)
* if c.negative: # <<<<<<<<<<<<<<
* c.next_random = fast_document_dmc_neg(c.negative, c.cum_table, c.cum_table_len, c.next_random,
* c.neu1, c.syn1neg, c.indexes[i], c.alpha, c.work,
*/
__pyx_t_13 = (__pyx_v_c.negative != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":688
* c.layer1_size, c.vector_size, c.learn_hidden)
* if c.negative:
* c.next_random = fast_document_dmc_neg(c.negative, c.cum_table, c.cum_table_len, c.next_random, # <<<<<<<<<<<<<<
* c.neu1, c.syn1neg, c.indexes[i], c.alpha, c.work,
* c.layer1_size, c.vector_size, c.learn_hidden)
*/
__pyx_v_c.next_random = __pyx_f_6gensim_6models_13doc2vec_inner_fast_document_dmc_neg(__pyx_v_c.negative, __pyx_v_c.cum_table, __pyx_v_c.cum_table_len, __pyx_v_c.next_random, __pyx_v_c.neu1, __pyx_v_c.syn1neg, (__pyx_v_c.indexes[__pyx_v_i]), __pyx_v_c.alpha, __pyx_v_c.work, __pyx_v_c.layer1_size, __pyx_v_c.vector_size, __pyx_v_c.learn_hidden);
/* "gensim/models/doc2vec_inner.pyx":687
* c.neu1, c.syn1, c.alpha, c.work,
* c.layer1_size, c.vector_size, c.learn_hidden)
* if c.negative: # <<<<<<<<<<<<<<
* c.next_random = fast_document_dmc_neg(c.negative, c.cum_table, c.cum_table_len, c.next_random,
* c.neu1, c.syn1neg, c.indexes[i], c.alpha, c.work,
*/
}
/* "gensim/models/doc2vec_inner.pyx":692
* c.layer1_size, c.vector_size, c.learn_hidden)
*
* if c.learn_doctags: # <<<<<<<<<<<<<<
* for m in range(c.doctag_len):
* our_saxpy(&c.vector_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], &c.work[m * c.vector_size],
*/
__pyx_t_13 = (__pyx_v_c.learn_doctags != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":693
*
* if c.learn_doctags:
* for m in range(c.doctag_len): # <<<<<<<<<<<<<<
* our_saxpy(&c.vector_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], &c.work[m * c.vector_size],
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size], &ONE)
*/
__pyx_t_17 = __pyx_v_c.doctag_len;
__pyx_t_18 = __pyx_t_17;
for (__pyx_t_19 = 0; __pyx_t_19 < __pyx_t_18; __pyx_t_19+=1) {
__pyx_v_m = __pyx_t_19;
/* "gensim/models/doc2vec_inner.pyx":694
* if c.learn_doctags:
* for m in range(c.doctag_len):
* our_saxpy(&c.vector_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], &c.work[m * c.vector_size], # <<<<<<<<<<<<<<
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size], &ONE)
* if c.learn_words:
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_c.vector_size), (&(__pyx_v_c.doctags_lockf[((__pyx_v_c.doctag_indexes[__pyx_v_m]) % __pyx_v_c.doctags_lockf_len)])), (&(__pyx_v_c.work[(__pyx_v_m * __pyx_v_c.vector_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_c.doctag_vectors[((__pyx_v_c.doctag_indexes[__pyx_v_m]) * __pyx_v_c.vector_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
}
/* "gensim/models/doc2vec_inner.pyx":692
* c.layer1_size, c.vector_size, c.learn_hidden)
*
* if c.learn_doctags: # <<<<<<<<<<<<<<
* for m in range(c.doctag_len):
* our_saxpy(&c.vector_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], &c.work[m * c.vector_size],
*/
}
/* "gensim/models/doc2vec_inner.pyx":696
* our_saxpy(&c.vector_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], &c.work[m * c.vector_size],
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size], &ONE)
* if c.learn_words: # <<<<<<<<<<<<<<
* for m in range(2 * c.window):
* our_saxpy(&c.vector_size, &c.words_lockf[c.window_indexes[m] % c.words_lockf_len], &c.work[(c.doctag_len + m) * c.vector_size],
*/
__pyx_t_13 = (__pyx_v_c.learn_words != 0);
if (__pyx_t_13) {
/* "gensim/models/doc2vec_inner.pyx":697
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size], &ONE)
* if c.learn_words:
* for m in range(2 * c.window): # <<<<<<<<<<<<<<
* our_saxpy(&c.vector_size, &c.words_lockf[c.window_indexes[m] % c.words_lockf_len], &c.work[(c.doctag_len + m) * c.vector_size],
* &ONE, &c.word_vectors[c.window_indexes[m] * c.vector_size], &ONE)
*/
__pyx_t_4 = (2 * __pyx_v_c.window);
__pyx_t_21 = __pyx_t_4;
for (__pyx_t_17 = 0; __pyx_t_17 < __pyx_t_21; __pyx_t_17+=1) {
__pyx_v_m = __pyx_t_17;
/* "gensim/models/doc2vec_inner.pyx":698
* if c.learn_words:
* for m in range(2 * c.window):
* our_saxpy(&c.vector_size, &c.words_lockf[c.window_indexes[m] % c.words_lockf_len], &c.work[(c.doctag_len + m) * c.vector_size], # <<<<<<<<<<<<<<
* &ONE, &c.word_vectors[c.window_indexes[m] * c.vector_size], &ONE)
*
*/
__pyx_v_6gensim_6models_14word2vec_inner_our_saxpy((&__pyx_v_c.vector_size), (&(__pyx_v_c.words_lockf[((__pyx_v_c.window_indexes[__pyx_v_m]) % __pyx_v_c.words_lockf_len)])), (&(__pyx_v_c.work[((__pyx_v_c.doctag_len + __pyx_v_m) * __pyx_v_c.vector_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE), (&(__pyx_v_c.word_vectors[((__pyx_v_c.window_indexes[__pyx_v_m]) * __pyx_v_c.vector_size)])), (&__pyx_v_6gensim_6models_13doc2vec_inner_ONE));
}
/* "gensim/models/doc2vec_inner.pyx":696
* our_saxpy(&c.vector_size, &c.doctags_lockf[c.doctag_indexes[m] % c.doctags_lockf_len], &c.work[m * c.vector_size],
* &ONE, &c.doctag_vectors[c.doctag_indexes[m] * c.vector_size], &ONE)
* if c.learn_words: # <<<<<<<<<<<<<<
* for m in range(2 * c.window):
* our_saxpy(&c.vector_size, &c.words_lockf[c.window_indexes[m] % c.words_lockf_len], &c.work[(c.doctag_len + m) * c.vector_size],
*/
}
}
}
/* "gensim/models/doc2vec_inner.pyx":658
*
* # release GIL & train on the document
* with nogil: # <<<<<<<<<<<<<<
* for i in range(c.document_len):
* j = i - c.window # negative OK: will pad with null word
*/
/*finally:*/ {
/*normal exit:*/{
#ifdef WITH_THREAD
__Pyx_FastGIL_Forget();
Py_BLOCK_THREADS
#endif
goto __pyx_L18;
}
__pyx_L18:;
}
}
/* "gensim/models/doc2vec_inner.pyx":701
* &ONE, &c.word_vectors[c.window_indexes[m] * c.vector_size], &ONE)
*
* return result # <<<<<<<<<<<<<<
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = __Pyx_PyInt_From_long(__pyx_v_result); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 701, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
/* "gensim/models/doc2vec_inner.pyx":569
*
*
* def train_document_dm_concat(model, doc_words, doctag_indexes, alpha, work=None, neu1=None, # <<<<<<<<<<<<<<
* learn_doctags=True, learn_words=True, learn_hidden=True,
* word_vectors=None, words_lockf=None, doctag_vectors=None, doctags_lockf=None):
*/
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_7);
__Pyx_XDECREF(__pyx_t_9);
__Pyx_XDECREF(__pyx_t_10);
__Pyx_XDECREF(__pyx_t_12);
__Pyx_AddTraceback("gensim.models.doc2vec_inner.train_document_dm_concat", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XDECREF(__pyx_v_vocab_codes);
__Pyx_XDECREF(__pyx_v_vocab_points);
__Pyx_XDECREF(__pyx_v_token);
__Pyx_XDECREF(__pyx_v_word_index);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":735
* ctypedef npy_cdouble complex_t
*
* cdef inline object PyArray_MultiIterNew1(a): # <<<<<<<<<<<<<<
* return PyArray_MultiIterNew(1, a)
*
*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew1(PyObject *__pyx_v_a) {
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("PyArray_MultiIterNew1", 0);
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":736
*
* cdef inline object PyArray_MultiIterNew1(a):
* return PyArray_MultiIterNew(1, a) # <<<<<<<<<<<<<<
*
* cdef inline object PyArray_MultiIterNew2(a, b):
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = PyArray_MultiIterNew(1, ((void *)__pyx_v_a)); if (unlikely(!__pyx_t_1)) __PYX_ERR(1, 736, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":735
* ctypedef npy_cdouble complex_t
*
* cdef inline object PyArray_MultiIterNew1(a): # <<<<<<<<<<<<<<
* return PyArray_MultiIterNew(1, a)
*
*/
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew1", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = 0;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":738
* return PyArray_MultiIterNew(1, a)
*
* cdef inline object PyArray_MultiIterNew2(a, b): # <<<<<<<<<<<<<<
* return PyArray_MultiIterNew(2, a, b)
*
*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew2(PyObject *__pyx_v_a, PyObject *__pyx_v_b) {
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("PyArray_MultiIterNew2", 0);
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":739
*
* cdef inline object PyArray_MultiIterNew2(a, b):
* return PyArray_MultiIterNew(2, a, b) # <<<<<<<<<<<<<<
*
* cdef inline object PyArray_MultiIterNew3(a, b, c):
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = PyArray_MultiIterNew(2, ((void *)__pyx_v_a), ((void *)__pyx_v_b)); if (unlikely(!__pyx_t_1)) __PYX_ERR(1, 739, __pyx_L1_error)
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":738
* return PyArray_MultiIterNew(1, a)
*
* cdef inline object PyArray_MultiIterNew2(a, b): # <<<<<<<<<<<<<<
* return PyArray_MultiIterNew(2, a, b)
*
*/
/* function exit code */
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew2", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = 0;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":741
* return PyArray_MultiIterNew(2, a, b)
*
* cdef inline object PyArray_MultiIterNew3(a, b, c): # <<<<<<<<<<<<<<
* return PyArray_MultiIterNew(3, a, b, c)
*
*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew3(PyObject *__pyx_v_a, PyObject *__pyx_v_b, PyObject *__pyx_v_c) {
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("PyArray_MultiIterNew3", 0);
/* "../pip-build-env-vo_sekbg/overlay/lib/python3.10/site-packages/numpy/__init__.pxd":742
*
* cdef inline object PyArray_MultiIterNew3(a, b, c):
* return PyArray_MultiIterNew(3, a, b,