35265 lines
1.4 MiB
35265 lines
1.4 MiB
/* Generated by Cython 0.20.2 on Fri Nov 28 18:58:48 2014 */
|
|
|
|
#define PY_SSIZE_T_CLEAN
|
|
#ifndef CYTHON_USE_PYLONG_INTERNALS
|
|
#ifdef PYLONG_BITS_IN_DIGIT
|
|
#define CYTHON_USE_PYLONG_INTERNALS 0
|
|
#else
|
|
#include "pyconfig.h"
|
|
#ifdef PYLONG_BITS_IN_DIGIT
|
|
#define CYTHON_USE_PYLONG_INTERNALS 1
|
|
#else
|
|
#define CYTHON_USE_PYLONG_INTERNALS 0
|
|
#endif
|
|
#endif
|
|
#endif
|
|
#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 < 0x02040000
|
|
#error Cython requires Python 2.4+.
|
|
#else
|
|
#define CYTHON_ABI "0_20_2"
|
|
#include <stddef.h> /* For offsetof */
|
|
#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
|
|
#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_CPYTHON 0
|
|
#else
|
|
#define CYTHON_COMPILING_IN_PYPY 0
|
|
#define CYTHON_COMPILING_IN_CPYTHON 1
|
|
#endif
|
|
#if CYTHON_COMPILING_IN_PYPY && PY_VERSION_HEX < 0x02070600
|
|
#define Py_OptimizeFlag 0
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
typedef int Py_ssize_t;
|
|
#define PY_SSIZE_T_MAX INT_MAX
|
|
#define PY_SSIZE_T_MIN INT_MIN
|
|
#define PY_FORMAT_SIZE_T ""
|
|
#define CYTHON_FORMAT_SSIZE_T ""
|
|
#define PyInt_FromSsize_t(z) PyInt_FromLong(z)
|
|
#define PyInt_AsSsize_t(o) __Pyx_PyInt_As_int(o)
|
|
#define PyNumber_Index(o) ((PyNumber_Check(o) && !PyFloat_Check(o)) ? PyNumber_Int(o) : \
|
|
(PyErr_Format(PyExc_TypeError, \
|
|
"expected index value, got %.200s", Py_TYPE(o)->tp_name), \
|
|
(PyObject*)0))
|
|
#define __Pyx_PyIndex_Check(o) (PyNumber_Check(o) && !PyFloat_Check(o) && \
|
|
!PyComplex_Check(o))
|
|
#define PyIndex_Check __Pyx_PyIndex_Check
|
|
#define PyErr_WarnEx(category, message, stacklevel) PyErr_Warn(category, message)
|
|
#define __PYX_BUILD_PY_SSIZE_T "i"
|
|
#else
|
|
#define __PYX_BUILD_PY_SSIZE_T "n"
|
|
#define CYTHON_FORMAT_SSIZE_T "z"
|
|
#define __Pyx_PyIndex_Check PyIndex_Check
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define Py_REFCNT(ob) (((PyObject*)(ob))->ob_refcnt)
|
|
#define Py_TYPE(ob) (((PyObject*)(ob))->ob_type)
|
|
#define Py_SIZE(ob) (((PyVarObject*)(ob))->ob_size)
|
|
#define PyVarObject_HEAD_INIT(type, size) \
|
|
PyObject_HEAD_INIT(type) size,
|
|
#define PyType_Modified(t)
|
|
typedef struct {
|
|
void *buf;
|
|
PyObject *obj;
|
|
Py_ssize_t len;
|
|
Py_ssize_t itemsize;
|
|
int readonly;
|
|
int ndim;
|
|
char *format;
|
|
Py_ssize_t *shape;
|
|
Py_ssize_t *strides;
|
|
Py_ssize_t *suboffsets;
|
|
void *internal;
|
|
} Py_buffer;
|
|
#define PyBUF_SIMPLE 0
|
|
#define PyBUF_WRITABLE 0x0001
|
|
#define PyBUF_FORMAT 0x0004
|
|
#define PyBUF_ND 0x0008
|
|
#define PyBUF_STRIDES (0x0010 | PyBUF_ND)
|
|
#define PyBUF_C_CONTIGUOUS (0x0020 | PyBUF_STRIDES)
|
|
#define PyBUF_F_CONTIGUOUS (0x0040 | PyBUF_STRIDES)
|
|
#define PyBUF_ANY_CONTIGUOUS (0x0080 | PyBUF_STRIDES)
|
|
#define PyBUF_INDIRECT (0x0100 | PyBUF_STRIDES)
|
|
#define PyBUF_RECORDS (PyBUF_STRIDES | PyBUF_FORMAT | PyBUF_WRITABLE)
|
|
#define PyBUF_FULL (PyBUF_INDIRECT | PyBUF_FORMAT | PyBUF_WRITABLE)
|
|
typedef int (*getbufferproc)(PyObject *, Py_buffer *, int);
|
|
typedef void (*releasebufferproc)(PyObject *, Py_buffer *);
|
|
#endif
|
|
#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_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 PyType_Type
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define PyUnicode_FromString(s) PyUnicode_Decode(s, strlen(s), "UTF-8", "strict")
|
|
#endif
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define Py_TPFLAGS_CHECKTYPES 0
|
|
#define Py_TPFLAGS_HAVE_INDEX 0
|
|
#endif
|
|
#if (PY_VERSION_HEX < 0x02060000) || (PY_MAJOR_VERSION >= 3)
|
|
#define Py_TPFLAGS_HAVE_NEWBUFFER 0
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define Py_TPFLAGS_HAVE_VERSION_TAG 0
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000 && !defined(Py_TPFLAGS_IS_ABSTRACT)
|
|
#define Py_TPFLAGS_IS_ABSTRACT 0
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x030400a1 && !defined(Py_TPFLAGS_HAVE_FINALIZE)
|
|
#define Py_TPFLAGS_HAVE_FINALIZE 0
|
|
#endif
|
|
#if PY_VERSION_HEX > 0x03030000 && defined(PyUnicode_KIND)
|
|
#define CYTHON_PEP393_ENABLED 1
|
|
#define __Pyx_PyUnicode_READY(op) (likely(PyUnicode_IS_READY(op)) ? \
|
|
0 : _PyUnicode_Ready((PyObject *)(op)))
|
|
#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_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)
|
|
#else
|
|
#define CYTHON_PEP393_ENABLED 0
|
|
#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_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]))
|
|
#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
|
|
#define __Pyx_PyString_FormatSafe(a, b) ((unlikely((a) == Py_None)) ? PyNumber_Remainder(a, b) : __Pyx_PyString_Format(a, b))
|
|
#define __Pyx_PyUnicode_FormatSafe(a, b) ((unlikely((a) == Py_None)) ? 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
|
|
#define PyBaseString_Type PyUnicode_Type
|
|
#define PyStringObject PyUnicodeObject
|
|
#define PyString_Type PyUnicode_Type
|
|
#define PyString_Check PyUnicode_Check
|
|
#define PyString_CheckExact PyUnicode_CheckExact
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define PyBytesObject PyStringObject
|
|
#define PyBytes_Type PyString_Type
|
|
#define PyBytes_Check PyString_Check
|
|
#define PyBytes_CheckExact PyString_CheckExact
|
|
#define PyBytes_FromString PyString_FromString
|
|
#define PyBytes_FromStringAndSize PyString_FromStringAndSize
|
|
#define PyBytes_FromFormat PyString_FromFormat
|
|
#define PyBytes_DecodeEscape PyString_DecodeEscape
|
|
#define PyBytes_AsString PyString_AsString
|
|
#define PyBytes_AsStringAndSize PyString_AsStringAndSize
|
|
#define PyBytes_Size PyString_Size
|
|
#define PyBytes_AS_STRING PyString_AS_STRING
|
|
#define PyBytes_GET_SIZE PyString_GET_SIZE
|
|
#define PyBytes_Repr PyString_Repr
|
|
#define PyBytes_Concat PyString_Concat
|
|
#define PyBytes_ConcatAndDel PyString_ConcatAndDel
|
|
#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_CheckExact(obj) || PyUnicode_CheckExact(obj) || \
|
|
PyString_Check(obj) || PyUnicode_Check(obj))
|
|
#define __Pyx_PyBaseString_CheckExact(obj) (PyString_CheckExact(obj) || PyUnicode_CheckExact(obj))
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
#define PySet_Check(obj) PyObject_TypeCheck(obj, &PySet_Type)
|
|
#define PyFrozenSet_Check(obj) PyObject_TypeCheck(obj, &PyFrozenSet_Type)
|
|
#endif
|
|
#ifndef PySet_CheckExact
|
|
#define PySet_CheckExact(obj) (Py_TYPE(obj) == &PySet_Type)
|
|
#endif
|
|
#define __Pyx_TypeCheck(obj, type) PyObject_TypeCheck(obj, (PyTypeObject *)type)
|
|
#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_VERSION_HEX < 0x030200A4
|
|
typedef long Py_hash_t;
|
|
#define __Pyx_PyInt_FromHash_t PyInt_FromLong
|
|
#define __Pyx_PyInt_AsHash_t PyInt_AsLong
|
|
#else
|
|
#define __Pyx_PyInt_FromHash_t PyInt_FromSsize_t
|
|
#define __Pyx_PyInt_AsHash_t PyInt_AsSsize_t
|
|
#endif
|
|
#if (PY_MAJOR_VERSION < 3) || (PY_VERSION_HEX >= 0x03010300)
|
|
#define __Pyx_PySequence_GetSlice(obj, a, b) PySequence_GetSlice(obj, a, b)
|
|
#define __Pyx_PySequence_SetSlice(obj, a, b, value) PySequence_SetSlice(obj, a, b, value)
|
|
#define __Pyx_PySequence_DelSlice(obj, a, b) PySequence_DelSlice(obj, a, b)
|
|
#else
|
|
#define __Pyx_PySequence_GetSlice(obj, a, b) (unlikely(!(obj)) ? \
|
|
(PyErr_SetString(PyExc_SystemError, "null argument to internal routine"), (PyObject*)0) : \
|
|
(likely((obj)->ob_type->tp_as_mapping) ? (PySequence_GetSlice(obj, a, b)) : \
|
|
(PyErr_Format(PyExc_TypeError, "'%.200s' object is unsliceable", (obj)->ob_type->tp_name), (PyObject*)0)))
|
|
#define __Pyx_PySequence_SetSlice(obj, a, b, value) (unlikely(!(obj)) ? \
|
|
(PyErr_SetString(PyExc_SystemError, "null argument to internal routine"), -1) : \
|
|
(likely((obj)->ob_type->tp_as_mapping) ? (PySequence_SetSlice(obj, a, b, value)) : \
|
|
(PyErr_Format(PyExc_TypeError, "'%.200s' object doesn't support slice assignment", (obj)->ob_type->tp_name), -1)))
|
|
#define __Pyx_PySequence_DelSlice(obj, a, b) (unlikely(!(obj)) ? \
|
|
(PyErr_SetString(PyExc_SystemError, "null argument to internal routine"), -1) : \
|
|
(likely((obj)->ob_type->tp_as_mapping) ? (PySequence_DelSlice(obj, a, b)) : \
|
|
(PyErr_Format(PyExc_TypeError, "'%.200s' object doesn't support slice deletion", (obj)->ob_type->tp_name), -1)))
|
|
#endif
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define PyMethod_New(func, self, klass) ((self) ? PyMethod_New(func, self) : PyInstanceMethod_New(func))
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
#define __Pyx_GetAttrString(o,n) PyObject_GetAttrString((o),((char *)(n)))
|
|
#define __Pyx_SetAttrString(o,n,a) PyObject_SetAttrString((o),((char *)(n)),(a))
|
|
#define __Pyx_DelAttrString(o,n) PyObject_DelAttrString((o),((char *)(n)))
|
|
#else
|
|
#define __Pyx_GetAttrString(o,n) PyObject_GetAttrString((o),(n))
|
|
#define __Pyx_SetAttrString(o,n,a) PyObject_SetAttrString((o),(n),(a))
|
|
#define __Pyx_DelAttrString(o,n) PyObject_DelAttrString((o),(n))
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
#define __Pyx_NAMESTR(n) ((char *)(n))
|
|
#define __Pyx_DOCSTR(n) ((char *)(n))
|
|
#else
|
|
#define __Pyx_NAMESTR(n) (n)
|
|
#define __Pyx_DOCSTR(n) (n)
|
|
#endif
|
|
#ifndef CYTHON_INLINE
|
|
#if defined(__GNUC__)
|
|
#define CYTHON_INLINE __inline__
|
|
#elif defined(_MSC_VER)
|
|
#define CYTHON_INLINE __inline
|
|
#elif defined (__STDC_VERSION__) && __STDC_VERSION__ >= 199901L
|
|
#define CYTHON_INLINE inline
|
|
#else
|
|
#define CYTHON_INLINE
|
|
#endif
|
|
#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
|
|
#ifdef NAN
|
|
#define __PYX_NAN() ((float) NAN)
|
|
#else
|
|
static CYTHON_INLINE float __PYX_NAN() {
|
|
/* Initialize NaN. The sign is irrelevant, an exponent with all bits 1 and
|
|
a nonzero mantissa means NaN. If the first bit in the mantissa is 1, it is
|
|
a quiet NaN. */
|
|
float value;
|
|
memset(&value, 0xFF, sizeof(value));
|
|
return value;
|
|
}
|
|
#endif
|
|
#ifdef __cplusplus
|
|
template<typename T>
|
|
void __Pyx_call_destructor(T* x) {
|
|
x->~T();
|
|
}
|
|
#endif
|
|
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#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
|
|
|
|
#ifndef __PYX_EXTERN_C
|
|
#ifdef __cplusplus
|
|
#define __PYX_EXTERN_C extern "C"
|
|
#else
|
|
#define __PYX_EXTERN_C extern
|
|
#endif
|
|
#endif
|
|
|
|
#if defined(WIN32) || defined(MS_WINDOWS)
|
|
#define _USE_MATH_DEFINES
|
|
#endif
|
|
#include <math.h>
|
|
#define __PYX_HAVE__sklearn__tree___tree
|
|
#define __PYX_HAVE_API__sklearn__tree___tree
|
|
#include "string.h"
|
|
#include "stdio.h"
|
|
#include "stdlib.h"
|
|
#include "numpy/arrayobject.h"
|
|
#include "numpy/ufuncobject.h"
|
|
#include "math.h"
|
|
#include "pythread.h"
|
|
#ifdef _OPENMP
|
|
#include <omp.h>
|
|
#endif /* _OPENMP */
|
|
|
|
#ifdef PYREX_WITHOUT_ASSERTIONS
|
|
#define CYTHON_WITHOUT_ASSERTIONS
|
|
#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
|
|
typedef struct {PyObject **p; char *s; const Py_ssize_t n; const char* encoding;
|
|
const char is_unicode; const char is_str; const char intern; } __Pyx_StringTabEntry; /*proto*/
|
|
|
|
#define __PYX_DEFAULT_STRING_ENCODING_IS_ASCII 0
|
|
#define __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT 0
|
|
#define __PYX_DEFAULT_STRING_ENCODING ""
|
|
#define __Pyx_PyObject_FromString __Pyx_PyBytes_FromString
|
|
#define __Pyx_PyObject_FromStringAndSize __Pyx_PyBytes_FromStringAndSize
|
|
#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 char* __Pyx_PyObject_AsString(PyObject*);
|
|
static CYTHON_INLINE 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_PyObject_AsSString(s) ((signed char*) __Pyx_PyObject_AsString(s))
|
|
#define __Pyx_PyObject_AsUString(s) ((unsigned char*) __Pyx_PyObject_AsString(s))
|
|
#define __Pyx_PyObject_FromUString(s) __Pyx_PyObject_FromString((const char*)s)
|
|
#define __Pyx_PyBytes_FromUString(s) __Pyx_PyBytes_FromString((const char*)s)
|
|
#define __Pyx_PyByteArray_FromUString(s) __Pyx_PyByteArray_FromString((const char*)s)
|
|
#define __Pyx_PyStr_FromUString(s) __Pyx_PyStr_FromString((const char*)s)
|
|
#define __Pyx_PyUnicode_FromUString(s) __Pyx_PyUnicode_FromString((const char*)s)
|
|
#if PY_MAJOR_VERSION < 3
|
|
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);
|
|
}
|
|
#else
|
|
#define __Pyx_Py_UNICODE_strlen Py_UNICODE_strlen
|
|
#endif
|
|
#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_Owned_Py_None(b) (Py_INCREF(Py_None), Py_None)
|
|
#define __Pyx_PyBool_FromLong(b) ((b) ? (Py_INCREF(Py_True), Py_True) : (Py_INCREF(Py_False), Py_False))
|
|
static CYTHON_INLINE int __Pyx_PyObject_IsTrue(PyObject*);
|
|
static CYTHON_INLINE PyObject* __Pyx_PyNumber_Int(PyObject* x);
|
|
static CYTHON_INLINE Py_ssize_t __Pyx_PyIndex_AsSsize_t(PyObject*);
|
|
static CYTHON_INLINE PyObject * __Pyx_PyInt_FromSize_t(size_t);
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
#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 && __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*) (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;
|
|
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));
|
|
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 PyObject *__pyx_m;
|
|
static PyObject *__pyx_d;
|
|
static PyObject *__pyx_b;
|
|
static PyObject *__pyx_empty_tuple;
|
|
static PyObject *__pyx_empty_bytes;
|
|
static int __pyx_lineno;
|
|
static int __pyx_clineno = 0;
|
|
static const char * __pyx_cfilenm= __FILE__;
|
|
static const char *__pyx_filename;
|
|
|
|
#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 <complex>
|
|
#else
|
|
#include <complex.h>
|
|
#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[] = {
|
|
"_tree.pyx",
|
|
"_tree.pxd",
|
|
"__init__.pxd",
|
|
"type.pxd",
|
|
"bool.pxd",
|
|
"complex.pxd",
|
|
"_utils.pxd",
|
|
};
|
|
#define IS_UNSIGNED(type) (((type) -1) > 0)
|
|
struct __Pyx_StructField_;
|
|
#define __PYX_BUF_FLAGS_PACKED_STRUCT (1 << 0)
|
|
typedef struct {
|
|
const char* name; /* for error messages only */
|
|
struct __Pyx_StructField_* fields;
|
|
size_t size; /* sizeof(type) */
|
|
size_t arraysize[8]; /* length of array in each dimension */
|
|
int ndim;
|
|
char typegroup; /* _R_eal, _C_omplex, Signed _I_nt, _U_nsigned int, _S_truct, _P_ointer, _O_bject, c_H_ar */
|
|
char is_unsigned;
|
|
int flags;
|
|
} __Pyx_TypeInfo;
|
|
typedef struct __Pyx_StructField_ {
|
|
__Pyx_TypeInfo* type;
|
|
const char* name;
|
|
size_t offset;
|
|
} __Pyx_StructField;
|
|
typedef struct {
|
|
__Pyx_StructField* field;
|
|
size_t parent_offset;
|
|
} __Pyx_BufFmt_StackElem;
|
|
typedef struct {
|
|
__Pyx_StructField root;
|
|
__Pyx_BufFmt_StackElem* head;
|
|
size_t fmt_offset;
|
|
size_t new_count, enc_count;
|
|
size_t struct_alignment;
|
|
int is_complex;
|
|
char enc_type;
|
|
char new_packmode;
|
|
char enc_packmode;
|
|
char is_valid_array;
|
|
} __Pyx_BufFmt_Context;
|
|
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":723
|
|
* # 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":724
|
|
*
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":725
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":726
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":730
|
|
* #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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":731
|
|
*
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":732
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":733
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":737
|
|
* #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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":738
|
|
*
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":747
|
|
* # 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":748
|
|
* # 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":749
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":751
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":752
|
|
*
|
|
* ctypedef npy_ulong uint_t
|
|
* ctypedef npy_ulonglong ulong_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_ulonglong ulonglong_t
|
|
*
|
|
*/
|
|
typedef npy_ulonglong __pyx_t_5numpy_ulong_t;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":753
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":755
|
|
* ctypedef npy_ulonglong ulonglong_t
|
|
*
|
|
* ctypedef npy_intp intp_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_uintp uintp_t
|
|
*
|
|
*/
|
|
typedef npy_intp __pyx_t_5numpy_intp_t;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":756
|
|
*
|
|
* ctypedef npy_intp intp_t
|
|
* ctypedef npy_uintp uintp_t # <<<<<<<<<<<<<<
|
|
*
|
|
* ctypedef npy_double float_t
|
|
*/
|
|
typedef npy_uintp __pyx_t_5numpy_uintp_t;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":758
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":759
|
|
*
|
|
* ctypedef npy_double float_t
|
|
* ctypedef npy_double double_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_longdouble longdouble_t
|
|
*
|
|
*/
|
|
typedef npy_double __pyx_t_5numpy_double_t;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":760
|
|
* 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;
|
|
|
|
/* "sklearn/tree/_utils.pxd":12
|
|
* cimport numpy as np
|
|
*
|
|
* ctypedef np.npy_intp SIZE_t # Type for indices and counters # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
typedef npy_intp __pyx_t_7sklearn_4tree_6_utils_SIZE_t;
|
|
|
|
/* "sklearn/tree/_tree.pxd":14
|
|
* cimport numpy as np
|
|
*
|
|
* ctypedef np.npy_float32 DTYPE_t # Type of X # <<<<<<<<<<<<<<
|
|
* ctypedef np.npy_float64 DOUBLE_t # Type of y, sample_weight
|
|
* ctypedef np.npy_intp SIZE_t # Type for indices and counters
|
|
*/
|
|
typedef npy_float32 __pyx_t_7sklearn_4tree_5_tree_DTYPE_t;
|
|
|
|
/* "sklearn/tree/_tree.pxd":15
|
|
*
|
|
* ctypedef np.npy_float32 DTYPE_t # Type of X
|
|
* ctypedef np.npy_float64 DOUBLE_t # Type of y, sample_weight # <<<<<<<<<<<<<<
|
|
* ctypedef np.npy_intp SIZE_t # Type for indices and counters
|
|
* ctypedef np.npy_int32 INT32_t # Signed 32 bit integer
|
|
*/
|
|
typedef npy_float64 __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t;
|
|
|
|
/* "sklearn/tree/_tree.pxd":16
|
|
* ctypedef np.npy_float32 DTYPE_t # Type of X
|
|
* ctypedef np.npy_float64 DOUBLE_t # Type of y, sample_weight
|
|
* ctypedef np.npy_intp SIZE_t # Type for indices and counters # <<<<<<<<<<<<<<
|
|
* ctypedef np.npy_int32 INT32_t # Signed 32 bit integer
|
|
* ctypedef np.npy_uint32 UINT32_t # Unsigned 32 bit integer
|
|
*/
|
|
typedef npy_intp __pyx_t_7sklearn_4tree_5_tree_SIZE_t;
|
|
|
|
/* "sklearn/tree/_tree.pxd":17
|
|
* ctypedef np.npy_float64 DOUBLE_t # Type of y, sample_weight
|
|
* ctypedef np.npy_intp SIZE_t # Type for indices and counters
|
|
* ctypedef np.npy_int32 INT32_t # Signed 32 bit integer # <<<<<<<<<<<<<<
|
|
* ctypedef np.npy_uint32 UINT32_t # Unsigned 32 bit integer
|
|
*
|
|
*/
|
|
typedef npy_int32 __pyx_t_7sklearn_4tree_5_tree_INT32_t;
|
|
|
|
/* "sklearn/tree/_tree.pxd":18
|
|
* ctypedef np.npy_intp SIZE_t # Type for indices and counters
|
|
* ctypedef np.npy_int32 INT32_t # Signed 32 bit integer
|
|
* ctypedef np.npy_uint32 UINT32_t # Unsigned 32 bit integer # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
typedef npy_uint32 __pyx_t_7sklearn_4tree_5_tree_UINT32_t;
|
|
#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
|
|
|
|
#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
|
|
|
|
|
|
/*--- Type declarations ---*/
|
|
struct __pyx_obj_7sklearn_4tree_6_utils_Stack;
|
|
struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Tree;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Entropy;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Gini;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_MSE;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_FriedmanMSE;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":762
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":763
|
|
*
|
|
* ctypedef npy_cfloat cfloat_t
|
|
* ctypedef npy_cdouble cdouble_t # <<<<<<<<<<<<<<
|
|
* ctypedef npy_clongdouble clongdouble_t
|
|
*
|
|
*/
|
|
typedef npy_cdouble __pyx_t_5numpy_cdouble_t;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":764
|
|
* 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;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":766
|
|
* 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_7sklearn_4tree_6_utils_StackRecord;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord;
|
|
|
|
/* "sklearn/tree/_utils.pxd":20
|
|
*
|
|
* # A record on the stack for depth-first tree growing
|
|
* cdef struct StackRecord: # <<<<<<<<<<<<<<
|
|
* SIZE_t start
|
|
* SIZE_t end
|
|
*/
|
|
struct __pyx_t_7sklearn_4tree_6_utils_StackRecord {
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t start;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t end;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t depth;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t parent;
|
|
int is_left;
|
|
double impurity;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t n_constant_features;
|
|
};
|
|
|
|
/* "sklearn/tree/_utils.pxd":46
|
|
*
|
|
* # A record on the frontier for best-first tree growing
|
|
* cdef struct PriorityHeapRecord: # <<<<<<<<<<<<<<
|
|
* SIZE_t node_id
|
|
* SIZE_t start
|
|
*/
|
|
struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord {
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t node_id;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t start;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t end;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t pos;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t depth;
|
|
int is_leaf;
|
|
double impurity;
|
|
double impurity_left;
|
|
double impurity_right;
|
|
double improvement;
|
|
};
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree_compute_feature_importances;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_build;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_build;
|
|
|
|
/* "sklearn/tree/_tree.pxd":67
|
|
* # =============================================================================
|
|
*
|
|
* cdef struct SplitRecord: # <<<<<<<<<<<<<<
|
|
* # Data to track sample split
|
|
* SIZE_t feature # Which feature to split on.
|
|
*/
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t feature;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t pos;
|
|
double threshold;
|
|
double improvement;
|
|
double impurity_left;
|
|
double impurity_right;
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pxd":145
|
|
* # =============================================================================
|
|
*
|
|
* cdef struct Node: # <<<<<<<<<<<<<<
|
|
* # Base storage structure for the nodes in a Tree object
|
|
*
|
|
*/
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t left_child;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t right_child;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t feature;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t threshold;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t impurity;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t weighted_n_node_samples;
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pxd":183
|
|
* double weighted_n_samples) nogil
|
|
* cdef void _resize(self, SIZE_t capacity) except *
|
|
* cdef int _resize_c(self, SIZE_t capacity=*) nogil # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef np.ndarray _get_value_ndarray(self)
|
|
*/
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c {
|
|
int __pyx_n;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t capacity;
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pxd":193
|
|
* cdef np.ndarray _apply_sparse_csr(self, object X)
|
|
*
|
|
* cpdef compute_feature_importances(self, normalize=*) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree_compute_feature_importances {
|
|
int __pyx_n;
|
|
PyObject *normalize;
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pxd":215
|
|
* cdef SIZE_t max_depth # Maximal tree depth
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=*)
|
|
* cdef _check_input(self, object X, np.ndarray y, np.ndarray sample_weight)
|
|
*/
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build {
|
|
int __pyx_n;
|
|
PyArrayObject *sample_weight;
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pyx":69
|
|
* cdef int IS_NOT_LEFT = 0
|
|
*
|
|
* cdef enum: # <<<<<<<<<<<<<<
|
|
* # Max value for our rand_r replacement (near the bottom).
|
|
* # We don't use RAND_MAX because it's different across platforms and
|
|
*/
|
|
enum {
|
|
__pyx_e_7sklearn_4tree_5_tree_RAND_R_MAX = 0x7FFFFFFF
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pyx":2617
|
|
* self.max_depth = max_depth
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_build {
|
|
int __pyx_n;
|
|
PyArrayObject *sample_weight;
|
|
};
|
|
|
|
/* "sklearn/tree/_tree.pyx":2771
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_build {
|
|
int __pyx_n;
|
|
PyArrayObject *sample_weight;
|
|
};
|
|
|
|
/* "sklearn/tree/_utils.pxd":29
|
|
* SIZE_t n_constant_features
|
|
*
|
|
* cdef class Stack: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t capacity
|
|
* cdef SIZE_t top
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_6_utils_Stack {
|
|
PyObject_HEAD
|
|
struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *__pyx_vtab;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t capacity;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t top;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_StackRecord *stack_;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_utils.pxd":58
|
|
* double improvement
|
|
*
|
|
* cdef class PriorityHeap: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t capacity
|
|
* cdef SIZE_t heap_ptr
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap {
|
|
PyObject_HEAD
|
|
struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap *__pyx_vtab;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t capacity;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t heap_ptr;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *heap_;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pxd":25
|
|
* # =============================================================================
|
|
*
|
|
* cdef class Criterion: # <<<<<<<<<<<<<<
|
|
* # The criterion computes the impurity of a node and the reduction of
|
|
* # impurity of a split on that node. It also computes the output statistics
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion {
|
|
PyObject_HEAD
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *__pyx_vtab;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *y;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t y_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *sample_weight;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t pos;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_node_samples;
|
|
double weighted_n_samples;
|
|
double weighted_n_node_samples;
|
|
double weighted_n_left;
|
|
double weighted_n_right;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pxd":79
|
|
*
|
|
*
|
|
* cdef class Splitter: # <<<<<<<<<<<<<<
|
|
* # The splitter searches in the input space for a feature and a threshold
|
|
* # to split the samples samples[start:end].
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter {
|
|
PyObject_HEAD
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *__pyx_vtab;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *criterion;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t min_samples_leaf;
|
|
double min_weight_leaf;
|
|
PyObject *random_state;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t rand_r_state;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_samples;
|
|
double weighted_n_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *constant_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *feature_values;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t end;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *y;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t y_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *sample_weight;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pxd":157
|
|
*
|
|
*
|
|
* cdef class Tree: # <<<<<<<<<<<<<<
|
|
* # The Tree object is a binary tree structure constructed by the
|
|
* # TreeBuilder. The tree structure is used for predictions and
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Tree {
|
|
PyObject_HEAD
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *__pyx_vtab;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t max_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t max_depth;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t node_count;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t capacity;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *nodes;
|
|
double *value;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t value_stride;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pxd":200
|
|
* # =============================================================================
|
|
*
|
|
* cdef class TreeBuilder: # <<<<<<<<<<<<<<
|
|
* # The TreeBuilder recursively builds a Tree object from training samples,
|
|
* # using a Splitter object for splitting internal nodes and assigning
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder {
|
|
PyObject_HEAD
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder *__pyx_vtab;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *splitter;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t min_samples_split;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t min_samples_leaf;
|
|
double min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t max_depth;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":150
|
|
*
|
|
*
|
|
* cdef class ClassificationCriterion(Criterion): # <<<<<<<<<<<<<<
|
|
* """Abstract criterion for classification."""
|
|
* cdef SIZE_t* n_classes
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion __pyx_base;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t label_count_stride;
|
|
double *label_count_left;
|
|
double *label_count_right;
|
|
double *label_count_total;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":368
|
|
*
|
|
*
|
|
* cdef class Entropy(ClassificationCriterion): # <<<<<<<<<<<<<<
|
|
* """Cross Entropy impurity criteria.
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Entropy {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":459
|
|
*
|
|
*
|
|
* cdef class Gini(ClassificationCriterion): # <<<<<<<<<<<<<<
|
|
* """Gini Index impurity criteria.
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Gini {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":553
|
|
*
|
|
*
|
|
* cdef class RegressionCriterion(Criterion): # <<<<<<<<<<<<<<
|
|
* """Abstract criterion for regression.
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion __pyx_base;
|
|
double *mean_left;
|
|
double *mean_right;
|
|
double *mean_total;
|
|
double *sq_sum_left;
|
|
double *sq_sum_right;
|
|
double *sq_sum_total;
|
|
double *var_left;
|
|
double *var_right;
|
|
double *sum_left;
|
|
double *sum_right;
|
|
double *sum_total;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":836
|
|
*
|
|
*
|
|
* cdef class MSE(RegressionCriterion): # <<<<<<<<<<<<<<
|
|
* """Mean squared error impurity criterion.
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_MSE {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":877
|
|
*
|
|
*
|
|
* cdef class FriedmanMSE(MSE): # <<<<<<<<<<<<<<
|
|
* """Mean squared error impurity criterion with improvement score by Friedman
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_FriedmanMSE {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_MSE __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1031
|
|
*
|
|
*
|
|
* cdef class BaseDenseSplitter(Splitter): # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* X
|
|
* cdef SIZE_t X_sample_stride
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter __pyx_base;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *X;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t X_sample_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t X_fx_stride;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1062
|
|
*
|
|
*
|
|
* cdef class BestSplitter(BaseDenseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self):
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1361
|
|
*
|
|
*
|
|
* cdef class RandomSplitter(BaseDenseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self):
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1563
|
|
*
|
|
*
|
|
* cdef class PresortBestSplitter(BaseDenseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best split, using presorting."""
|
|
* cdef DTYPE_t* X_old
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_base;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *X_old;
|
|
PyArrayObject *X_argsorted;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *X_argsorted_ptr;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t X_argsorted_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_total_samples;
|
|
unsigned char *sample_mask;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1811
|
|
*
|
|
*
|
|
* cdef class BaseSparseSplitter(Splitter): # <<<<<<<<<<<<<<
|
|
* # The sparse splitter works only with csc sparse matrix format
|
|
* cdef DTYPE_t* X_data
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter __pyx_base;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *X_data;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *X_indices;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *X_indptr;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t n_total_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *index_to_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *sorted_samples;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2130
|
|
*
|
|
*
|
|
* cdef class BestSparseSplitter(BaseSparseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best split, using the sparse data."""
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2348
|
|
*
|
|
*
|
|
* cdef class RandomSparseSplitter(BaseSparseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding a random split, using the sparse data."""
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2605
|
|
* # Depth first builder ---------------------------------------------------------
|
|
*
|
|
* cdef class DepthFirstTreeBuilder(TreeBuilder): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree in depth-first fashion."""
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder __pyx_base;
|
|
};
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2751
|
|
*
|
|
*
|
|
* cdef class BestFirstTreeBuilder(TreeBuilder): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree in best-first fashion.
|
|
*
|
|
*/
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder __pyx_base;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t max_leaf_nodes;
|
|
};
|
|
|
|
|
|
|
|
/* "sklearn/tree/_utils.pxd":29
|
|
* SIZE_t n_constant_features
|
|
*
|
|
* cdef class Stack: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t capacity
|
|
* cdef SIZE_t top
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack {
|
|
int (*is_empty)(struct __pyx_obj_7sklearn_4tree_6_utils_Stack *);
|
|
int (*push)(struct __pyx_obj_7sklearn_4tree_6_utils_Stack *, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, int, double, __pyx_t_7sklearn_4tree_6_utils_SIZE_t);
|
|
int (*pop)(struct __pyx_obj_7sklearn_4tree_6_utils_Stack *, struct __pyx_t_7sklearn_4tree_6_utils_StackRecord *);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *__pyx_vtabptr_7sklearn_4tree_6_utils_Stack;
|
|
|
|
|
|
/* "sklearn/tree/_utils.pxd":58
|
|
* double improvement
|
|
*
|
|
* cdef class PriorityHeap: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t capacity
|
|
* cdef SIZE_t heap_ptr
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap {
|
|
int (*is_empty)(struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *);
|
|
int (*push)(struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, __pyx_t_7sklearn_4tree_6_utils_SIZE_t, int, double, double, double, double);
|
|
int (*pop)(struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *, struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap *__pyx_vtabptr_7sklearn_4tree_6_utils_PriorityHeap;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":97
|
|
* # =============================================================================
|
|
*
|
|
* cdef class Criterion: # <<<<<<<<<<<<<<
|
|
* """Interface for impurity criteria."""
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion {
|
|
void (*init)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t);
|
|
void (*reset)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *);
|
|
void (*update)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t);
|
|
double (*node_impurity)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *);
|
|
void (*children_impurity)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *);
|
|
void (*node_value)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *);
|
|
double (*impurity_improvement)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *__pyx_vtabptr_7sklearn_4tree_5_tree_Criterion;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":922
|
|
*
|
|
*
|
|
* cdef class Splitter: # <<<<<<<<<<<<<<
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features,
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter {
|
|
void (*init)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, PyObject *, PyArrayObject *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *);
|
|
void (*node_reset)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double *);
|
|
void (*node_split)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *);
|
|
void (*node_value)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double *);
|
|
double (*node_impurity)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *__pyx_vtabptr_7sklearn_4tree_5_tree_Splitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2955
|
|
* # =============================================================================
|
|
*
|
|
* cdef class Tree: # <<<<<<<<<<<<<<
|
|
* """Array-based representation of a binary decision tree.
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t (*_add_node)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, int, int, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double);
|
|
void (*_resize)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t);
|
|
int (*_resize_c)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c *__pyx_optional_args);
|
|
PyArrayObject *(*_get_value_ndarray)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *);
|
|
PyArrayObject *(*_get_node_ndarray)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *);
|
|
PyArrayObject *(*predict)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, int __pyx_skip_dispatch);
|
|
PyArrayObject *(*apply)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, int __pyx_skip_dispatch);
|
|
PyArrayObject *(*_apply_dense)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *);
|
|
PyArrayObject *(*_apply_sparse_csr)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *);
|
|
PyObject *(*compute_feature_importances)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree_compute_feature_importances *__pyx_optional_args);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *__pyx_vtabptr_7sklearn_4tree_5_tree_Tree;
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_dense(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *);
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_sparse_csr(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *);
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2566
|
|
* # Tree builders
|
|
* # =============================================================================
|
|
* cdef class TreeBuilder: # <<<<<<<<<<<<<<
|
|
* """Interface for different tree building strategies. """
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder {
|
|
PyObject *(*build)(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, PyArrayObject *, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build *__pyx_optional_args);
|
|
PyObject *(*_check_input)(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, PyObject *, PyArrayObject *, PyArrayObject *);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder *__pyx_vtabptr_7sklearn_4tree_5_tree_TreeBuilder;
|
|
static PyObject *__pyx_f_7sklearn_4tree_5_tree_11TreeBuilder__check_input(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, PyObject *, PyArrayObject *, PyArrayObject *);
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":150
|
|
*
|
|
*
|
|
* cdef class ClassificationCriterion(Criterion): # <<<<<<<<<<<<<<
|
|
* """Abstract criterion for classification."""
|
|
* cdef SIZE_t* n_classes
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_vtabptr_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":368
|
|
*
|
|
*
|
|
* cdef class Entropy(ClassificationCriterion): # <<<<<<<<<<<<<<
|
|
* """Cross Entropy impurity criteria.
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Entropy {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Entropy *__pyx_vtabptr_7sklearn_4tree_5_tree_Entropy;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":459
|
|
*
|
|
*
|
|
* cdef class Gini(ClassificationCriterion): # <<<<<<<<<<<<<<
|
|
* """Gini Index impurity criteria.
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Gini {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Gini *__pyx_vtabptr_7sklearn_4tree_5_tree_Gini;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":553
|
|
*
|
|
*
|
|
* cdef class RegressionCriterion(Criterion): # <<<<<<<<<<<<<<
|
|
* """Abstract criterion for regression.
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RegressionCriterion {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_vtabptr_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":836
|
|
*
|
|
*
|
|
* cdef class MSE(RegressionCriterion): # <<<<<<<<<<<<<<
|
|
* """Mean squared error impurity criterion.
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_MSE {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RegressionCriterion __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_MSE *__pyx_vtabptr_7sklearn_4tree_5_tree_MSE;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":877
|
|
*
|
|
*
|
|
* cdef class FriedmanMSE(MSE): # <<<<<<<<<<<<<<
|
|
* """Mean squared error impurity criterion with improvement score by Friedman
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_FriedmanMSE {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_MSE __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_FriedmanMSE *__pyx_vtabptr_7sklearn_4tree_5_tree_FriedmanMSE;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1031
|
|
*
|
|
*
|
|
* cdef class BaseDenseSplitter(Splitter): # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* X
|
|
* cdef SIZE_t X_sample_stride
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseDenseSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseDenseSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1062
|
|
*
|
|
*
|
|
* cdef class BestSplitter(BaseDenseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self):
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_BestSplitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1361
|
|
*
|
|
*
|
|
* cdef class RandomSplitter(BaseDenseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self):
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RandomSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RandomSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_RandomSplitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1563
|
|
*
|
|
*
|
|
* cdef class PresortBestSplitter(BaseDenseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best split, using presorting."""
|
|
* cdef DTYPE_t* X_old
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_PresortBestSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_PresortBestSplitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":1811
|
|
*
|
|
*
|
|
* cdef class BaseSparseSplitter(Splitter): # <<<<<<<<<<<<<<
|
|
* # The sparse splitter works only with csc sparse matrix format
|
|
* cdef DTYPE_t* X_data
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseSparseSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter __pyx_base;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t (*_partition)(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t);
|
|
void (*extract_nnz)(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, int *);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter__partition(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t);
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, int *);
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2130
|
|
*
|
|
*
|
|
* cdef class BestSparseSplitter(BaseSparseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding the best split, using the sparse data."""
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestSparseSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseSparseSplitter __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestSparseSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_BestSparseSplitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2348
|
|
*
|
|
*
|
|
* cdef class RandomSparseSplitter(BaseSparseSplitter): # <<<<<<<<<<<<<<
|
|
* """Splitter for finding a random split, using the sparse data."""
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RandomSparseSplitter {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseSparseSplitter __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RandomSparseSplitter *__pyx_vtabptr_7sklearn_4tree_5_tree_RandomSparseSplitter;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2605
|
|
* # Depth first builder ---------------------------------------------------------
|
|
*
|
|
* cdef class DepthFirstTreeBuilder(TreeBuilder): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree in depth-first fashion."""
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_DepthFirstTreeBuilder {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder __pyx_base;
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *__pyx_vtabptr_7sklearn_4tree_5_tree_DepthFirstTreeBuilder;
|
|
|
|
|
|
/* "sklearn/tree/_tree.pyx":2751
|
|
*
|
|
*
|
|
* cdef class BestFirstTreeBuilder(TreeBuilder): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree in best-first fashion.
|
|
*
|
|
*/
|
|
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestFirstTreeBuilder {
|
|
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder __pyx_base;
|
|
int (*_add_split_node)(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double, int, int, struct __pyx_t_7sklearn_4tree_5_tree_Node *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *);
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_vtabptr_7sklearn_4tree_5_tree_BestFirstTreeBuilder;
|
|
static CYTHON_INLINE int __pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder__add_split_node(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double, int, int, struct __pyx_t_7sklearn_4tree_5_tree_Node *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *);
|
|
#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); /*proto*/
|
|
#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 /* CYTHON_REFNANNY */
|
|
#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)
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
static CYTHON_INLINE PyObject* __Pyx_PyObject_GetAttrStr(PyObject* obj, PyObject* attr_name) {
|
|
PyTypeObject* tp = Py_TYPE(obj);
|
|
if (likely(tp->tp_getattro))
|
|
return tp->tp_getattro(obj, attr_name);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(tp->tp_getattr))
|
|
return tp->tp_getattr(obj, PyString_AS_STRING(attr_name));
|
|
#endif
|
|
return PyObject_GetAttr(obj, attr_name);
|
|
}
|
|
#else
|
|
#define __Pyx_PyObject_GetAttrStr(o,n) PyObject_GetAttr(o,n)
|
|
#endif
|
|
|
|
static PyObject *__Pyx_GetBuiltinName(PyObject *name); /*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); /*proto*/
|
|
|
|
static void __Pyx_RaiseDoubleKeywordsError(const char* func_name, PyObject* kw_name); /*proto*/
|
|
|
|
static int __Pyx_ParseOptionalKeywords(PyObject *kwds, PyObject **argnames[], \
|
|
PyObject *kwds2, PyObject *values[], Py_ssize_t num_pos_args, \
|
|
const char* function_name); /*proto*/
|
|
|
|
static CYTHON_INLINE int __Pyx_ArgTypeTest(PyObject *obj, PyTypeObject *type, int none_allowed,
|
|
const char *name, int exact); /*proto*/
|
|
|
|
static CYTHON_INLINE int __Pyx_GetBufferAndValidate(Py_buffer* buf, PyObject* obj,
|
|
__Pyx_TypeInfo* dtype, int flags, int nd, int cast, __Pyx_BufFmt_StackElem* stack);
|
|
static CYTHON_INLINE void __Pyx_SafeReleaseBuffer(Py_buffer* info);
|
|
|
|
#define __Pyx_BufPtrStrided1d(type, buf, i0, s0) (type)((char*)buf + i0 * s0)
|
|
static CYTHON_INLINE void __Pyx_ErrRestore(PyObject *type, PyObject *value, PyObject *tb); /*proto*/
|
|
static CYTHON_INLINE void __Pyx_ErrFetch(PyObject **type, PyObject **value, PyObject **tb); /*proto*/
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
static CYTHON_INLINE PyObject* __Pyx_PyObject_Call(PyObject *func, PyObject *arg, PyObject *kw); /*proto*/
|
|
#else
|
|
#define __Pyx_PyObject_Call(func, arg, kw) PyObject_Call(func, arg, kw)
|
|
#endif
|
|
|
|
#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 CYTHON_INLINE 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);
|
|
|
|
static void __Pyx_WriteUnraisable(const char *name, int clineno,
|
|
int lineno, const char *filename,
|
|
int full_traceback); /*proto*/
|
|
|
|
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type); /*proto*/
|
|
|
|
static CYTHON_INLINE PyObject *__Pyx_GetModuleGlobalName(PyObject *name); /*proto*/
|
|
|
|
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause); /*proto*/
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
#define __Pyx_PyObject_DelAttrStr(o,n) __Pyx_PyObject_SetAttrStr(o,n,NULL)
|
|
static CYTHON_INLINE int __Pyx_PyObject_SetAttrStr(PyObject* obj, PyObject* attr_name, PyObject* value) {
|
|
PyTypeObject* tp = Py_TYPE(obj);
|
|
if (likely(tp->tp_setattro))
|
|
return tp->tp_setattro(obj, attr_name, value);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(tp->tp_setattr))
|
|
return tp->tp_setattr(obj, PyString_AS_STRING(attr_name), value);
|
|
#endif
|
|
return PyObject_SetAttr(obj, attr_name, value);
|
|
}
|
|
#else
|
|
#define __Pyx_PyObject_DelAttrStr(o,n) PyObject_DelAttr(o,n)
|
|
#define __Pyx_PyObject_SetAttrStr(o,n,v) PyObject_SetAttr(o,n,v)
|
|
#endif
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseTooManyValuesError(Py_ssize_t expected);
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNeedMoreValuesError(Py_ssize_t index);
|
|
|
|
static CYTHON_INLINE int __Pyx_IterFinish(void); /*proto*/
|
|
|
|
static int __Pyx_IternextUnpackEndCheck(PyObject *retval, Py_ssize_t expected); /*proto*/
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyObject_GetSlice(
|
|
PyObject* obj, Py_ssize_t cstart, Py_ssize_t cstop,
|
|
PyObject** py_start, PyObject** py_stop, PyObject** py_slice,
|
|
int has_cstart, int has_cstop, int wraparound);
|
|
|
|
static CYTHON_INLINE int __Pyx_PySequence_Contains(PyObject* item, PyObject* seq, int eq) {
|
|
int result = PySequence_Contains(seq, item);
|
|
return unlikely(result < 0) ? result : (result == (eq == Py_EQ));
|
|
}
|
|
|
|
static void __Pyx_RaiseBufferFallbackError(void); /*proto*/
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNoneNotIterableError(void);
|
|
|
|
static int __Pyx_SetVtable(PyObject *dict, void *vtable); /*proto*/
|
|
|
|
static void* __Pyx_GetVtable(PyObject *dict); /*proto*/
|
|
|
|
static PyObject* __Pyx_ImportFrom(PyObject* module, PyObject* name); /*proto*/
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_Py_intptr_t(Py_intptr_t value);
|
|
|
|
static CYTHON_INLINE Py_intptr_t __Pyx_PyInt_As_Py_intptr_t(PyObject *);
|
|
|
|
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, int level); /*proto*/
|
|
|
|
typedef struct {
|
|
Py_ssize_t shape, strides, suboffsets;
|
|
} __Pyx_Buf_DimInfo;
|
|
typedef struct {
|
|
size_t refcount;
|
|
Py_buffer pybuffer;
|
|
} __Pyx_Buffer;
|
|
typedef struct {
|
|
__Pyx_Buffer *rcbuffer;
|
|
char *data;
|
|
__Pyx_Buf_DimInfo diminfo[8];
|
|
} __Pyx_LocalBuf_ND;
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
static int __Pyx_GetBuffer(PyObject *obj, Py_buffer *view, int flags);
|
|
static void __Pyx_ReleaseBuffer(Py_buffer *view);
|
|
#else
|
|
#define __Pyx_GetBuffer PyObject_GetBuffer
|
|
#define __Pyx_ReleaseBuffer PyBuffer_Release
|
|
#endif
|
|
|
|
|
|
static Py_ssize_t __Pyx_zeros[] = {0, 0, 0, 0, 0, 0, 0, 0};
|
|
static Py_ssize_t __Pyx_minusones[] = {-1, -1, -1, -1, -1, -1, -1, -1};
|
|
|
|
static CYTHON_INLINE int __Pyx_PyInt_As_int(PyObject *);
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_int(int value);
|
|
|
|
static CYTHON_INLINE npy_uint32 __Pyx_PyInt_As_npy_uint32(PyObject *);
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_long(long value);
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_npy_int32(npy_int32 value);
|
|
|
|
static CYTHON_INLINE npy_int32 __Pyx_PyInt_As_npy_int32(PyObject *);
|
|
|
|
static CYTHON_INLINE long __Pyx_pow_long(long, long); /* proto */
|
|
|
|
#ifndef __PYX_FORCE_INIT_THREADS
|
|
#define __PYX_FORCE_INIT_THREADS 0
|
|
#endif
|
|
|
|
#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(_WIN32) || defined(__clang__)) && defined(__cplusplus) && CYTHON_CCOMPLEX
|
|
#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
|
|
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float, float);
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#define __Pyx_c_eqf(a, b) ((a)==(b))
|
|
#define __Pyx_c_sumf(a, b) ((a)+(b))
|
|
#define __Pyx_c_difff(a, b) ((a)-(b))
|
|
#define __Pyx_c_prodf(a, b) ((a)*(b))
|
|
#define __Pyx_c_quotf(a, b) ((a)/(b))
|
|
#define __Pyx_c_negf(a) (-(a))
|
|
#ifdef __cplusplus
|
|
#define __Pyx_c_is_zerof(z) ((z)==(float)0)
|
|
#define __Pyx_c_conjf(z) (::std::conj(z))
|
|
#if 1
|
|
#define __Pyx_c_absf(z) (::std::abs(z))
|
|
#define __Pyx_c_powf(a, b) (::std::pow(a, b))
|
|
#endif
|
|
#else
|
|
#define __Pyx_c_is_zerof(z) ((z)==0)
|
|
#define __Pyx_c_conjf(z) (conjf(z))
|
|
#if 1
|
|
#define __Pyx_c_absf(z) (cabsf(z))
|
|
#define __Pyx_c_powf(a, b) (cpowf(a, b))
|
|
#endif
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eqf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_sumf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_difff(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_prodf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_quotf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_negf(__pyx_t_float_complex);
|
|
static CYTHON_INLINE int __Pyx_c_is_zerof(__pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_conjf(__pyx_t_float_complex);
|
|
#if 1
|
|
static CYTHON_INLINE float __Pyx_c_absf(__pyx_t_float_complex);
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_powf(__pyx_t_float_complex, __pyx_t_float_complex);
|
|
#endif
|
|
#endif
|
|
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double, double);
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#define __Pyx_c_eq(a, b) ((a)==(b))
|
|
#define __Pyx_c_sum(a, b) ((a)+(b))
|
|
#define __Pyx_c_diff(a, b) ((a)-(b))
|
|
#define __Pyx_c_prod(a, b) ((a)*(b))
|
|
#define __Pyx_c_quot(a, b) ((a)/(b))
|
|
#define __Pyx_c_neg(a) (-(a))
|
|
#ifdef __cplusplus
|
|
#define __Pyx_c_is_zero(z) ((z)==(double)0)
|
|
#define __Pyx_c_conj(z) (::std::conj(z))
|
|
#if 1
|
|
#define __Pyx_c_abs(z) (::std::abs(z))
|
|
#define __Pyx_c_pow(a, b) (::std::pow(a, b))
|
|
#endif
|
|
#else
|
|
#define __Pyx_c_is_zero(z) ((z)==0)
|
|
#define __Pyx_c_conj(z) (conj(z))
|
|
#if 1
|
|
#define __Pyx_c_abs(z) (cabs(z))
|
|
#define __Pyx_c_pow(a, b) (cpow(a, b))
|
|
#endif
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eq(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_sum(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_diff(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_prod(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_quot(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_neg(__pyx_t_double_complex);
|
|
static CYTHON_INLINE int __Pyx_c_is_zero(__pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_conj(__pyx_t_double_complex);
|
|
#if 1
|
|
static CYTHON_INLINE double __Pyx_c_abs(__pyx_t_double_complex);
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_pow(__pyx_t_double_complex, __pyx_t_double_complex);
|
|
#endif
|
|
#endif
|
|
|
|
static CYTHON_INLINE long __Pyx_PyInt_As_long(PyObject *);
|
|
|
|
static int __Pyx_check_binary_version(void);
|
|
|
|
#if !defined(__Pyx_PyIdentifier_FromString)
|
|
#if PY_MAJOR_VERSION < 3
|
|
#define __Pyx_PyIdentifier_FromString(s) PyString_FromString(s)
|
|
#else
|
|
#define __Pyx_PyIdentifier_FromString(s) PyUnicode_FromString(s)
|
|
#endif
|
|
#endif
|
|
|
|
static PyObject *__Pyx_ImportModule(const char *name); /*proto*/
|
|
|
|
static PyTypeObject *__Pyx_ImportType(const char *module_name, const char *class_name, size_t size, int strict); /*proto*/
|
|
|
|
typedef struct {
|
|
int code_line;
|
|
PyCodeObject* code_object;
|
|
} __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);
|
|
|
|
static void __Pyx_AddTraceback(const char *funcname, int c_line,
|
|
int py_line, const char *filename); /*proto*/
|
|
|
|
static int __Pyx_InitStrings(__Pyx_StringTabEntry *t); /*proto*/
|
|
|
|
|
|
/* Module declarations from 'cpython.buffer' */
|
|
|
|
/* Module declarations from 'cpython.ref' */
|
|
|
|
/* Module declarations from 'libc.string' */
|
|
|
|
/* Module declarations from 'libc.stdio' */
|
|
|
|
/* Module declarations from 'cpython.object' */
|
|
|
|
/* Module declarations from '__builtin__' */
|
|
|
|
/* Module declarations from 'cpython.type' */
|
|
static PyTypeObject *__pyx_ptype_7cpython_4type_type = 0;
|
|
|
|
/* Module declarations from 'libc.stdlib' */
|
|
|
|
/* 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_ufunc = 0;
|
|
static CYTHON_INLINE char *__pyx_f_5numpy__util_dtypestring(PyArray_Descr *, char *, char *, int *); /*proto*/
|
|
|
|
/* Module declarations from 'libc.math' */
|
|
|
|
/* Module declarations from 'cpython.version' */
|
|
|
|
/* Module declarations from 'cpython.exc' */
|
|
|
|
/* Module declarations from 'cpython.module' */
|
|
|
|
/* Module declarations from 'cpython.mem' */
|
|
|
|
/* Module declarations from 'cpython.tuple' */
|
|
|
|
/* Module declarations from 'cpython.list' */
|
|
|
|
/* Module declarations from 'cpython.sequence' */
|
|
|
|
/* Module declarations from 'cpython.mapping' */
|
|
|
|
/* Module declarations from 'cpython.iterator' */
|
|
|
|
/* Module declarations from 'cpython.number' */
|
|
|
|
/* Module declarations from 'cpython.int' */
|
|
|
|
/* Module declarations from '__builtin__' */
|
|
|
|
/* Module declarations from 'cpython.bool' */
|
|
static PyTypeObject *__pyx_ptype_7cpython_4bool_bool = 0;
|
|
|
|
/* Module declarations from 'cpython.long' */
|
|
|
|
/* Module declarations from 'cpython.float' */
|
|
|
|
/* Module declarations from '__builtin__' */
|
|
|
|
/* Module declarations from 'cpython.complex' */
|
|
static PyTypeObject *__pyx_ptype_7cpython_7complex_complex = 0;
|
|
|
|
/* Module declarations from 'cpython.string' */
|
|
|
|
/* Module declarations from 'cpython.unicode' */
|
|
|
|
/* Module declarations from 'cpython.dict' */
|
|
|
|
/* Module declarations from 'cpython.instance' */
|
|
|
|
/* Module declarations from 'cpython.function' */
|
|
|
|
/* Module declarations from 'cpython.method' */
|
|
|
|
/* Module declarations from 'cpython.weakref' */
|
|
|
|
/* Module declarations from 'cpython.getargs' */
|
|
|
|
/* Module declarations from 'cpython.pythread' */
|
|
|
|
/* Module declarations from 'cpython.pystate' */
|
|
|
|
/* Module declarations from 'cpython.cobject' */
|
|
|
|
/* Module declarations from 'cpython.oldbuffer' */
|
|
|
|
/* Module declarations from 'cpython.set' */
|
|
|
|
/* Module declarations from 'cpython.bytes' */
|
|
|
|
/* Module declarations from 'cpython.pycapsule' */
|
|
|
|
/* Module declarations from 'cpython' */
|
|
|
|
/* Module declarations from 'sklearn.tree._utils' */
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_6_utils_Stack = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_6_utils_PriorityHeap = 0;
|
|
|
|
/* Module declarations from 'sklearn.tree._tree' */
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Criterion = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Splitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Tree = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_TreeBuilder = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Entropy = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Gini = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_MSE = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_FriedmanMSE = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_BaseDenseSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_BestSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_RandomSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_PresortBestSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_BaseSparseSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_BestSparseSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_RandomSparseSplitter = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_DepthFirstTreeBuilder = 0;
|
|
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_BestFirstTreeBuilder = 0;
|
|
static double __pyx_v_7sklearn_4tree_5_tree_INFINITY;
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF;
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED;
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_7sklearn_4tree_5_tree_INITIAL_STACK_SIZE;
|
|
static __pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_7sklearn_4tree_5_tree_MIN_IMPURITY_SPLIT;
|
|
static __pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD;
|
|
static __pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_7sklearn_4tree_5_tree_EXTRACT_NNZ_SWITCH;
|
|
static int __pyx_v_7sklearn_4tree_5_tree_IS_FIRST;
|
|
static int __pyx_v_7sklearn_4tree_5_tree_IS_NOT_FIRST;
|
|
static int __pyx_v_7sklearn_4tree_5_tree_IS_LEFT;
|
|
static int __pyx_v_7sklearn_4tree_5_tree_IS_NOT_LEFT;
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree__init_split(struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_sort(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_swap(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_f_7sklearn_4tree_5_tree_median3(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static void __pyx_f_7sklearn_4tree_5_tree_introsort(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, int); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_sift_down(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static void __pyx_f_7sklearn_4tree_5_tree_heapsort(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static int __pyx_f_7sklearn_4tree_5_tree_compare_SIZE_t(void const *, void const *); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_binary_search(__pyx_t_7sklearn_4tree_5_tree_INT32_t *, __pyx_t_7sklearn_4tree_5_tree_INT32_t, __pyx_t_7sklearn_4tree_5_tree_INT32_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_INT32_t *); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_extract_nnz_index_to_samples(__pyx_t_7sklearn_4tree_5_tree_INT32_t *, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_INT32_t, __pyx_t_7sklearn_4tree_5_tree_INT32_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_extract_nnz_binary_search(__pyx_t_7sklearn_4tree_5_tree_INT32_t *, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_INT32_t, __pyx_t_7sklearn_4tree_5_tree_INT32_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, int *); /*proto*/
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static CYTHON_INLINE int __pyx_f_7sklearn_4tree_5_tree__add_to_frontier(struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *, struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *); /*proto*/
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_UINT32_t __pyx_f_7sklearn_4tree_5_tree_our_rand_r(__pyx_t_7sklearn_4tree_5_tree_UINT32_t *); /*proto*/
|
|
static CYTHON_INLINE PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_sizet_ptr_to_ndarray(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t); /*proto*/
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_UINT32_t *); /*proto*/
|
|
static CYTHON_INLINE double __pyx_f_7sklearn_4tree_5_tree_rand_uniform(double, double, __pyx_t_7sklearn_4tree_5_tree_UINT32_t *); /*proto*/
|
|
static CYTHON_INLINE double __pyx_f_7sklearn_4tree_5_tree_log(double); /*proto*/
|
|
static __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_fuse_0__pyx_f_7sklearn_4tree_5_tree_safe_realloc(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t **, size_t); /*proto*/
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc(__pyx_t_7sklearn_4tree_5_tree_SIZE_t **, size_t); /*proto*/
|
|
static unsigned char *__pyx_fuse_2__pyx_f_7sklearn_4tree_5_tree_safe_realloc(unsigned char **, size_t); /*proto*/
|
|
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_SIZE_t = { "SIZE_t", NULL, sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t), { 0 }, 0, IS_UNSIGNED(__pyx_t_7sklearn_4tree_5_tree_SIZE_t) ? 'U' : 'I', IS_UNSIGNED(__pyx_t_7sklearn_4tree_5_tree_SIZE_t), 0 };
|
|
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DOUBLE_t = { "DOUBLE_t", NULL, sizeof(__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t), { 0 }, 0, 'R', 0, 0 };
|
|
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t = { "DTYPE_t", NULL, sizeof(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t), { 0 }, 0, 'R', 0, 0 };
|
|
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_INT32_t = { "INT32_t", NULL, sizeof(__pyx_t_7sklearn_4tree_5_tree_INT32_t), { 0 }, 0, IS_UNSIGNED(__pyx_t_7sklearn_4tree_5_tree_INT32_t) ? 'U' : 'I', IS_UNSIGNED(__pyx_t_7sklearn_4tree_5_tree_INT32_t), 0 };
|
|
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t = { "float64_t", NULL, sizeof(__pyx_t_5numpy_float64_t), { 0 }, 0, 'R', 0, 0 };
|
|
#define __Pyx_MODULE_NAME "sklearn.tree._tree"
|
|
int __pyx_module_is_main_sklearn__tree___tree = 0;
|
|
|
|
/* Implementation of 'sklearn.tree._tree' */
|
|
static PyObject *__pyx_builtin_range;
|
|
static PyObject *__pyx_builtin_MemoryError;
|
|
static PyObject *__pyx_builtin_ValueError;
|
|
static PyObject *__pyx_builtin_RuntimeError;
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs, PyArrayObject *__pyx_v_n_classes); /* proto */
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_6__getstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_8__setstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v_d); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs); /* proto */
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_6__getstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_8__setstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v_d); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, double __pyx_v_min_weight_leaf, PyObject *__pyx_v_random_state); /* proto */
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_8Splitter_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_4__getstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_6__setstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v_d); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion_4__del__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_12max_features___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_12max_features_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_17BaseDenseSplitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, CYTHON_UNUSED double __pyx_v_min_weight_leaf, CYTHON_UNUSED PyObject *__pyx_v_random_state); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_12BestSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_14RandomSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, CYTHON_UNUSED double __pyx_v_min_weight_leaf, CYTHON_UNUSED PyObject *__pyx_v_random_state); /* proto */
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_18BaseSparseSplitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, CYTHON_UNUSED double __pyx_v_min_weight_leaf, CYTHON_UNUSED PyObject *__pyx_v_random_state); /* proto */
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_18BaseSparseSplitter_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_18BestSparseSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_20RandomSparseSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_11TreeBuilder_build(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, double __pyx_v_min_weight_leaf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_2build(struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_20BestFirstTreeBuilder___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, PyObject *__pyx_v_min_weight_leaf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_leaf_nodes); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_2build(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_classes___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_13children_left___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_14children_right___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_7feature___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9threshold___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_8impurity___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_14n_node_samples___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_23weighted_n_node_samples___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_5value___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, int __pyx_v_n_features, PyArrayObject *__pyx_v_n_classes, int __pyx_v_n_outputs); /* proto */
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_4Tree_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_6__getstate__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_8__setstate__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_d); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_10predict(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_12apply(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_14compute_feature_importances(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_normalize); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_10n_features___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_10n_features_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_outputs___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_outputs_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_13max_n_classes___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_13max_n_classes_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9max_depth___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_9max_depth_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_10node_count___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_10node_count_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_8capacity___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self); /* proto */
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_8capacity_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree__realloc_test(CYTHON_UNUSED PyObject *__pyx_self); /* proto */
|
|
static int __pyx_pf_5numpy_7ndarray___getbuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags); /* proto */
|
|
static void __pyx_pf_5numpy_7ndarray_2__releasebuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info); /* proto */
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Criterion(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Splitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Tree(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_TreeBuilder(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_ClassificationCriterion(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Entropy(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Gini(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_RegressionCriterion(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_MSE(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_FriedmanMSE(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BaseDenseSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BestSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_RandomSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_PresortBestSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BaseSparseSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BestSparseSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_RandomSparseSplitter(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_DepthFirstTreeBuilder(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BestFirstTreeBuilder(PyTypeObject *t, PyObject *a, PyObject *k); /*proto*/
|
|
static char __pyx_k_B[] = "B";
|
|
static char __pyx_k_C[] = "C";
|
|
static char __pyx_k_H[] = "H";
|
|
static char __pyx_k_I[] = "I";
|
|
static char __pyx_k_L[] = "L";
|
|
static char __pyx_k_O[] = "O";
|
|
static char __pyx_k_Q[] = "Q";
|
|
static char __pyx_k_X[] = "X";
|
|
static char __pyx_k_b[] = "b";
|
|
static char __pyx_k_d[] = "d";
|
|
static char __pyx_k_f[] = "f";
|
|
static char __pyx_k_g[] = "g";
|
|
static char __pyx_k_h[] = "h";
|
|
static char __pyx_k_i[] = "i";
|
|
static char __pyx_k_l[] = "l";
|
|
static char __pyx_k_p[] = "p";
|
|
static char __pyx_k_q[] = "q";
|
|
static char __pyx_k_y[] = "y";
|
|
static char __pyx_k_Zd[] = "Zd";
|
|
static char __pyx_k_Zf[] = "Zf";
|
|
static char __pyx_k_Zg[] = "Zg";
|
|
static char __pyx_k_np[] = "np";
|
|
static char __pyx_k_inf[] = "inf";
|
|
static char __pyx_k_max[] = "max";
|
|
static char __pyx_k_sum[] = "sum";
|
|
static char __pyx_k_axis[] = "axis";
|
|
static char __pyx_k_clip[] = "clip";
|
|
static char __pyx_k_copy[] = "copy";
|
|
static char __pyx_k_data[] = "data";
|
|
static char __pyx_k_intp[] = "intp";
|
|
static char __pyx_k_main[] = "__main__";
|
|
static char __pyx_k_mode[] = "mode";
|
|
static char __pyx_k_ndim[] = "ndim";
|
|
static char __pyx_k_take[] = "take";
|
|
static char __pyx_k_test[] = "__test__";
|
|
static char __pyx_k_tree[] = "tree";
|
|
static char __pyx_k_DTYPE[] = "DTYPE";
|
|
static char __pyx_k_apply[] = "apply";
|
|
static char __pyx_k_build[] = "build";
|
|
static char __pyx_k_dtype[] = "dtype";
|
|
static char __pyx_k_flags[] = "flags";
|
|
static char __pyx_k_int32[] = "int32";
|
|
static char __pyx_k_names[] = "names";
|
|
static char __pyx_k_nodes[] = "nodes";
|
|
static char __pyx_k_numpy[] = "numpy";
|
|
static char __pyx_k_order[] = "order";
|
|
static char __pyx_k_range[] = "range";
|
|
static char __pyx_k_shape[] = "shape";
|
|
static char __pyx_k_tocsc[] = "tocsc";
|
|
static char __pyx_k_zeros[] = "zeros";
|
|
static char __pyx_k_DOUBLE[] = "DOUBLE";
|
|
static char __pyx_k_import[] = "__import__";
|
|
static char __pyx_k_indptr[] = "indptr";
|
|
static char __pyx_k_values[] = "values";
|
|
static char __pyx_k_argsort[] = "argsort";
|
|
static char __pyx_k_asarray[] = "asarray";
|
|
static char __pyx_k_feature[] = "feature";
|
|
static char __pyx_k_float32[] = "float32";
|
|
static char __pyx_k_float64[] = "float64";
|
|
static char __pyx_k_formats[] = "formats";
|
|
static char __pyx_k_indices[] = "indices";
|
|
static char __pyx_k_offsets[] = "offsets";
|
|
static char __pyx_k_predict[] = "predict";
|
|
static char __pyx_k_randint[] = "randint";
|
|
static char __pyx_k_reshape[] = "reshape";
|
|
static char __pyx_k_strides[] = "strides";
|
|
static char __pyx_k_getstate[] = "__getstate__";
|
|
static char __pyx_k_impurity[] = "impurity";
|
|
static char __pyx_k_issparse[] = "issparse";
|
|
static char __pyx_k_itemsize[] = "itemsize";
|
|
static char __pyx_k_splitter[] = "splitter";
|
|
static char __pyx_k_TREE_LEAF[] = "TREE_LEAF";
|
|
static char __pyx_k_criterion[] = "criterion";
|
|
static char __pyx_k_max_depth[] = "max_depth";
|
|
static char __pyx_k_n_classes[] = "n_classes";
|
|
static char __pyx_k_n_outputs[] = "n_outputs";
|
|
static char __pyx_k_normalize[] = "normalize";
|
|
static char __pyx_k_threshold[] = "threshold";
|
|
static char __pyx_k_NODE_DTYPE[] = "NODE_DTYPE";
|
|
static char __pyx_k_ValueError[] = "ValueError";
|
|
static char __pyx_k_contiguous[] = "contiguous";
|
|
static char __pyx_k_csc_matrix[] = "csc_matrix";
|
|
static char __pyx_k_csr_matrix[] = "csr_matrix";
|
|
static char __pyx_k_left_child[] = "left_child";
|
|
static char __pyx_k_n_features[] = "n_features";
|
|
static char __pyx_k_node_count[] = "node_count";
|
|
static char __pyx_k_pyx_vtable[] = "__pyx_vtable__";
|
|
static char __pyx_k_MemoryError[] = "MemoryError";
|
|
static char __pyx_k_right_child[] = "right_child";
|
|
static char __pyx_k_RuntimeError[] = "RuntimeError";
|
|
static char __pyx_k_c_contiguous[] = "c_contiguous";
|
|
static char __pyx_k_max_features[] = "max_features";
|
|
static char __pyx_k_random_state[] = "random_state";
|
|
static char __pyx_k_realloc_test[] = "_realloc_test";
|
|
static char __pyx_k_scipy_sparse[] = "scipy.sparse";
|
|
static char __pyx_k_sort_indices[] = "sort_indices";
|
|
static char __pyx_k_pyx_getbuffer[] = "__pyx_getbuffer";
|
|
static char __pyx_k_sample_weight[] = "sample_weight";
|
|
static char __pyx_k_TREE_UNDEFINED[] = "TREE_UNDEFINED";
|
|
static char __pyx_k_asfortranarray[] = "asfortranarray";
|
|
static char __pyx_k_max_leaf_nodes[] = "max_leaf_nodes";
|
|
static char __pyx_k_n_node_samples[] = "n_node_samples";
|
|
static char __pyx_k_min_weight_leaf[] = "min_weight_leaf";
|
|
static char __pyx_k_min_samples_leaf[] = "min_samples_leaf";
|
|
static char __pyx_k_ascontiguousarray[] = "ascontiguousarray";
|
|
static char __pyx_k_min_samples_split[] = "min_samples_split";
|
|
static char __pyx_k_pyx_releasebuffer[] = "__pyx_releasebuffer";
|
|
static char __pyx_k_resizing_tree_to_d[] = "resizing tree to %d";
|
|
static char __pyx_k_sklearn_tree__tree[] = "sklearn.tree._tree";
|
|
static char __pyx_k_weighted_n_node_samples[] = "weighted_n_node_samples";
|
|
static char __pyx_k_X_should_be_in_csc_format[] = "X should be in csc format";
|
|
static char __pyx_k_could_not_allocate_d_bytes[] = "could not allocate %d bytes";
|
|
static char __pyx_k_compute_feature_importances[] = "compute_feature_importances";
|
|
static char __pyx_k_ndarray_is_not_C_contiguous[] = "ndarray is not C contiguous";
|
|
static char __pyx_k_could_not_allocate_d_d_bytes[] = "could not allocate (%d * %d) bytes";
|
|
static char __pyx_k_Users_ajoly_git_scikit_learn_sk[] = "/Users/ajoly/git/scikit-learn/sklearn/tree/_tree.pyx";
|
|
static char __pyx_k_unknown_dtype_code_in_numpy_pxd[] = "unknown dtype code in numpy.pxd (%d)";
|
|
static char __pyx_k_Did_not_recognise_loaded_array_l[] = "Did not recognise loaded array layout";
|
|
static char __pyx_k_Format_string_allocated_too_shor[] = "Format string allocated too short, see comment in numpy.pxd";
|
|
static char __pyx_k_No_support_for_np_int64_index_ba[] = "No support for np.int64 index based sparse matrices";
|
|
static char __pyx_k_Non_native_byte_order_not_suppor[] = "Non-native byte order not supported";
|
|
static char __pyx_k_X_dtype_should_be_np_float32_got[] = "X.dtype should be np.float32, got %s";
|
|
static char __pyx_k_X_should_be_in_csr_matrix_format[] = "X should be in csr_matrix format, got %s";
|
|
static char __pyx_k_X_should_be_in_np_ndarray_format[] = "X should be in np.ndarray format, got %s";
|
|
static char __pyx_k_You_have_loaded_Tree_version_whi[] = "You have loaded Tree version which cannot be imported";
|
|
static char __pyx_k_ndarray_is_not_Fortran_contiguou[] = "ndarray is not Fortran contiguous";
|
|
static char __pyx_k_Format_string_allocated_too_shor_2[] = "Format string allocated too short.";
|
|
static PyObject *__pyx_n_s_C;
|
|
static PyObject *__pyx_n_s_DOUBLE;
|
|
static PyObject *__pyx_n_s_DTYPE;
|
|
static PyObject *__pyx_kp_s_Did_not_recognise_loaded_array_l;
|
|
static PyObject *__pyx_kp_u_Format_string_allocated_too_shor;
|
|
static PyObject *__pyx_kp_u_Format_string_allocated_too_shor_2;
|
|
static PyObject *__pyx_n_s_MemoryError;
|
|
static PyObject *__pyx_n_s_NODE_DTYPE;
|
|
static PyObject *__pyx_kp_s_No_support_for_np_int64_index_ba;
|
|
static PyObject *__pyx_kp_u_Non_native_byte_order_not_suppor;
|
|
static PyObject *__pyx_n_s_RuntimeError;
|
|
static PyObject *__pyx_n_s_TREE_LEAF;
|
|
static PyObject *__pyx_n_s_TREE_UNDEFINED;
|
|
static PyObject *__pyx_kp_s_Users_ajoly_git_scikit_learn_sk;
|
|
static PyObject *__pyx_n_s_ValueError;
|
|
static PyObject *__pyx_n_s_X;
|
|
static PyObject *__pyx_kp_s_X_dtype_should_be_np_float32_got;
|
|
static PyObject *__pyx_kp_s_X_should_be_in_csc_format;
|
|
static PyObject *__pyx_kp_s_X_should_be_in_csr_matrix_format;
|
|
static PyObject *__pyx_kp_s_X_should_be_in_np_ndarray_format;
|
|
static PyObject *__pyx_kp_s_You_have_loaded_Tree_version_whi;
|
|
static PyObject *__pyx_n_s_apply;
|
|
static PyObject *__pyx_n_s_argsort;
|
|
static PyObject *__pyx_n_s_asarray;
|
|
static PyObject *__pyx_n_s_ascontiguousarray;
|
|
static PyObject *__pyx_n_s_asfortranarray;
|
|
static PyObject *__pyx_n_s_axis;
|
|
static PyObject *__pyx_n_s_build;
|
|
static PyObject *__pyx_n_s_c_contiguous;
|
|
static PyObject *__pyx_n_s_clip;
|
|
static PyObject *__pyx_n_s_compute_feature_importances;
|
|
static PyObject *__pyx_n_s_contiguous;
|
|
static PyObject *__pyx_n_s_copy;
|
|
static PyObject *__pyx_kp_s_could_not_allocate_d_bytes;
|
|
static PyObject *__pyx_kp_s_could_not_allocate_d_d_bytes;
|
|
static PyObject *__pyx_n_s_criterion;
|
|
static PyObject *__pyx_n_s_csc_matrix;
|
|
static PyObject *__pyx_n_s_csr_matrix;
|
|
static PyObject *__pyx_n_s_data;
|
|
static PyObject *__pyx_n_s_dtype;
|
|
static PyObject *__pyx_n_s_feature;
|
|
static PyObject *__pyx_n_s_flags;
|
|
static PyObject *__pyx_n_s_float32;
|
|
static PyObject *__pyx_n_s_float64;
|
|
static PyObject *__pyx_n_s_formats;
|
|
static PyObject *__pyx_n_s_getstate;
|
|
static PyObject *__pyx_n_s_import;
|
|
static PyObject *__pyx_n_s_impurity;
|
|
static PyObject *__pyx_n_s_indices;
|
|
static PyObject *__pyx_n_s_indptr;
|
|
static PyObject *__pyx_n_s_inf;
|
|
static PyObject *__pyx_n_s_int32;
|
|
static PyObject *__pyx_n_s_intp;
|
|
static PyObject *__pyx_n_s_issparse;
|
|
static PyObject *__pyx_n_s_itemsize;
|
|
static PyObject *__pyx_n_s_left_child;
|
|
static PyObject *__pyx_n_s_main;
|
|
static PyObject *__pyx_n_s_max;
|
|
static PyObject *__pyx_n_s_max_depth;
|
|
static PyObject *__pyx_n_s_max_features;
|
|
static PyObject *__pyx_n_s_max_leaf_nodes;
|
|
static PyObject *__pyx_n_s_min_samples_leaf;
|
|
static PyObject *__pyx_n_s_min_samples_split;
|
|
static PyObject *__pyx_n_s_min_weight_leaf;
|
|
static PyObject *__pyx_n_s_mode;
|
|
static PyObject *__pyx_n_s_n_classes;
|
|
static PyObject *__pyx_n_s_n_features;
|
|
static PyObject *__pyx_n_s_n_node_samples;
|
|
static PyObject *__pyx_n_s_n_outputs;
|
|
static PyObject *__pyx_n_s_names;
|
|
static PyObject *__pyx_kp_u_ndarray_is_not_C_contiguous;
|
|
static PyObject *__pyx_kp_u_ndarray_is_not_Fortran_contiguou;
|
|
static PyObject *__pyx_n_s_ndim;
|
|
static PyObject *__pyx_n_s_node_count;
|
|
static PyObject *__pyx_n_s_nodes;
|
|
static PyObject *__pyx_n_s_normalize;
|
|
static PyObject *__pyx_n_s_np;
|
|
static PyObject *__pyx_n_s_numpy;
|
|
static PyObject *__pyx_n_s_offsets;
|
|
static PyObject *__pyx_n_s_order;
|
|
static PyObject *__pyx_n_s_p;
|
|
static PyObject *__pyx_n_s_predict;
|
|
static PyObject *__pyx_n_s_pyx_getbuffer;
|
|
static PyObject *__pyx_n_s_pyx_releasebuffer;
|
|
static PyObject *__pyx_n_s_pyx_vtable;
|
|
static PyObject *__pyx_n_s_randint;
|
|
static PyObject *__pyx_n_s_random_state;
|
|
static PyObject *__pyx_n_s_range;
|
|
static PyObject *__pyx_n_s_realloc_test;
|
|
static PyObject *__pyx_n_s_reshape;
|
|
static PyObject *__pyx_kp_s_resizing_tree_to_d;
|
|
static PyObject *__pyx_n_s_right_child;
|
|
static PyObject *__pyx_n_s_sample_weight;
|
|
static PyObject *__pyx_n_s_scipy_sparse;
|
|
static PyObject *__pyx_n_s_shape;
|
|
static PyObject *__pyx_n_s_sklearn_tree__tree;
|
|
static PyObject *__pyx_n_s_sort_indices;
|
|
static PyObject *__pyx_n_s_splitter;
|
|
static PyObject *__pyx_n_s_strides;
|
|
static PyObject *__pyx_n_s_sum;
|
|
static PyObject *__pyx_n_s_take;
|
|
static PyObject *__pyx_n_s_test;
|
|
static PyObject *__pyx_n_s_threshold;
|
|
static PyObject *__pyx_n_s_tocsc;
|
|
static PyObject *__pyx_n_s_tree;
|
|
static PyObject *__pyx_kp_u_unknown_dtype_code_in_numpy_pxd;
|
|
static PyObject *__pyx_n_s_values;
|
|
static PyObject *__pyx_n_s_weighted_n_node_samples;
|
|
static PyObject *__pyx_n_s_y;
|
|
static PyObject *__pyx_n_s_zeros;
|
|
static PyObject *__pyx_int_0;
|
|
static PyObject *__pyx_int_1;
|
|
static PyObject *__pyx_int_neg_1;
|
|
static PyObject *__pyx_int_neg_2;
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_k__5;
|
|
static PyObject *__pyx_tuple_;
|
|
static PyObject *__pyx_tuple__2;
|
|
static PyObject *__pyx_tuple__3;
|
|
static PyObject *__pyx_tuple__4;
|
|
static PyObject *__pyx_tuple__6;
|
|
static PyObject *__pyx_tuple__7;
|
|
static PyObject *__pyx_tuple__8;
|
|
static PyObject *__pyx_tuple__9;
|
|
static PyObject *__pyx_tuple__10;
|
|
static PyObject *__pyx_tuple__11;
|
|
static PyObject *__pyx_tuple__12;
|
|
static PyObject *__pyx_codeobj__13;
|
|
|
|
/* "sklearn/tree/_tree.pyx":100
|
|
* """Interface for impurity criteria."""
|
|
*
|
|
* cdef void init(self, DOUBLE_t* y, SIZE_t y_stride, DOUBLE_t* sample_weight, # <<<<<<<<<<<<<<
|
|
* double weighted_n_samples, SIZE_t* samples, SIZE_t start,
|
|
* SIZE_t end) nogil:
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_init(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_y, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_y_stride, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight, CYTHON_UNUSED double __pyx_v_weighted_n_samples, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":107
|
|
* pass
|
|
*
|
|
* cdef void reset(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Reset the criterion at pos=start."""
|
|
* pass
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_reset(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":111
|
|
* pass
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil: # <<<<<<<<<<<<<<
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_update(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_new_pos) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":116
|
|
* pass
|
|
*
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_9Criterion_node_impurity(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self) {
|
|
double __pyx_r;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":121
|
|
* pass
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_children_impurity(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, CYTHON_UNUSED double *__pyx_v_impurity_left, CYTHON_UNUSED double *__pyx_v_impurity_right) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":127
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* pass
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_node_value(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, CYTHON_UNUSED double *__pyx_v_dest) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":131
|
|
* pass
|
|
*
|
|
* cdef double impurity_improvement(self, double impurity) nogil: # <<<<<<<<<<<<<<
|
|
* """Weighted impurity improvement, i.e.
|
|
*
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_9Criterion_impurity_improvement(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, double __pyx_v_impurity) {
|
|
double __pyx_v_impurity_left;
|
|
double __pyx_v_impurity_right;
|
|
double __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":143
|
|
* cdef double impurity_right
|
|
*
|
|
* self.children_impurity(&impurity_left, &impurity_right) # <<<<<<<<<<<<<<
|
|
*
|
|
* return ((self.weighted_n_node_samples / self.weighted_n_samples) *
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_vtab)->children_impurity(__pyx_v_self, (&__pyx_v_impurity_left), (&__pyx_v_impurity_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":145
|
|
* self.children_impurity(&impurity_left, &impurity_right)
|
|
*
|
|
* return ((self.weighted_n_node_samples / self.weighted_n_samples) * # <<<<<<<<<<<<<<
|
|
* (impurity - self.weighted_n_right / self.weighted_n_node_samples * impurity_right
|
|
* - self.weighted_n_left / self.weighted_n_node_samples * impurity_left))
|
|
*/
|
|
__pyx_r = ((__pyx_v_self->weighted_n_node_samples / __pyx_v_self->weighted_n_samples) * ((__pyx_v_impurity - ((__pyx_v_self->weighted_n_right / __pyx_v_self->weighted_n_node_samples) * __pyx_v_impurity_right)) - ((__pyx_v_self->weighted_n_left / __pyx_v_self->weighted_n_node_samples) * __pyx_v_impurity_left)));
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":131
|
|
* pass
|
|
*
|
|
* cdef double impurity_improvement(self, double impurity) nogil: # <<<<<<<<<<<<<<
|
|
* """Weighted impurity improvement, i.e.
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":158
|
|
* cdef double* label_count_total
|
|
*
|
|
* def __cinit__(self, SIZE_t n_outputs, # <<<<<<<<<<<<<<
|
|
* np.ndarray[SIZE_t, ndim=1] n_classes):
|
|
* # Default values
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
PyArrayObject *__pyx_v_n_classes = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_n_outputs,&__pyx_n_s_n_classes,0};
|
|
PyObject* values[2] = {0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_n_outputs)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_n_classes)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 2, 2, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 158; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 158; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 2) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
}
|
|
__pyx_v_n_outputs = __Pyx_PyInt_As_Py_intptr_t(values[0]); if (unlikely((__pyx_v_n_outputs == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 158; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_n_classes = ((PyArrayObject *)values[1]);
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 2, 2, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 158; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_n_classes), __pyx_ptype_5numpy_ndarray, 1, "n_classes", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 159; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self), __pyx_v_n_outputs, __pyx_v_n_classes);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs, PyArrayObject *__pyx_v_n_classes) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_elements;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_n_classes;
|
|
__Pyx_Buffer __pyx_pybuffer_n_classes;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
int __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
__pyx_pybuffer_n_classes.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_n_classes.refcount = 0;
|
|
__pyx_pybuffernd_n_classes.data = NULL;
|
|
__pyx_pybuffernd_n_classes.rcbuffer = &__pyx_pybuffer_n_classes;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_n_classes.rcbuffer->pybuffer, (PyObject*)__pyx_v_n_classes, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_SIZE_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 158; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_n_classes.diminfo[0].strides = __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_n_classes.diminfo[0].shape = __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.shape[0];
|
|
|
|
/* "sklearn/tree/_tree.pyx":161
|
|
* np.ndarray[SIZE_t, ndim=1] n_classes):
|
|
* # Default values
|
|
* self.y = NULL # <<<<<<<<<<<<<<
|
|
* self.y_stride = 0
|
|
* self.sample_weight = NULL
|
|
*/
|
|
__pyx_v_self->__pyx_base.y = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":162
|
|
* # Default values
|
|
* self.y = NULL
|
|
* self.y_stride = 0 # <<<<<<<<<<<<<<
|
|
* self.sample_weight = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.y_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":163
|
|
* self.y = NULL
|
|
* self.y_stride = 0
|
|
* self.sample_weight = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* self.samples = NULL
|
|
*/
|
|
__pyx_v_self->__pyx_base.sample_weight = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":165
|
|
* self.sample_weight = NULL
|
|
*
|
|
* self.samples = NULL # <<<<<<<<<<<<<<
|
|
* self.start = 0
|
|
* self.pos = 0
|
|
*/
|
|
__pyx_v_self->__pyx_base.samples = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":166
|
|
*
|
|
* self.samples = NULL
|
|
* self.start = 0 # <<<<<<<<<<<<<<
|
|
* self.pos = 0
|
|
* self.end = 0
|
|
*/
|
|
__pyx_v_self->__pyx_base.start = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":167
|
|
* self.samples = NULL
|
|
* self.start = 0
|
|
* self.pos = 0 # <<<<<<<<<<<<<<
|
|
* self.end = 0
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.pos = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":168
|
|
* self.start = 0
|
|
* self.pos = 0
|
|
* self.end = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* self.n_outputs = n_outputs
|
|
*/
|
|
__pyx_v_self->__pyx_base.end = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":170
|
|
* self.end = 0
|
|
*
|
|
* self.n_outputs = n_outputs # <<<<<<<<<<<<<<
|
|
* self.n_node_samples = 0
|
|
* self.weighted_n_node_samples = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.n_outputs = __pyx_v_n_outputs;
|
|
|
|
/* "sklearn/tree/_tree.pyx":171
|
|
*
|
|
* self.n_outputs = n_outputs
|
|
* self.n_node_samples = 0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_node_samples = 0.0
|
|
* self.weighted_n_left = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.n_node_samples = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":172
|
|
* self.n_outputs = n_outputs
|
|
* self.n_node_samples = 0
|
|
* self.weighted_n_node_samples = 0.0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_left = 0.0
|
|
* self.weighted_n_right = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_node_samples = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":173
|
|
* self.n_node_samples = 0
|
|
* self.weighted_n_node_samples = 0.0
|
|
* self.weighted_n_left = 0.0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_right = 0.0
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":174
|
|
* self.weighted_n_node_samples = 0.0
|
|
* self.weighted_n_left = 0.0
|
|
* self.weighted_n_right = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* self.label_count_left = NULL
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":176
|
|
* self.weighted_n_right = 0.0
|
|
*
|
|
* self.label_count_left = NULL # <<<<<<<<<<<<<<
|
|
* self.label_count_right = NULL
|
|
* self.label_count_total = NULL
|
|
*/
|
|
__pyx_v_self->label_count_left = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":177
|
|
*
|
|
* self.label_count_left = NULL
|
|
* self.label_count_right = NULL # <<<<<<<<<<<<<<
|
|
* self.label_count_total = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->label_count_right = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":178
|
|
* self.label_count_left = NULL
|
|
* self.label_count_right = NULL
|
|
* self.label_count_total = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* # Count labels for each output
|
|
*/
|
|
__pyx_v_self->label_count_total = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":181
|
|
*
|
|
* # Count labels for each output
|
|
* self.n_classes = NULL # <<<<<<<<<<<<<<
|
|
* safe_realloc(&self.n_classes, n_outputs)
|
|
*
|
|
*/
|
|
__pyx_v_self->n_classes = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":182
|
|
* # Count labels for each output
|
|
* self.n_classes = NULL
|
|
* safe_realloc(&self.n_classes, n_outputs) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t k = 0
|
|
*/
|
|
__pyx_t_1 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->n_classes), __pyx_v_n_outputs); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 182; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":184
|
|
* safe_realloc(&self.n_classes, n_outputs)
|
|
*
|
|
* cdef SIZE_t k = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = 0
|
|
*
|
|
*/
|
|
__pyx_v_k = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":185
|
|
*
|
|
* cdef SIZE_t k = 0
|
|
* cdef SIZE_t label_count_stride = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_label_count_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":187
|
|
* cdef SIZE_t label_count_stride = 0
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* self.n_classes[k] = n_classes[k]
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_outputs;
|
|
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
|
|
__pyx_v_k = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":188
|
|
*
|
|
* for k in range(n_outputs):
|
|
* self.n_classes[k] = n_classes[k] # <<<<<<<<<<<<<<
|
|
*
|
|
* if n_classes[k] > label_count_stride:
|
|
*/
|
|
__pyx_t_4 = __pyx_v_k;
|
|
(__pyx_v_self->n_classes[__pyx_v_k]) = (*__Pyx_BufPtrStrided1d(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.buf, __pyx_t_4, __pyx_pybuffernd_n_classes.diminfo[0].strides));
|
|
|
|
/* "sklearn/tree/_tree.pyx":190
|
|
* self.n_classes[k] = n_classes[k]
|
|
*
|
|
* if n_classes[k] > label_count_stride: # <<<<<<<<<<<<<<
|
|
* label_count_stride = n_classes[k]
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_k;
|
|
__pyx_t_6 = (((*__Pyx_BufPtrStrided1d(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.buf, __pyx_t_5, __pyx_pybuffernd_n_classes.diminfo[0].strides)) > __pyx_v_label_count_stride) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":191
|
|
*
|
|
* if n_classes[k] > label_count_stride:
|
|
* label_count_stride = n_classes[k] # <<<<<<<<<<<<<<
|
|
*
|
|
* self.label_count_stride = label_count_stride
|
|
*/
|
|
__pyx_t_7 = __pyx_v_k;
|
|
__pyx_v_label_count_stride = (*__Pyx_BufPtrStrided1d(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.buf, __pyx_t_7, __pyx_pybuffernd_n_classes.diminfo[0].strides));
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":193
|
|
* label_count_stride = n_classes[k]
|
|
*
|
|
* self.label_count_stride = label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* # Allocate counters
|
|
*/
|
|
__pyx_v_self->label_count_stride = __pyx_v_label_count_stride;
|
|
|
|
/* "sklearn/tree/_tree.pyx":196
|
|
*
|
|
* # Allocate counters
|
|
* cdef SIZE_t n_elements = n_outputs * label_count_stride # <<<<<<<<<<<<<<
|
|
* self.label_count_left = <double*> calloc(n_elements, sizeof(double))
|
|
* self.label_count_right = <double*> calloc(n_elements, sizeof(double))
|
|
*/
|
|
__pyx_v_n_elements = (__pyx_v_n_outputs * __pyx_v_label_count_stride);
|
|
|
|
/* "sklearn/tree/_tree.pyx":197
|
|
* # Allocate counters
|
|
* cdef SIZE_t n_elements = n_outputs * label_count_stride
|
|
* self.label_count_left = <double*> calloc(n_elements, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.label_count_right = <double*> calloc(n_elements, sizeof(double))
|
|
* self.label_count_total = <double*> calloc(n_elements, sizeof(double))
|
|
*/
|
|
__pyx_v_self->label_count_left = ((double *)calloc(__pyx_v_n_elements, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":198
|
|
* cdef SIZE_t n_elements = n_outputs * label_count_stride
|
|
* self.label_count_left = <double*> calloc(n_elements, sizeof(double))
|
|
* self.label_count_right = <double*> calloc(n_elements, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.label_count_total = <double*> calloc(n_elements, sizeof(double))
|
|
*
|
|
*/
|
|
__pyx_v_self->label_count_right = ((double *)calloc(__pyx_v_n_elements, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":199
|
|
* self.label_count_left = <double*> calloc(n_elements, sizeof(double))
|
|
* self.label_count_right = <double*> calloc(n_elements, sizeof(double))
|
|
* self.label_count_total = <double*> calloc(n_elements, sizeof(double)) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Check for allocation errors
|
|
*/
|
|
__pyx_v_self->label_count_total = ((double *)calloc(__pyx_v_n_elements, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":202
|
|
*
|
|
* # Check for allocation errors
|
|
* if (self.label_count_left == NULL or # <<<<<<<<<<<<<<
|
|
* self.label_count_right == NULL or
|
|
* self.label_count_total == NULL):
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_self->label_count_left == NULL) != 0);
|
|
if (!__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":203
|
|
* # Check for allocation errors
|
|
* if (self.label_count_left == NULL or
|
|
* self.label_count_right == NULL or # <<<<<<<<<<<<<<
|
|
* self.label_count_total == NULL):
|
|
* raise MemoryError()
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_self->label_count_right == NULL) != 0);
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":204
|
|
* if (self.label_count_left == NULL or
|
|
* self.label_count_right == NULL or
|
|
* self.label_count_total == NULL): # <<<<<<<<<<<<<<
|
|
* raise MemoryError()
|
|
*
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_self->label_count_total == NULL) != 0);
|
|
__pyx_t_10 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_10 = __pyx_t_8;
|
|
}
|
|
__pyx_t_8 = __pyx_t_10;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_6;
|
|
}
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":205
|
|
* self.label_count_right == NULL or
|
|
* self.label_count_total == NULL):
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
* def __dealloc__(self):
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 205; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":158
|
|
* cdef double* label_count_total
|
|
*
|
|
* def __cinit__(self, SIZE_t n_outputs, # <<<<<<<<<<<<<<
|
|
* np.ndarray[SIZE_t, ndim=1] n_classes):
|
|
* # Default values
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_n_classes.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_n_classes.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":207
|
|
* raise MemoryError()
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.n_classes)
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_3__dealloc__(PyObject *__pyx_v_self); /*proto*/
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_3__dealloc__(PyObject *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0);
|
|
__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_2__dealloc__(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":209
|
|
* def __dealloc__(self):
|
|
* """Destructor."""
|
|
* free(self.n_classes) # <<<<<<<<<<<<<<
|
|
* free(self.label_count_left)
|
|
* free(self.label_count_right)
|
|
*/
|
|
free(__pyx_v_self->n_classes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":210
|
|
* """Destructor."""
|
|
* free(self.n_classes)
|
|
* free(self.label_count_left) # <<<<<<<<<<<<<<
|
|
* free(self.label_count_right)
|
|
* free(self.label_count_total)
|
|
*/
|
|
free(__pyx_v_self->label_count_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":211
|
|
* free(self.n_classes)
|
|
* free(self.label_count_left)
|
|
* free(self.label_count_right) # <<<<<<<<<<<<<<
|
|
* free(self.label_count_total)
|
|
*
|
|
*/
|
|
free(__pyx_v_self->label_count_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":212
|
|
* free(self.label_count_left)
|
|
* free(self.label_count_right)
|
|
* free(self.label_count_total) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __reduce__(self):
|
|
*/
|
|
free(__pyx_v_self->label_count_total);
|
|
|
|
/* "sklearn/tree/_tree.pyx":207
|
|
* raise MemoryError()
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.n_classes)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":214
|
|
* free(self.label_count_total)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (ClassificationCriterion,
|
|
* (self.n_outputs,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_4__reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":215
|
|
*
|
|
* def __reduce__(self):
|
|
* return (ClassificationCriterion, # <<<<<<<<<<<<<<
|
|
* (self.n_outputs,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs)),
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":216
|
|
* def __reduce__(self):
|
|
* return (ClassificationCriterion,
|
|
* (self.n_outputs, # <<<<<<<<<<<<<<
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs)),
|
|
* self.__getstate__())
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.n_outputs); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 216; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":217
|
|
* return (ClassificationCriterion,
|
|
* (self.n_outputs,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs)), # <<<<<<<<<<<<<<
|
|
* self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_2 = ((PyObject *)__pyx_f_7sklearn_4tree_5_tree_sizet_ptr_to_ndarray(__pyx_v_self->n_classes, __pyx_v_self->__pyx_base.n_outputs)); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 217; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":216
|
|
* def __reduce__(self):
|
|
* return (ClassificationCriterion,
|
|
* (self.n_outputs, # <<<<<<<<<<<<<<
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs)),
|
|
* self.__getstate__())
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(2); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 216; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":218
|
|
* (self.n_outputs,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs)),
|
|
* self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __getstate__(self):
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 218; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 218; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":215
|
|
*
|
|
* def __reduce__(self):
|
|
* return (ClassificationCriterion, # <<<<<<<<<<<<<<
|
|
* (self.n_outputs,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs)),
|
|
*/
|
|
__pyx_t_2 = PyTuple_New(3); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion)));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion)));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 2, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_1 = 0;
|
|
__pyx_r = __pyx_t_2;
|
|
__pyx_t_2 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":214
|
|
* free(self.label_count_total)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (ClassificationCriterion,
|
|
* (self.n_outputs,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":220
|
|
* self.__getstate__())
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* return {}
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_7__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_7__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__getstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_6__getstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_6__getstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
|
|
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("__getstate__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":221
|
|
*
|
|
* def __getstate__(self):
|
|
* return {} # <<<<<<<<<<<<<<
|
|
*
|
|
* def __setstate__(self, d):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 221; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":220
|
|
* self.__getstate__())
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* return {}
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.__getstate__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":223
|
|
* return {}
|
|
*
|
|
* def __setstate__(self, d): # <<<<<<<<<<<<<<
|
|
* pass
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_9__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_9__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_8__setstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self), ((PyObject *)__pyx_v_d));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_8__setstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__", 0);
|
|
|
|
/* function exit code */
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":226
|
|
* pass
|
|
*
|
|
* cdef void init(self, DOUBLE_t* y, SIZE_t y_stride, # <<<<<<<<<<<<<<
|
|
* DOUBLE_t* sample_weight, double weighted_n_samples,
|
|
* SIZE_t* samples, SIZE_t start, SIZE_t end) nogil:
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_init(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_y_stride, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight, double __pyx_v_weighted_n_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end) {
|
|
double __pyx_v_weighted_n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_total;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_c;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_w;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_offset;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_2;
|
|
double *__pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
int __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":232
|
|
* children samples[start:start] and samples[start:end]."""
|
|
* # Initialize fields
|
|
* self.y = y # <<<<<<<<<<<<<<
|
|
* self.y_stride = y_stride
|
|
* self.sample_weight = sample_weight
|
|
*/
|
|
__pyx_v_self->__pyx_base.y = __pyx_v_y;
|
|
|
|
/* "sklearn/tree/_tree.pyx":233
|
|
* # Initialize fields
|
|
* self.y = y
|
|
* self.y_stride = y_stride # <<<<<<<<<<<<<<
|
|
* self.sample_weight = sample_weight
|
|
* self.samples = samples
|
|
*/
|
|
__pyx_v_self->__pyx_base.y_stride = __pyx_v_y_stride;
|
|
|
|
/* "sklearn/tree/_tree.pyx":234
|
|
* self.y = y
|
|
* self.y_stride = y_stride
|
|
* self.sample_weight = sample_weight # <<<<<<<<<<<<<<
|
|
* self.samples = samples
|
|
* self.start = start
|
|
*/
|
|
__pyx_v_self->__pyx_base.sample_weight = __pyx_v_sample_weight;
|
|
|
|
/* "sklearn/tree/_tree.pyx":235
|
|
* self.y_stride = y_stride
|
|
* self.sample_weight = sample_weight
|
|
* self.samples = samples # <<<<<<<<<<<<<<
|
|
* self.start = start
|
|
* self.end = end
|
|
*/
|
|
__pyx_v_self->__pyx_base.samples = __pyx_v_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":236
|
|
* self.sample_weight = sample_weight
|
|
* self.samples = samples
|
|
* self.start = start # <<<<<<<<<<<<<<
|
|
* self.end = end
|
|
* self.n_node_samples = end - start
|
|
*/
|
|
__pyx_v_self->__pyx_base.start = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":237
|
|
* self.samples = samples
|
|
* self.start = start
|
|
* self.end = end # <<<<<<<<<<<<<<
|
|
* self.n_node_samples = end - start
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
*/
|
|
__pyx_v_self->__pyx_base.end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":238
|
|
* self.start = start
|
|
* self.end = end
|
|
* self.n_node_samples = end - start # <<<<<<<<<<<<<<
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
* cdef double weighted_n_node_samples = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.n_node_samples = (__pyx_v_end - __pyx_v_start);
|
|
|
|
/* "sklearn/tree/_tree.pyx":239
|
|
* self.end = end
|
|
* self.n_node_samples = end - start
|
|
* self.weighted_n_samples = weighted_n_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_node_samples = 0.0
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_samples = __pyx_v_weighted_n_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":240
|
|
* self.n_node_samples = end - start
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
* cdef double weighted_n_node_samples = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize label_count_total and weighted_n_node_samples
|
|
*/
|
|
__pyx_v_weighted_n_node_samples = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":243
|
|
*
|
|
* # Initialize label_count_total and weighted_n_node_samples
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":244
|
|
* # Initialize label_count_total and weighted_n_node_samples
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->n_classes;
|
|
__pyx_v_n_classes = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":245
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":246
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t i = 0
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->label_count_total;
|
|
__pyx_v_label_count_total = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":248
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*
|
|
* cdef SIZE_t i = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t p = 0
|
|
* cdef SIZE_t k = 0
|
|
*/
|
|
__pyx_v_i = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":249
|
|
*
|
|
* cdef SIZE_t i = 0
|
|
* cdef SIZE_t p = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t k = 0
|
|
* cdef SIZE_t c = 0
|
|
*/
|
|
__pyx_v_p = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":250
|
|
* cdef SIZE_t i = 0
|
|
* cdef SIZE_t p = 0
|
|
* cdef SIZE_t k = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t c = 0
|
|
* cdef DOUBLE_t w = 1.0
|
|
*/
|
|
__pyx_v_k = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":251
|
|
* cdef SIZE_t p = 0
|
|
* cdef SIZE_t k = 0
|
|
* cdef SIZE_t c = 0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t w = 1.0
|
|
* cdef SIZE_t offset = 0
|
|
*/
|
|
__pyx_v_c = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":252
|
|
* cdef SIZE_t k = 0
|
|
* cdef SIZE_t c = 0
|
|
* cdef DOUBLE_t w = 1.0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t offset = 0
|
|
*
|
|
*/
|
|
__pyx_v_w = 1.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":253
|
|
* cdef SIZE_t c = 0
|
|
* cdef DOUBLE_t w = 1.0
|
|
* cdef SIZE_t offset = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_offset = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":255
|
|
* cdef SIZE_t offset = 0
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* memset(label_count_total + offset, 0,
|
|
* n_classes[k] * sizeof(double))
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_4 = 0; __pyx_t_4 < __pyx_t_1; __pyx_t_4+=1) {
|
|
__pyx_v_k = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":256
|
|
*
|
|
* for k in range(n_outputs):
|
|
* memset(label_count_total + offset, 0, # <<<<<<<<<<<<<<
|
|
* n_classes[k] * sizeof(double))
|
|
* offset += label_count_stride
|
|
*/
|
|
memset((__pyx_v_label_count_total + __pyx_v_offset), 0, ((__pyx_v_n_classes[__pyx_v_k]) * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":258
|
|
* memset(label_count_total + offset, 0,
|
|
* n_classes[k] * sizeof(double))
|
|
* offset += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* for p in range(start, end):
|
|
*/
|
|
__pyx_v_offset = (__pyx_v_offset + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":260
|
|
* offset += label_count_stride
|
|
*
|
|
* for p in range(start, end): # <<<<<<<<<<<<<<
|
|
* i = samples[p]
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_end;
|
|
for (__pyx_t_4 = __pyx_v_start; __pyx_t_4 < __pyx_t_1; __pyx_t_4+=1) {
|
|
__pyx_v_p = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":261
|
|
*
|
|
* for p in range(start, end):
|
|
* i = samples[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if sample_weight != NULL:
|
|
*/
|
|
__pyx_v_i = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":263
|
|
* i = samples[p]
|
|
*
|
|
* if sample_weight != NULL: # <<<<<<<<<<<<<<
|
|
* w = sample_weight[i]
|
|
*
|
|
*/
|
|
__pyx_t_5 = ((__pyx_v_sample_weight != NULL) != 0);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":264
|
|
*
|
|
* if sample_weight != NULL:
|
|
* w = sample_weight[i] # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_w = (__pyx_v_sample_weight[__pyx_v_i]);
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":266
|
|
* w = sample_weight[i]
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* c = <SIZE_t> y[i * y_stride + k]
|
|
* label_count_total[k * label_count_stride + c] += w
|
|
*/
|
|
__pyx_t_6 = __pyx_v_n_outputs;
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_6; __pyx_t_7+=1) {
|
|
__pyx_v_k = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":267
|
|
*
|
|
* for k in range(n_outputs):
|
|
* c = <SIZE_t> y[i * y_stride + k] # <<<<<<<<<<<<<<
|
|
* label_count_total[k * label_count_stride + c] += w
|
|
*
|
|
*/
|
|
__pyx_v_c = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_y[((__pyx_v_i * __pyx_v_y_stride) + __pyx_v_k)]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":268
|
|
* for k in range(n_outputs):
|
|
* c = <SIZE_t> y[i * y_stride + k]
|
|
* label_count_total[k * label_count_stride + c] += w # <<<<<<<<<<<<<<
|
|
*
|
|
* weighted_n_node_samples += w
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_k * __pyx_v_label_count_stride) + __pyx_v_c);
|
|
(__pyx_v_label_count_total[__pyx_t_8]) = ((__pyx_v_label_count_total[__pyx_t_8]) + __pyx_v_w);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":270
|
|
* label_count_total[k * label_count_stride + c] += w
|
|
*
|
|
* weighted_n_node_samples += w # <<<<<<<<<<<<<<
|
|
*
|
|
* self.weighted_n_node_samples = weighted_n_node_samples
|
|
*/
|
|
__pyx_v_weighted_n_node_samples = (__pyx_v_weighted_n_node_samples + __pyx_v_w);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":272
|
|
* weighted_n_node_samples += w
|
|
*
|
|
* self.weighted_n_node_samples = weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reset to pos=start
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_node_samples = __pyx_v_weighted_n_node_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":275
|
|
*
|
|
* # Reset to pos=start
|
|
* self.reset() # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void reset(self) nogil:
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.reset(((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self));
|
|
|
|
/* "sklearn/tree/_tree.pyx":226
|
|
* pass
|
|
*
|
|
* cdef void init(self, DOUBLE_t* y, SIZE_t y_stride, # <<<<<<<<<<<<<<
|
|
* DOUBLE_t* sample_weight, double weighted_n_samples,
|
|
* SIZE_t* samples, SIZE_t start, SIZE_t end) nogil:
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":277
|
|
* self.reset()
|
|
*
|
|
* cdef void reset(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Reset the criterion at pos=start."""
|
|
* self.pos = self.start
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_reset(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_total;
|
|
double *__pyx_v_label_count_left;
|
|
double *__pyx_v_label_count_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
double __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":279
|
|
* cdef void reset(self) nogil:
|
|
* """Reset the criterion at pos=start."""
|
|
* self.pos = self.start # <<<<<<<<<<<<<<
|
|
*
|
|
* self.weighted_n_left = 0.0
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.start;
|
|
__pyx_v_self->__pyx_base.pos = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":281
|
|
* self.pos = self.start
|
|
*
|
|
* self.weighted_n_left = 0.0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_right = self.weighted_n_node_samples
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":282
|
|
*
|
|
* self.weighted_n_left = 0.0
|
|
* self.weighted_n_right = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_self->__pyx_base.weighted_n_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":284
|
|
* self.weighted_n_right = self.weighted_n_node_samples
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":285
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->n_classes;
|
|
__pyx_v_n_classes = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":286
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef double* label_count_left = self.label_count_left
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":287
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->label_count_total;
|
|
__pyx_v_label_count_total = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":288
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef double* label_count_left = self.label_count_left # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->label_count_left;
|
|
__pyx_v_label_count_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":289
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t k = 0
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->label_count_right;
|
|
__pyx_v_label_count_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":291
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
* cdef SIZE_t k = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_k = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":293
|
|
* cdef SIZE_t k = 0
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* memset(label_count_left, 0, n_classes[k] * sizeof(double))
|
|
* memcpy(label_count_right, label_count_total,
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_1; __pyx_t_5+=1) {
|
|
__pyx_v_k = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":294
|
|
*
|
|
* for k in range(n_outputs):
|
|
* memset(label_count_left, 0, n_classes[k] * sizeof(double)) # <<<<<<<<<<<<<<
|
|
* memcpy(label_count_right, label_count_total,
|
|
* n_classes[k] * sizeof(double))
|
|
*/
|
|
memset(__pyx_v_label_count_left, 0, ((__pyx_v_n_classes[__pyx_v_k]) * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":295
|
|
* for k in range(n_outputs):
|
|
* memset(label_count_left, 0, n_classes[k] * sizeof(double))
|
|
* memcpy(label_count_right, label_count_total, # <<<<<<<<<<<<<<
|
|
* n_classes[k] * sizeof(double))
|
|
*
|
|
*/
|
|
memcpy(__pyx_v_label_count_right, __pyx_v_label_count_total, ((__pyx_v_n_classes[__pyx_v_k]) * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":298
|
|
* n_classes[k] * sizeof(double))
|
|
*
|
|
* label_count_total += label_count_stride # <<<<<<<<<<<<<<
|
|
* label_count_left += label_count_stride
|
|
* label_count_right += label_count_stride
|
|
*/
|
|
__pyx_v_label_count_total = (__pyx_v_label_count_total + __pyx_v_label_count_stride);
|
|
|
|
/* "sklearn/tree/_tree.pyx":299
|
|
*
|
|
* label_count_total += label_count_stride
|
|
* label_count_left += label_count_stride # <<<<<<<<<<<<<<
|
|
* label_count_right += label_count_stride
|
|
*
|
|
*/
|
|
__pyx_v_label_count_left = (__pyx_v_label_count_left + __pyx_v_label_count_stride);
|
|
|
|
/* "sklearn/tree/_tree.pyx":300
|
|
* label_count_total += label_count_stride
|
|
* label_count_left += label_count_stride
|
|
* label_count_right += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil:
|
|
*/
|
|
__pyx_v_label_count_right = (__pyx_v_label_count_right + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":277
|
|
* self.reset()
|
|
*
|
|
* cdef void reset(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Reset the criterion at pos=start."""
|
|
* self.pos = self.start
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":302
|
|
* label_count_right += label_count_stride
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil: # <<<<<<<<<<<<<<
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_update(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_new_pos) {
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_y;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_y_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_pos;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
CYTHON_UNUSED double *__pyx_v_label_count_total;
|
|
double *__pyx_v_label_count_left;
|
|
double *__pyx_v_label_count_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_index;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_w;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_diff_w;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
int __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":305
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
* cdef DOUBLE_t* y = self.y # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t y_stride = self.y_stride
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.y;
|
|
__pyx_v_y = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":306
|
|
* the right child to the left child."""
|
|
* cdef DOUBLE_t* y = self.y
|
|
* cdef SIZE_t y_stride = self.y_stride # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.y_stride;
|
|
__pyx_v_y_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":307
|
|
* cdef DOUBLE_t* y = self.y
|
|
* cdef SIZE_t y_stride = self.y_stride
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.sample_weight;
|
|
__pyx_v_sample_weight = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":309
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight
|
|
*
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t pos = self.pos
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":310
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t pos = self.pos # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.pos;
|
|
__pyx_v_pos = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":312
|
|
* cdef SIZE_t pos = self.pos
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":313
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->n_classes;
|
|
__pyx_v_n_classes = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":314
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef double* label_count_left = self.label_count_left
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":315
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->label_count_total;
|
|
__pyx_v_label_count_total = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":316
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef double* label_count_left = self.label_count_left # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->label_count_left;
|
|
__pyx_v_label_count_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":317
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t i
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->label_count_right;
|
|
__pyx_v_label_count_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":323
|
|
* cdef SIZE_t k
|
|
* cdef SIZE_t label_index
|
|
* cdef DOUBLE_t w = 1.0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t diff_w = 0.0
|
|
*
|
|
*/
|
|
__pyx_v_w = 1.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":324
|
|
* cdef SIZE_t label_index
|
|
* cdef DOUBLE_t w = 1.0
|
|
* cdef DOUBLE_t diff_w = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Note: We assume start <= pos < new_pos <= end
|
|
*/
|
|
__pyx_v_diff_w = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":328
|
|
* # Note: We assume start <= pos < new_pos <= end
|
|
*
|
|
* for p in range(pos, new_pos): # <<<<<<<<<<<<<<
|
|
* i = samples[p]
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_new_pos;
|
|
for (__pyx_t_5 = __pyx_v_pos; __pyx_t_5 < __pyx_t_2; __pyx_t_5+=1) {
|
|
__pyx_v_p = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":329
|
|
*
|
|
* for p in range(pos, new_pos):
|
|
* i = samples[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if sample_weight != NULL:
|
|
*/
|
|
__pyx_v_i = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":331
|
|
* i = samples[p]
|
|
*
|
|
* if sample_weight != NULL: # <<<<<<<<<<<<<<
|
|
* w = sample_weight[i]
|
|
*
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_sample_weight != NULL) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":332
|
|
*
|
|
* if sample_weight != NULL:
|
|
* w = sample_weight[i] # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_w = (__pyx_v_sample_weight[__pyx_v_i]);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":334
|
|
* w = sample_weight[i]
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* label_index = (k * label_count_stride +
|
|
* <SIZE_t> y[i * y_stride + k])
|
|
*/
|
|
__pyx_t_7 = __pyx_v_n_outputs;
|
|
for (__pyx_t_8 = 0; __pyx_t_8 < __pyx_t_7; __pyx_t_8+=1) {
|
|
__pyx_v_k = __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":335
|
|
*
|
|
* for k in range(n_outputs):
|
|
* label_index = (k * label_count_stride + # <<<<<<<<<<<<<<
|
|
* <SIZE_t> y[i * y_stride + k])
|
|
* label_count_left[label_index] += w
|
|
*/
|
|
__pyx_v_label_index = ((__pyx_v_k * __pyx_v_label_count_stride) + ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_y[((__pyx_v_i * __pyx_v_y_stride) + __pyx_v_k)])));
|
|
|
|
/* "sklearn/tree/_tree.pyx":337
|
|
* label_index = (k * label_count_stride +
|
|
* <SIZE_t> y[i * y_stride + k])
|
|
* label_count_left[label_index] += w # <<<<<<<<<<<<<<
|
|
* label_count_right[label_index] -= w
|
|
*
|
|
*/
|
|
__pyx_t_9 = __pyx_v_label_index;
|
|
(__pyx_v_label_count_left[__pyx_t_9]) = ((__pyx_v_label_count_left[__pyx_t_9]) + __pyx_v_w);
|
|
|
|
/* "sklearn/tree/_tree.pyx":338
|
|
* <SIZE_t> y[i * y_stride + k])
|
|
* label_count_left[label_index] += w
|
|
* label_count_right[label_index] -= w # <<<<<<<<<<<<<<
|
|
*
|
|
* diff_w += w
|
|
*/
|
|
__pyx_t_9 = __pyx_v_label_index;
|
|
(__pyx_v_label_count_right[__pyx_t_9]) = ((__pyx_v_label_count_right[__pyx_t_9]) - __pyx_v_w);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":340
|
|
* label_count_right[label_index] -= w
|
|
*
|
|
* diff_w += w # <<<<<<<<<<<<<<
|
|
*
|
|
* self.weighted_n_left += diff_w
|
|
*/
|
|
__pyx_v_diff_w = (__pyx_v_diff_w + __pyx_v_w);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":342
|
|
* diff_w += w
|
|
*
|
|
* self.weighted_n_left += diff_w # <<<<<<<<<<<<<<
|
|
* self.weighted_n_right -= diff_w
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_left = (__pyx_v_self->__pyx_base.weighted_n_left + __pyx_v_diff_w);
|
|
|
|
/* "sklearn/tree/_tree.pyx":343
|
|
*
|
|
* self.weighted_n_left += diff_w
|
|
* self.weighted_n_right -= diff_w # <<<<<<<<<<<<<<
|
|
*
|
|
* self.pos = new_pos
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_right = (__pyx_v_self->__pyx_base.weighted_n_right - __pyx_v_diff_w);
|
|
|
|
/* "sklearn/tree/_tree.pyx":345
|
|
* self.weighted_n_right -= diff_w
|
|
*
|
|
* self.pos = new_pos # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double node_impurity(self) nogil:
|
|
*/
|
|
__pyx_v_self->__pyx_base.pos = __pyx_v_new_pos;
|
|
|
|
/* "sklearn/tree/_tree.pyx":302
|
|
* label_count_right += label_count_stride
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil: # <<<<<<<<<<<<<<
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":347
|
|
* self.pos = new_pos
|
|
*
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* pass
|
|
*
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_node_impurity(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
|
|
double __pyx_r;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":350
|
|
* pass
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* pass
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_children_impurity(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, CYTHON_UNUSED double *__pyx_v_impurity_left, CYTHON_UNUSED double *__pyx_v_impurity_right) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":354
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_node_value(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, double *__pyx_v_dest) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_total;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_2;
|
|
double *__pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":356
|
|
* cdef void node_value(self, double* dest) nogil:
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":357
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->n_classes;
|
|
__pyx_v_n_classes = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":358
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_total = self.label_count_total
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":359
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t k
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->label_count_total;
|
|
__pyx_v_label_count_total = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":362
|
|
* cdef SIZE_t k
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* memcpy(dest, label_count_total, n_classes[k] * sizeof(double))
|
|
* dest += label_count_stride
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_4 = 0; __pyx_t_4 < __pyx_t_1; __pyx_t_4+=1) {
|
|
__pyx_v_k = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":363
|
|
*
|
|
* for k in range(n_outputs):
|
|
* memcpy(dest, label_count_total, n_classes[k] * sizeof(double)) # <<<<<<<<<<<<<<
|
|
* dest += label_count_stride
|
|
* label_count_total += label_count_stride
|
|
*/
|
|
memcpy(__pyx_v_dest, __pyx_v_label_count_total, ((__pyx_v_n_classes[__pyx_v_k]) * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":364
|
|
* for k in range(n_outputs):
|
|
* memcpy(dest, label_count_total, n_classes[k] * sizeof(double))
|
|
* dest += label_count_stride # <<<<<<<<<<<<<<
|
|
* label_count_total += label_count_stride
|
|
*
|
|
*/
|
|
__pyx_v_dest = (__pyx_v_dest + __pyx_v_label_count_stride);
|
|
|
|
/* "sklearn/tree/_tree.pyx":365
|
|
* memcpy(dest, label_count_total, n_classes[k] * sizeof(double))
|
|
* dest += label_count_stride
|
|
* label_count_total += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_v_label_count_total = (__pyx_v_label_count_total + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":354
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":382
|
|
* cross-entropy = - \sum_{k=0}^{K-1} pmk log(pmk)
|
|
* """
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_7Entropy_node_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_Entropy *__pyx_v_self) {
|
|
double __pyx_v_weighted_n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_total;
|
|
double __pyx_v_entropy;
|
|
double __pyx_v_total;
|
|
double __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_c;
|
|
double __pyx_r;
|
|
double __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":385
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_weighted_n_node_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":387
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":388
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.n_classes;
|
|
__pyx_v_n_classes = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":389
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":390
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double entropy = 0.0
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.label_count_total;
|
|
__pyx_v_label_count_total = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":392
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*
|
|
* cdef double entropy = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total = 0.0
|
|
* cdef double tmp
|
|
*/
|
|
__pyx_v_entropy = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":393
|
|
*
|
|
* cdef double entropy = 0.0
|
|
* cdef double total = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double tmp
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_total = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":398
|
|
* cdef SIZE_t c
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* entropy = 0.0
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_outputs;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_2; __pyx_t_5+=1) {
|
|
__pyx_v_k = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":399
|
|
*
|
|
* for k in range(n_outputs):
|
|
* entropy = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
*/
|
|
__pyx_v_entropy = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":401
|
|
* entropy = 0.0
|
|
*
|
|
* for c in range(n_classes[k]): # <<<<<<<<<<<<<<
|
|
* tmp = label_count_total[c]
|
|
* if tmp > 0.0:
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_n_classes[__pyx_v_k]);
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_6; __pyx_t_7+=1) {
|
|
__pyx_v_c = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":402
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_total[c] # <<<<<<<<<<<<<<
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_node_samples
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_label_count_total[__pyx_v_c]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":403
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_total[c]
|
|
* if tmp > 0.0: # <<<<<<<<<<<<<<
|
|
* tmp /= weighted_n_node_samples
|
|
* entropy -= tmp * log(tmp)
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_tmp > 0.0) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":404
|
|
* tmp = label_count_total[c]
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
* entropy -= tmp * log(tmp)
|
|
*
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_tmp / __pyx_v_weighted_n_node_samples);
|
|
|
|
/* "sklearn/tree/_tree.pyx":405
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_node_samples
|
|
* entropy -= tmp * log(tmp) # <<<<<<<<<<<<<<
|
|
*
|
|
* total += entropy
|
|
*/
|
|
__pyx_v_entropy = (__pyx_v_entropy - (__pyx_v_tmp * __pyx_f_7sklearn_4tree_5_tree_log(__pyx_v_tmp)));
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":407
|
|
* entropy -= tmp * log(tmp)
|
|
*
|
|
* total += entropy # <<<<<<<<<<<<<<
|
|
* label_count_total += label_count_stride
|
|
*
|
|
*/
|
|
__pyx_v_total = (__pyx_v_total + __pyx_v_entropy);
|
|
|
|
/* "sklearn/tree/_tree.pyx":408
|
|
*
|
|
* total += entropy
|
|
* label_count_total += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* return total / n_outputs
|
|
*/
|
|
__pyx_v_label_count_total = (__pyx_v_label_count_total + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":410
|
|
* label_count_total += label_count_stride
|
|
*
|
|
* return total / n_outputs # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left,
|
|
*/
|
|
__pyx_r = (__pyx_v_total / __pyx_v_n_outputs);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":382
|
|
* cross-entropy = - \sum_{k=0}^{K-1} pmk log(pmk)
|
|
* """
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":412
|
|
* return total / n_outputs
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of the
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_7Entropy_children_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_Entropy *__pyx_v_self, double *__pyx_v_impurity_left, double *__pyx_v_impurity_right) {
|
|
CYTHON_UNUSED double __pyx_v_weighted_n_node_samples;
|
|
double __pyx_v_weighted_n_left;
|
|
double __pyx_v_weighted_n_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_left;
|
|
double *__pyx_v_label_count_right;
|
|
double __pyx_v_entropy_left;
|
|
double __pyx_v_entropy_right;
|
|
double __pyx_v_total_left;
|
|
double __pyx_v_total_right;
|
|
double __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_c;
|
|
double __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":417
|
|
* left child (samples[start:pos]) and the impurity the right child
|
|
* (samples[pos:end])."""
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_weighted_n_node_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":418
|
|
* (samples[pos:end])."""
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double weighted_n_left = self.weighted_n_left # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_left;
|
|
__pyx_v_weighted_n_left = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":419
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_right;
|
|
__pyx_v_weighted_n_right = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":421
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":422
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_left = self.label_count_left
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.n_classes;
|
|
__pyx_v_n_classes = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":423
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":424
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_left = self.label_count_left # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.label_count_left;
|
|
__pyx_v_label_count_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":425
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double entropy_left = 0.0
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.label_count_right;
|
|
__pyx_v_label_count_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":427
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
* cdef double entropy_left = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double entropy_right = 0.0
|
|
* cdef double total_left = 0.0
|
|
*/
|
|
__pyx_v_entropy_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":428
|
|
*
|
|
* cdef double entropy_left = 0.0
|
|
* cdef double entropy_right = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total_left = 0.0
|
|
* cdef double total_right = 0.0
|
|
*/
|
|
__pyx_v_entropy_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":429
|
|
* cdef double entropy_left = 0.0
|
|
* cdef double entropy_right = 0.0
|
|
* cdef double total_left = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total_right = 0.0
|
|
* cdef double tmp
|
|
*/
|
|
__pyx_v_total_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":430
|
|
* cdef double entropy_right = 0.0
|
|
* cdef double total_left = 0.0
|
|
* cdef double total_right = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double tmp
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_total_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":435
|
|
* cdef SIZE_t c
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* entropy_left = 0.0
|
|
* entropy_right = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_outputs;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_2; __pyx_t_5+=1) {
|
|
__pyx_v_k = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":436
|
|
*
|
|
* for k in range(n_outputs):
|
|
* entropy_left = 0.0 # <<<<<<<<<<<<<<
|
|
* entropy_right = 0.0
|
|
*
|
|
*/
|
|
__pyx_v_entropy_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":437
|
|
* for k in range(n_outputs):
|
|
* entropy_left = 0.0
|
|
* entropy_right = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
*/
|
|
__pyx_v_entropy_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":439
|
|
* entropy_right = 0.0
|
|
*
|
|
* for c in range(n_classes[k]): # <<<<<<<<<<<<<<
|
|
* tmp = label_count_left[c]
|
|
* if tmp > 0.0:
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_n_classes[__pyx_v_k]);
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_6; __pyx_t_7+=1) {
|
|
__pyx_v_c = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":440
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_left[c] # <<<<<<<<<<<<<<
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_left
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_label_count_left[__pyx_v_c]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":441
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_left[c]
|
|
* if tmp > 0.0: # <<<<<<<<<<<<<<
|
|
* tmp /= weighted_n_left
|
|
* entropy_left -= tmp * log(tmp)
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_tmp > 0.0) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":442
|
|
* tmp = label_count_left[c]
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_left # <<<<<<<<<<<<<<
|
|
* entropy_left -= tmp * log(tmp)
|
|
*
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_tmp / __pyx_v_weighted_n_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":443
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_left
|
|
* entropy_left -= tmp * log(tmp) # <<<<<<<<<<<<<<
|
|
*
|
|
* tmp = label_count_right[c]
|
|
*/
|
|
__pyx_v_entropy_left = (__pyx_v_entropy_left - (__pyx_v_tmp * __pyx_f_7sklearn_4tree_5_tree_log(__pyx_v_tmp)));
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":445
|
|
* entropy_left -= tmp * log(tmp)
|
|
*
|
|
* tmp = label_count_right[c] # <<<<<<<<<<<<<<
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_right
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_label_count_right[__pyx_v_c]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":446
|
|
*
|
|
* tmp = label_count_right[c]
|
|
* if tmp > 0.0: # <<<<<<<<<<<<<<
|
|
* tmp /= weighted_n_right
|
|
* entropy_right -= tmp * log(tmp)
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_tmp > 0.0) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":447
|
|
* tmp = label_count_right[c]
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_right # <<<<<<<<<<<<<<
|
|
* entropy_right -= tmp * log(tmp)
|
|
*
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_tmp / __pyx_v_weighted_n_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":448
|
|
* if tmp > 0.0:
|
|
* tmp /= weighted_n_right
|
|
* entropy_right -= tmp * log(tmp) # <<<<<<<<<<<<<<
|
|
*
|
|
* total_left += entropy_left
|
|
*/
|
|
__pyx_v_entropy_right = (__pyx_v_entropy_right - (__pyx_v_tmp * __pyx_f_7sklearn_4tree_5_tree_log(__pyx_v_tmp)));
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":450
|
|
* entropy_right -= tmp * log(tmp)
|
|
*
|
|
* total_left += entropy_left # <<<<<<<<<<<<<<
|
|
* total_right += entropy_right
|
|
* label_count_left += label_count_stride
|
|
*/
|
|
__pyx_v_total_left = (__pyx_v_total_left + __pyx_v_entropy_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":451
|
|
*
|
|
* total_left += entropy_left
|
|
* total_right += entropy_right # <<<<<<<<<<<<<<
|
|
* label_count_left += label_count_stride
|
|
* label_count_right += label_count_stride
|
|
*/
|
|
__pyx_v_total_right = (__pyx_v_total_right + __pyx_v_entropy_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":452
|
|
* total_left += entropy_left
|
|
* total_right += entropy_right
|
|
* label_count_left += label_count_stride # <<<<<<<<<<<<<<
|
|
* label_count_right += label_count_stride
|
|
*
|
|
*/
|
|
__pyx_v_label_count_left = (__pyx_v_label_count_left + __pyx_v_label_count_stride);
|
|
|
|
/* "sklearn/tree/_tree.pyx":453
|
|
* total_right += entropy_right
|
|
* label_count_left += label_count_stride
|
|
* label_count_right += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs
|
|
*/
|
|
__pyx_v_label_count_right = (__pyx_v_label_count_right + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":455
|
|
* label_count_right += label_count_stride
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs # <<<<<<<<<<<<<<
|
|
* impurity_right[0] = total_right / n_outputs
|
|
*
|
|
*/
|
|
(__pyx_v_impurity_left[0]) = (__pyx_v_total_left / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":456
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs
|
|
* impurity_right[0] = total_right / n_outputs # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_impurity_right[0]) = (__pyx_v_total_right / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":412
|
|
* return total / n_outputs
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of the
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":474
|
|
* = 1 - \sum_{k=0}^{K-1} pmk ** 2
|
|
* """
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_4Gini_node_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_Gini *__pyx_v_self) {
|
|
double __pyx_v_weighted_n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_total;
|
|
double __pyx_v_gini;
|
|
double __pyx_v_total;
|
|
double __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_c;
|
|
double __pyx_r;
|
|
double __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":477
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_weighted_n_node_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":479
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":480
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.n_classes;
|
|
__pyx_v_n_classes = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":481
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":482
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_total = self.label_count_total # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double gini = 0.0
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.label_count_total;
|
|
__pyx_v_label_count_total = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":484
|
|
* cdef double* label_count_total = self.label_count_total
|
|
*
|
|
* cdef double gini = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total = 0.0
|
|
* cdef double tmp
|
|
*/
|
|
__pyx_v_gini = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":485
|
|
*
|
|
* cdef double gini = 0.0
|
|
* cdef double total = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double tmp
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_total = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":490
|
|
* cdef SIZE_t c
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* gini = 0.0
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_outputs;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_2; __pyx_t_5+=1) {
|
|
__pyx_v_k = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":491
|
|
*
|
|
* for k in range(n_outputs):
|
|
* gini = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
*/
|
|
__pyx_v_gini = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":493
|
|
* gini = 0.0
|
|
*
|
|
* for c in range(n_classes[k]): # <<<<<<<<<<<<<<
|
|
* tmp = label_count_total[c]
|
|
* gini += tmp * tmp
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_n_classes[__pyx_v_k]);
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_6; __pyx_t_7+=1) {
|
|
__pyx_v_c = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":494
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_total[c] # <<<<<<<<<<<<<<
|
|
* gini += tmp * tmp
|
|
*
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_label_count_total[__pyx_v_c]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":495
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_total[c]
|
|
* gini += tmp * tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* gini = 1.0 - gini / (weighted_n_node_samples *
|
|
*/
|
|
__pyx_v_gini = (__pyx_v_gini + (__pyx_v_tmp * __pyx_v_tmp));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":497
|
|
* gini += tmp * tmp
|
|
*
|
|
* gini = 1.0 - gini / (weighted_n_node_samples * # <<<<<<<<<<<<<<
|
|
* weighted_n_node_samples)
|
|
*
|
|
*/
|
|
__pyx_v_gini = (1.0 - (__pyx_v_gini / (__pyx_v_weighted_n_node_samples * __pyx_v_weighted_n_node_samples)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":500
|
|
* weighted_n_node_samples)
|
|
*
|
|
* total += gini # <<<<<<<<<<<<<<
|
|
* label_count_total += label_count_stride
|
|
*
|
|
*/
|
|
__pyx_v_total = (__pyx_v_total + __pyx_v_gini);
|
|
|
|
/* "sklearn/tree/_tree.pyx":501
|
|
*
|
|
* total += gini
|
|
* label_count_total += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* return total / n_outputs
|
|
*/
|
|
__pyx_v_label_count_total = (__pyx_v_label_count_total + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":503
|
|
* label_count_total += label_count_stride
|
|
*
|
|
* return total / n_outputs # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left,
|
|
*/
|
|
__pyx_r = (__pyx_v_total / __pyx_v_n_outputs);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":474
|
|
* = 1 - \sum_{k=0}^{K-1} pmk ** 2
|
|
* """
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":505
|
|
* return total / n_outputs
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of the
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_4Gini_children_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_Gini *__pyx_v_self, double *__pyx_v_impurity_left, double *__pyx_v_impurity_right) {
|
|
CYTHON_UNUSED double __pyx_v_weighted_n_node_samples;
|
|
double __pyx_v_weighted_n_left;
|
|
double __pyx_v_weighted_n_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_classes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_label_count_stride;
|
|
double *__pyx_v_label_count_left;
|
|
double *__pyx_v_label_count_right;
|
|
double __pyx_v_gini_left;
|
|
double __pyx_v_gini_right;
|
|
CYTHON_UNUSED double __pyx_v_total;
|
|
double __pyx_v_total_left;
|
|
double __pyx_v_total_right;
|
|
double __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_c;
|
|
double __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":510
|
|
* left child (samples[start:pos]) and the impurity the right child
|
|
* (samples[pos:end])."""
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_weighted_n_node_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":511
|
|
* (samples[pos:end])."""
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double weighted_n_left = self.weighted_n_left # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_left;
|
|
__pyx_v_weighted_n_left = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":512
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_right;
|
|
__pyx_v_weighted_n_right = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":514
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":515
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_left = self.label_count_left
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.n_classes;
|
|
__pyx_v_n_classes = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":516
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.label_count_stride;
|
|
__pyx_v_label_count_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":517
|
|
* cdef SIZE_t* n_classes = self.n_classes
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_left = self.label_count_left # <<<<<<<<<<<<<<
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.label_count_left;
|
|
__pyx_v_label_count_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":518
|
|
* cdef SIZE_t label_count_stride = self.label_count_stride
|
|
* cdef double* label_count_left = self.label_count_left
|
|
* cdef double* label_count_right = self.label_count_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double gini_left = 0.0
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.label_count_right;
|
|
__pyx_v_label_count_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":520
|
|
* cdef double* label_count_right = self.label_count_right
|
|
*
|
|
* cdef double gini_left = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double gini_right = 0.0
|
|
* cdef double total = 0.0
|
|
*/
|
|
__pyx_v_gini_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":521
|
|
*
|
|
* cdef double gini_left = 0.0
|
|
* cdef double gini_right = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total = 0.0
|
|
* cdef double total_left = 0.0
|
|
*/
|
|
__pyx_v_gini_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":522
|
|
* cdef double gini_left = 0.0
|
|
* cdef double gini_right = 0.0
|
|
* cdef double total = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total_left = 0.0
|
|
* cdef double total_right = 0.0
|
|
*/
|
|
__pyx_v_total = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":523
|
|
* cdef double gini_right = 0.0
|
|
* cdef double total = 0.0
|
|
* cdef double total_left = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total_right = 0.0
|
|
* cdef double tmp
|
|
*/
|
|
__pyx_v_total_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":524
|
|
* cdef double total = 0.0
|
|
* cdef double total_left = 0.0
|
|
* cdef double total_right = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double tmp
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_total_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":529
|
|
* cdef SIZE_t c
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* gini_left = 0.0
|
|
* gini_right = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_outputs;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_2; __pyx_t_5+=1) {
|
|
__pyx_v_k = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":530
|
|
*
|
|
* for k in range(n_outputs):
|
|
* gini_left = 0.0 # <<<<<<<<<<<<<<
|
|
* gini_right = 0.0
|
|
*
|
|
*/
|
|
__pyx_v_gini_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":531
|
|
* for k in range(n_outputs):
|
|
* gini_left = 0.0
|
|
* gini_right = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
*/
|
|
__pyx_v_gini_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":533
|
|
* gini_right = 0.0
|
|
*
|
|
* for c in range(n_classes[k]): # <<<<<<<<<<<<<<
|
|
* tmp = label_count_left[c]
|
|
* gini_left += tmp * tmp
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_n_classes[__pyx_v_k]);
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_6; __pyx_t_7+=1) {
|
|
__pyx_v_c = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":534
|
|
*
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_left[c] # <<<<<<<<<<<<<<
|
|
* gini_left += tmp * tmp
|
|
* tmp = label_count_right[c]
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_label_count_left[__pyx_v_c]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":535
|
|
* for c in range(n_classes[k]):
|
|
* tmp = label_count_left[c]
|
|
* gini_left += tmp * tmp # <<<<<<<<<<<<<<
|
|
* tmp = label_count_right[c]
|
|
* gini_right += tmp * tmp
|
|
*/
|
|
__pyx_v_gini_left = (__pyx_v_gini_left + (__pyx_v_tmp * __pyx_v_tmp));
|
|
|
|
/* "sklearn/tree/_tree.pyx":536
|
|
* tmp = label_count_left[c]
|
|
* gini_left += tmp * tmp
|
|
* tmp = label_count_right[c] # <<<<<<<<<<<<<<
|
|
* gini_right += tmp * tmp
|
|
*
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_label_count_right[__pyx_v_c]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":537
|
|
* gini_left += tmp * tmp
|
|
* tmp = label_count_right[c]
|
|
* gini_right += tmp * tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* gini_left = 1.0 - gini_left / (weighted_n_left *
|
|
*/
|
|
__pyx_v_gini_right = (__pyx_v_gini_right + (__pyx_v_tmp * __pyx_v_tmp));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":539
|
|
* gini_right += tmp * tmp
|
|
*
|
|
* gini_left = 1.0 - gini_left / (weighted_n_left * # <<<<<<<<<<<<<<
|
|
* weighted_n_left)
|
|
* gini_right = 1.0 - gini_right / (weighted_n_right *
|
|
*/
|
|
__pyx_v_gini_left = (1.0 - (__pyx_v_gini_left / (__pyx_v_weighted_n_left * __pyx_v_weighted_n_left)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":541
|
|
* gini_left = 1.0 - gini_left / (weighted_n_left *
|
|
* weighted_n_left)
|
|
* gini_right = 1.0 - gini_right / (weighted_n_right * # <<<<<<<<<<<<<<
|
|
* weighted_n_right)
|
|
*
|
|
*/
|
|
__pyx_v_gini_right = (1.0 - (__pyx_v_gini_right / (__pyx_v_weighted_n_right * __pyx_v_weighted_n_right)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":544
|
|
* weighted_n_right)
|
|
*
|
|
* total_left += gini_left # <<<<<<<<<<<<<<
|
|
* total_right += gini_right
|
|
* label_count_left += label_count_stride
|
|
*/
|
|
__pyx_v_total_left = (__pyx_v_total_left + __pyx_v_gini_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":545
|
|
*
|
|
* total_left += gini_left
|
|
* total_right += gini_right # <<<<<<<<<<<<<<
|
|
* label_count_left += label_count_stride
|
|
* label_count_right += label_count_stride
|
|
*/
|
|
__pyx_v_total_right = (__pyx_v_total_right + __pyx_v_gini_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":546
|
|
* total_left += gini_left
|
|
* total_right += gini_right
|
|
* label_count_left += label_count_stride # <<<<<<<<<<<<<<
|
|
* label_count_right += label_count_stride
|
|
*
|
|
*/
|
|
__pyx_v_label_count_left = (__pyx_v_label_count_left + __pyx_v_label_count_stride);
|
|
|
|
/* "sklearn/tree/_tree.pyx":547
|
|
* total_right += gini_right
|
|
* label_count_left += label_count_stride
|
|
* label_count_right += label_count_stride # <<<<<<<<<<<<<<
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs
|
|
*/
|
|
__pyx_v_label_count_right = (__pyx_v_label_count_right + __pyx_v_label_count_stride);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":549
|
|
* label_count_right += label_count_stride
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs # <<<<<<<<<<<<<<
|
|
* impurity_right[0] = total_right / n_outputs
|
|
*
|
|
*/
|
|
(__pyx_v_impurity_left[0]) = (__pyx_v_total_left / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":550
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs
|
|
* impurity_right[0] = total_right / n_outputs # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_impurity_right[0]) = (__pyx_v_total_right / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":505
|
|
* return total / n_outputs
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of the
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":574
|
|
* cdef double* sum_total
|
|
*
|
|
* def __cinit__(self, SIZE_t n_outputs): # <<<<<<<<<<<<<<
|
|
* # Default values
|
|
* self.y = NULL
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_n_outputs,0};
|
|
PyObject* values[1] = {0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_n_outputs)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 574; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 1) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
}
|
|
__pyx_v_n_outputs = __Pyx_PyInt_As_Py_intptr_t(values[0]); if (unlikely((__pyx_v_n_outputs == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 574; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 1, 1, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 574; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.RegressionCriterion.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self), __pyx_v_n_outputs);
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
int __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
int __pyx_t_11;
|
|
int __pyx_t_12;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":576
|
|
* def __cinit__(self, SIZE_t n_outputs):
|
|
* # Default values
|
|
* self.y = NULL # <<<<<<<<<<<<<<
|
|
* self.y_stride = 0
|
|
* self.sample_weight = NULL
|
|
*/
|
|
__pyx_v_self->__pyx_base.y = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":577
|
|
* # Default values
|
|
* self.y = NULL
|
|
* self.y_stride = 0 # <<<<<<<<<<<<<<
|
|
* self.sample_weight = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.y_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":578
|
|
* self.y = NULL
|
|
* self.y_stride = 0
|
|
* self.sample_weight = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* self.samples = NULL
|
|
*/
|
|
__pyx_v_self->__pyx_base.sample_weight = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":580
|
|
* self.sample_weight = NULL
|
|
*
|
|
* self.samples = NULL # <<<<<<<<<<<<<<
|
|
* self.start = 0
|
|
* self.pos = 0
|
|
*/
|
|
__pyx_v_self->__pyx_base.samples = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":581
|
|
*
|
|
* self.samples = NULL
|
|
* self.start = 0 # <<<<<<<<<<<<<<
|
|
* self.pos = 0
|
|
* self.end = 0
|
|
*/
|
|
__pyx_v_self->__pyx_base.start = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":582
|
|
* self.samples = NULL
|
|
* self.start = 0
|
|
* self.pos = 0 # <<<<<<<<<<<<<<
|
|
* self.end = 0
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.pos = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":583
|
|
* self.start = 0
|
|
* self.pos = 0
|
|
* self.end = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* self.n_outputs = n_outputs
|
|
*/
|
|
__pyx_v_self->__pyx_base.end = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":585
|
|
* self.end = 0
|
|
*
|
|
* self.n_outputs = n_outputs # <<<<<<<<<<<<<<
|
|
* self.n_node_samples = 0
|
|
* self.weighted_n_node_samples = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.n_outputs = __pyx_v_n_outputs;
|
|
|
|
/* "sklearn/tree/_tree.pyx":586
|
|
*
|
|
* self.n_outputs = n_outputs
|
|
* self.n_node_samples = 0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_node_samples = 0.0
|
|
* self.weighted_n_left = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.n_node_samples = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":587
|
|
* self.n_outputs = n_outputs
|
|
* self.n_node_samples = 0
|
|
* self.weighted_n_node_samples = 0.0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_left = 0.0
|
|
* self.weighted_n_right = 0.0
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_node_samples = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":588
|
|
* self.n_node_samples = 0
|
|
* self.weighted_n_node_samples = 0.0
|
|
* self.weighted_n_left = 0.0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_right = 0.0
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":589
|
|
* self.weighted_n_node_samples = 0.0
|
|
* self.weighted_n_left = 0.0
|
|
* self.weighted_n_right = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Allocate accumulators. Make sure they are NULL, not uninitialized,
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":593
|
|
* # Allocate accumulators. Make sure they are NULL, not uninitialized,
|
|
* # before an exception can be raised (which triggers __dealloc__).
|
|
* self.mean_left = NULL # <<<<<<<<<<<<<<
|
|
* self.mean_right = NULL
|
|
* self.mean_total = NULL
|
|
*/
|
|
__pyx_v_self->mean_left = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":594
|
|
* # before an exception can be raised (which triggers __dealloc__).
|
|
* self.mean_left = NULL
|
|
* self.mean_right = NULL # <<<<<<<<<<<<<<
|
|
* self.mean_total = NULL
|
|
* self.sq_sum_left = NULL
|
|
*/
|
|
__pyx_v_self->mean_right = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":595
|
|
* self.mean_left = NULL
|
|
* self.mean_right = NULL
|
|
* self.mean_total = NULL # <<<<<<<<<<<<<<
|
|
* self.sq_sum_left = NULL
|
|
* self.sq_sum_right = NULL
|
|
*/
|
|
__pyx_v_self->mean_total = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":596
|
|
* self.mean_right = NULL
|
|
* self.mean_total = NULL
|
|
* self.sq_sum_left = NULL # <<<<<<<<<<<<<<
|
|
* self.sq_sum_right = NULL
|
|
* self.sq_sum_total = NULL
|
|
*/
|
|
__pyx_v_self->sq_sum_left = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":597
|
|
* self.mean_total = NULL
|
|
* self.sq_sum_left = NULL
|
|
* self.sq_sum_right = NULL # <<<<<<<<<<<<<<
|
|
* self.sq_sum_total = NULL
|
|
* self.var_left = NULL
|
|
*/
|
|
__pyx_v_self->sq_sum_right = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":598
|
|
* self.sq_sum_left = NULL
|
|
* self.sq_sum_right = NULL
|
|
* self.sq_sum_total = NULL # <<<<<<<<<<<<<<
|
|
* self.var_left = NULL
|
|
* self.var_right = NULL
|
|
*/
|
|
__pyx_v_self->sq_sum_total = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":599
|
|
* self.sq_sum_right = NULL
|
|
* self.sq_sum_total = NULL
|
|
* self.var_left = NULL # <<<<<<<<<<<<<<
|
|
* self.var_right = NULL
|
|
* self.sum_left = NULL
|
|
*/
|
|
__pyx_v_self->var_left = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":600
|
|
* self.sq_sum_total = NULL
|
|
* self.var_left = NULL
|
|
* self.var_right = NULL # <<<<<<<<<<<<<<
|
|
* self.sum_left = NULL
|
|
* self.sum_right = NULL
|
|
*/
|
|
__pyx_v_self->var_right = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":601
|
|
* self.var_left = NULL
|
|
* self.var_right = NULL
|
|
* self.sum_left = NULL # <<<<<<<<<<<<<<
|
|
* self.sum_right = NULL
|
|
* self.sum_total = NULL
|
|
*/
|
|
__pyx_v_self->sum_left = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":602
|
|
* self.var_right = NULL
|
|
* self.sum_left = NULL
|
|
* self.sum_right = NULL # <<<<<<<<<<<<<<
|
|
* self.sum_total = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->sum_right = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":603
|
|
* self.sum_left = NULL
|
|
* self.sum_right = NULL
|
|
* self.sum_total = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* self.mean_left = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->sum_total = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":605
|
|
* self.sum_total = NULL
|
|
*
|
|
* self.mean_left = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.mean_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.mean_total = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->mean_left = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":606
|
|
*
|
|
* self.mean_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.mean_right = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.mean_total = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->mean_right = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":607
|
|
* self.mean_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.mean_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.mean_total = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.sq_sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->mean_total = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":608
|
|
* self.mean_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.mean_total = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_left = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.sq_sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->sq_sum_left = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":609
|
|
* self.mean_total = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_right = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.sq_sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.var_left = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->sq_sum_right = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":610
|
|
* self.sq_sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_total = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.var_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.var_right = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->sq_sum_total = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":611
|
|
* self.sq_sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sq_sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.var_left = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.var_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->var_left = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":612
|
|
* self.sq_sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.var_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.var_right = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->var_right = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":613
|
|
* self.var_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.var_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_left = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
*/
|
|
__pyx_v_self->sum_left = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":614
|
|
* self.var_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_right = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
* self.sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
*
|
|
*/
|
|
__pyx_v_self->sum_right = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":615
|
|
* self.sum_left = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_right = <double*> calloc(n_outputs, sizeof(double))
|
|
* self.sum_total = <double*> calloc(n_outputs, sizeof(double)) # <<<<<<<<<<<<<<
|
|
*
|
|
* if (self.mean_left == NULL or
|
|
*/
|
|
__pyx_v_self->sum_total = ((double *)calloc(__pyx_v_n_outputs, (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":617
|
|
* self.sum_total = <double*> calloc(n_outputs, sizeof(double))
|
|
*
|
|
* if (self.mean_left == NULL or # <<<<<<<<<<<<<<
|
|
* self.mean_right == NULL or
|
|
* self.mean_total == NULL or
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_self->mean_left == NULL) != 0);
|
|
if (!__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":618
|
|
*
|
|
* if (self.mean_left == NULL or
|
|
* self.mean_right == NULL or # <<<<<<<<<<<<<<
|
|
* self.mean_total == NULL or
|
|
* self.sq_sum_left == NULL or
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_self->mean_right == NULL) != 0);
|
|
if (!__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":619
|
|
* if (self.mean_left == NULL or
|
|
* self.mean_right == NULL or
|
|
* self.mean_total == NULL or # <<<<<<<<<<<<<<
|
|
* self.sq_sum_left == NULL or
|
|
* self.sq_sum_right == NULL or
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_self->mean_total == NULL) != 0);
|
|
if (!__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":620
|
|
* self.mean_right == NULL or
|
|
* self.mean_total == NULL or
|
|
* self.sq_sum_left == NULL or # <<<<<<<<<<<<<<
|
|
* self.sq_sum_right == NULL or
|
|
* self.sq_sum_total == NULL or
|
|
*/
|
|
__pyx_t_4 = ((__pyx_v_self->sq_sum_left == NULL) != 0);
|
|
if (!__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":621
|
|
* self.mean_total == NULL or
|
|
* self.sq_sum_left == NULL or
|
|
* self.sq_sum_right == NULL or # <<<<<<<<<<<<<<
|
|
* self.sq_sum_total == NULL or
|
|
* self.var_left == NULL or
|
|
*/
|
|
__pyx_t_5 = ((__pyx_v_self->sq_sum_right == NULL) != 0);
|
|
if (!__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":622
|
|
* self.sq_sum_left == NULL or
|
|
* self.sq_sum_right == NULL or
|
|
* self.sq_sum_total == NULL or # <<<<<<<<<<<<<<
|
|
* self.var_left == NULL or
|
|
* self.var_right == NULL or
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_self->sq_sum_total == NULL) != 0);
|
|
if (!__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":623
|
|
* self.sq_sum_right == NULL or
|
|
* self.sq_sum_total == NULL or
|
|
* self.var_left == NULL or # <<<<<<<<<<<<<<
|
|
* self.var_right == NULL or
|
|
* self.sum_left == NULL or
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_self->var_left == NULL) != 0);
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":624
|
|
* self.sq_sum_total == NULL or
|
|
* self.var_left == NULL or
|
|
* self.var_right == NULL or # <<<<<<<<<<<<<<
|
|
* self.sum_left == NULL or
|
|
* self.sum_right == NULL or
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_self->var_right == NULL) != 0);
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":625
|
|
* self.var_left == NULL or
|
|
* self.var_right == NULL or
|
|
* self.sum_left == NULL or # <<<<<<<<<<<<<<
|
|
* self.sum_right == NULL or
|
|
* self.sum_total == NULL):
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_self->sum_left == NULL) != 0);
|
|
if (!__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":626
|
|
* self.var_right == NULL or
|
|
* self.sum_left == NULL or
|
|
* self.sum_right == NULL or # <<<<<<<<<<<<<<
|
|
* self.sum_total == NULL):
|
|
* raise MemoryError()
|
|
*/
|
|
__pyx_t_10 = ((__pyx_v_self->sum_right == NULL) != 0);
|
|
if (!__pyx_t_10) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":627
|
|
* self.sum_left == NULL or
|
|
* self.sum_right == NULL or
|
|
* self.sum_total == NULL): # <<<<<<<<<<<<<<
|
|
* raise MemoryError()
|
|
*
|
|
*/
|
|
__pyx_t_11 = ((__pyx_v_self->sum_total == NULL) != 0);
|
|
__pyx_t_12 = __pyx_t_11;
|
|
} else {
|
|
__pyx_t_12 = __pyx_t_10;
|
|
}
|
|
__pyx_t_10 = __pyx_t_12;
|
|
} else {
|
|
__pyx_t_10 = __pyx_t_9;
|
|
}
|
|
__pyx_t_9 = __pyx_t_10;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_8;
|
|
}
|
|
__pyx_t_8 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_7;
|
|
}
|
|
__pyx_t_7 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
__pyx_t_6 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_5;
|
|
}
|
|
__pyx_t_5 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_5 = __pyx_t_4;
|
|
}
|
|
__pyx_t_4 = __pyx_t_5;
|
|
} else {
|
|
__pyx_t_4 = __pyx_t_3;
|
|
}
|
|
__pyx_t_3 = __pyx_t_4;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_2;
|
|
}
|
|
__pyx_t_2 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_1;
|
|
}
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":628
|
|
* self.sum_right == NULL or
|
|
* self.sum_total == NULL):
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
* def __dealloc__(self):
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 628; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":574
|
|
* cdef double* sum_total
|
|
*
|
|
* def __cinit__(self, SIZE_t n_outputs): # <<<<<<<<<<<<<<
|
|
* # Default values
|
|
* self.y = NULL
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.RegressionCriterion.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":630
|
|
* raise MemoryError()
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.mean_left)
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_3__dealloc__(PyObject *__pyx_v_self); /*proto*/
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_3__dealloc__(PyObject *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0);
|
|
__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_2__dealloc__(((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":632
|
|
* def __dealloc__(self):
|
|
* """Destructor."""
|
|
* free(self.mean_left) # <<<<<<<<<<<<<<
|
|
* free(self.mean_right)
|
|
* free(self.mean_total)
|
|
*/
|
|
free(__pyx_v_self->mean_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":633
|
|
* """Destructor."""
|
|
* free(self.mean_left)
|
|
* free(self.mean_right) # <<<<<<<<<<<<<<
|
|
* free(self.mean_total)
|
|
* free(self.sq_sum_left)
|
|
*/
|
|
free(__pyx_v_self->mean_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":634
|
|
* free(self.mean_left)
|
|
* free(self.mean_right)
|
|
* free(self.mean_total) # <<<<<<<<<<<<<<
|
|
* free(self.sq_sum_left)
|
|
* free(self.sq_sum_right)
|
|
*/
|
|
free(__pyx_v_self->mean_total);
|
|
|
|
/* "sklearn/tree/_tree.pyx":635
|
|
* free(self.mean_right)
|
|
* free(self.mean_total)
|
|
* free(self.sq_sum_left) # <<<<<<<<<<<<<<
|
|
* free(self.sq_sum_right)
|
|
* free(self.sq_sum_total)
|
|
*/
|
|
free(__pyx_v_self->sq_sum_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":636
|
|
* free(self.mean_total)
|
|
* free(self.sq_sum_left)
|
|
* free(self.sq_sum_right) # <<<<<<<<<<<<<<
|
|
* free(self.sq_sum_total)
|
|
* free(self.var_left)
|
|
*/
|
|
free(__pyx_v_self->sq_sum_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":637
|
|
* free(self.sq_sum_left)
|
|
* free(self.sq_sum_right)
|
|
* free(self.sq_sum_total) # <<<<<<<<<<<<<<
|
|
* free(self.var_left)
|
|
* free(self.var_right)
|
|
*/
|
|
free(__pyx_v_self->sq_sum_total);
|
|
|
|
/* "sklearn/tree/_tree.pyx":638
|
|
* free(self.sq_sum_right)
|
|
* free(self.sq_sum_total)
|
|
* free(self.var_left) # <<<<<<<<<<<<<<
|
|
* free(self.var_right)
|
|
* free(self.sum_left)
|
|
*/
|
|
free(__pyx_v_self->var_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":639
|
|
* free(self.sq_sum_total)
|
|
* free(self.var_left)
|
|
* free(self.var_right) # <<<<<<<<<<<<<<
|
|
* free(self.sum_left)
|
|
* free(self.sum_right)
|
|
*/
|
|
free(__pyx_v_self->var_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":640
|
|
* free(self.var_left)
|
|
* free(self.var_right)
|
|
* free(self.sum_left) # <<<<<<<<<<<<<<
|
|
* free(self.sum_right)
|
|
* free(self.sum_total)
|
|
*/
|
|
free(__pyx_v_self->sum_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":641
|
|
* free(self.var_right)
|
|
* free(self.sum_left)
|
|
* free(self.sum_right) # <<<<<<<<<<<<<<
|
|
* free(self.sum_total)
|
|
*
|
|
*/
|
|
free(__pyx_v_self->sum_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":642
|
|
* free(self.sum_left)
|
|
* free(self.sum_right)
|
|
* free(self.sum_total) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __reduce__(self):
|
|
*/
|
|
free(__pyx_v_self->sum_total);
|
|
|
|
/* "sklearn/tree/_tree.pyx":630
|
|
* raise MemoryError()
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.mean_left)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":644
|
|
* free(self.sum_total)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (RegressionCriterion, (self.n_outputs,), self.__getstate__())
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_4__reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":645
|
|
*
|
|
* def __reduce__(self):
|
|
* return (RegressionCriterion, (self.n_outputs,), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __getstate__(self):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.n_outputs); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 645; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 645; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 645; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 645; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyTuple_New(3); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 645; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion)));
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion)));
|
|
PyTuple_SET_ITEM(__pyx_t_1, 1, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 2, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":644
|
|
* free(self.sum_total)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (RegressionCriterion, (self.n_outputs,), self.__getstate__())
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.RegressionCriterion.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":647
|
|
* return (RegressionCriterion, (self.n_outputs,), self.__getstate__())
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* return {}
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_7__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_7__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__getstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_6__getstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_6__getstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
|
|
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("__getstate__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":648
|
|
*
|
|
* def __getstate__(self):
|
|
* return {} # <<<<<<<<<<<<<<
|
|
*
|
|
* def __setstate__(self, d):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":647
|
|
* return (RegressionCriterion, (self.n_outputs,), self.__getstate__())
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* return {}
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.RegressionCriterion.__getstate__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":650
|
|
* return {}
|
|
*
|
|
* def __setstate__(self, d): # <<<<<<<<<<<<<<
|
|
* pass
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_9__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_9__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_8__setstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self), ((PyObject *)__pyx_v_d));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_8__setstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__", 0);
|
|
|
|
/* function exit code */
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":653
|
|
* pass
|
|
*
|
|
* cdef void init(self, DOUBLE_t* y, SIZE_t y_stride, DOUBLE_t* sample_weight, # <<<<<<<<<<<<<<
|
|
* double weighted_n_samples, SIZE_t* samples, SIZE_t start,
|
|
* SIZE_t end) nogil:
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_init(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_y_stride, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight, double __pyx_v_weighted_n_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end) {
|
|
double __pyx_v_weighted_n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
double *__pyx_v_mean_left;
|
|
double *__pyx_v_mean_right;
|
|
double *__pyx_v_mean_total;
|
|
double *__pyx_v_sq_sum_left;
|
|
double *__pyx_v_sq_sum_right;
|
|
double *__pyx_v_sq_sum_total;
|
|
double *__pyx_v_var_left;
|
|
double *__pyx_v_var_right;
|
|
double *__pyx_v_sum_left;
|
|
double *__pyx_v_sum_right;
|
|
double *__pyx_v_sum_total;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_y_ik;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_w_y_ik;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_w;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_bytes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
double *__pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_3;
|
|
int __pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":659
|
|
* children samples[start:start] and samples[start:end]."""
|
|
* # Initialize fields
|
|
* self.y = y # <<<<<<<<<<<<<<
|
|
* self.y_stride = y_stride
|
|
* self.sample_weight = sample_weight
|
|
*/
|
|
__pyx_v_self->__pyx_base.y = __pyx_v_y;
|
|
|
|
/* "sklearn/tree/_tree.pyx":660
|
|
* # Initialize fields
|
|
* self.y = y
|
|
* self.y_stride = y_stride # <<<<<<<<<<<<<<
|
|
* self.sample_weight = sample_weight
|
|
* self.samples = samples
|
|
*/
|
|
__pyx_v_self->__pyx_base.y_stride = __pyx_v_y_stride;
|
|
|
|
/* "sklearn/tree/_tree.pyx":661
|
|
* self.y = y
|
|
* self.y_stride = y_stride
|
|
* self.sample_weight = sample_weight # <<<<<<<<<<<<<<
|
|
* self.samples = samples
|
|
* self.start = start
|
|
*/
|
|
__pyx_v_self->__pyx_base.sample_weight = __pyx_v_sample_weight;
|
|
|
|
/* "sklearn/tree/_tree.pyx":662
|
|
* self.y_stride = y_stride
|
|
* self.sample_weight = sample_weight
|
|
* self.samples = samples # <<<<<<<<<<<<<<
|
|
* self.start = start
|
|
* self.end = end
|
|
*/
|
|
__pyx_v_self->__pyx_base.samples = __pyx_v_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":663
|
|
* self.sample_weight = sample_weight
|
|
* self.samples = samples
|
|
* self.start = start # <<<<<<<<<<<<<<
|
|
* self.end = end
|
|
* self.n_node_samples = end - start
|
|
*/
|
|
__pyx_v_self->__pyx_base.start = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":664
|
|
* self.samples = samples
|
|
* self.start = start
|
|
* self.end = end # <<<<<<<<<<<<<<
|
|
* self.n_node_samples = end - start
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
*/
|
|
__pyx_v_self->__pyx_base.end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":665
|
|
* self.start = start
|
|
* self.end = end
|
|
* self.n_node_samples = end - start # <<<<<<<<<<<<<<
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
* cdef double weighted_n_node_samples = 0.
|
|
*/
|
|
__pyx_v_self->__pyx_base.n_node_samples = (__pyx_v_end - __pyx_v_start);
|
|
|
|
/* "sklearn/tree/_tree.pyx":666
|
|
* self.end = end
|
|
* self.n_node_samples = end - start
|
|
* self.weighted_n_samples = weighted_n_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_node_samples = 0.
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_samples = __pyx_v_weighted_n_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":667
|
|
* self.n_node_samples = end - start
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
* cdef double weighted_n_node_samples = 0. # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize accumulators
|
|
*/
|
|
__pyx_v_weighted_n_node_samples = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":670
|
|
*
|
|
* # Initialize accumulators
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":671
|
|
* # Initialize accumulators
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* mean_left = self.mean_left # <<<<<<<<<<<<<<
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* mean_total = self.mean_total
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->mean_left;
|
|
__pyx_v_mean_left = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":672
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right # <<<<<<<<<<<<<<
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->mean_right;
|
|
__pyx_v_mean_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":673
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* mean_total = self.mean_total # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->mean_total;
|
|
__pyx_v_mean_total = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":674
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double* sq_sum_left = self.sq_sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->sq_sum_left;
|
|
__pyx_v_sq_sum_left = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":675
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* var_left = self.var_left
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->sq_sum_right;
|
|
__pyx_v_sq_sum_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":676
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* sq_sum_total = self.sq_sum_total # <<<<<<<<<<<<<<
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->sq_sum_total;
|
|
__pyx_v_sq_sum_total = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":677
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* var_left = self.var_left # <<<<<<<<<<<<<<
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double* sum_left = self.sum_left
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->var_left;
|
|
__pyx_v_var_left = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":678
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right # <<<<<<<<<<<<<<
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->var_right;
|
|
__pyx_v_var_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":679
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double* sum_left = self.sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double* sum_total = self.sum_total
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->sum_left;
|
|
__pyx_v_sum_left = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":680
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right # <<<<<<<<<<<<<<
|
|
* cdef double* sum_total = self.sum_total
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->sum_right;
|
|
__pyx_v_sum_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":681
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double* sum_total = self.sum_total # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t i = 0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->sum_total;
|
|
__pyx_v_sum_total = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":683
|
|
* cdef double* sum_total = self.sum_total
|
|
*
|
|
* cdef SIZE_t i = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t p = 0
|
|
* cdef SIZE_t k = 0
|
|
*/
|
|
__pyx_v_i = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":684
|
|
*
|
|
* cdef SIZE_t i = 0
|
|
* cdef SIZE_t p = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t k = 0
|
|
* cdef DOUBLE_t y_ik = 0.0
|
|
*/
|
|
__pyx_v_p = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":685
|
|
* cdef SIZE_t i = 0
|
|
* cdef SIZE_t p = 0
|
|
* cdef SIZE_t k = 0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t y_ik = 0.0
|
|
* cdef DOUBLE_t w_y_ik = 0.0
|
|
*/
|
|
__pyx_v_k = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":686
|
|
* cdef SIZE_t p = 0
|
|
* cdef SIZE_t k = 0
|
|
* cdef DOUBLE_t y_ik = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t w_y_ik = 0.0
|
|
* cdef DOUBLE_t w = 1.0
|
|
*/
|
|
__pyx_v_y_ik = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":687
|
|
* cdef SIZE_t k = 0
|
|
* cdef DOUBLE_t y_ik = 0.0
|
|
* cdef DOUBLE_t w_y_ik = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t w = 1.0
|
|
*
|
|
*/
|
|
__pyx_v_w_y_ik = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":688
|
|
* cdef DOUBLE_t y_ik = 0.0
|
|
* cdef DOUBLE_t w_y_ik = 0.0
|
|
* cdef DOUBLE_t w = 1.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_bytes = n_outputs * sizeof(double)
|
|
*/
|
|
__pyx_v_w = 1.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":690
|
|
* cdef DOUBLE_t w = 1.0
|
|
*
|
|
* cdef SIZE_t n_bytes = n_outputs * sizeof(double) # <<<<<<<<<<<<<<
|
|
* memset(mean_left, 0, n_bytes)
|
|
* memset(mean_right, 0, n_bytes)
|
|
*/
|
|
__pyx_v_n_bytes = (__pyx_v_n_outputs * (sizeof(double)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":691
|
|
*
|
|
* cdef SIZE_t n_bytes = n_outputs * sizeof(double)
|
|
* memset(mean_left, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(mean_right, 0, n_bytes)
|
|
* memset(mean_total, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_mean_left, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":692
|
|
* cdef SIZE_t n_bytes = n_outputs * sizeof(double)
|
|
* memset(mean_left, 0, n_bytes)
|
|
* memset(mean_right, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(mean_total, 0, n_bytes)
|
|
* memset(sq_sum_left, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_mean_right, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":693
|
|
* memset(mean_left, 0, n_bytes)
|
|
* memset(mean_right, 0, n_bytes)
|
|
* memset(mean_total, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(sq_sum_left, 0, n_bytes)
|
|
* memset(sq_sum_right, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_mean_total, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":694
|
|
* memset(mean_right, 0, n_bytes)
|
|
* memset(mean_total, 0, n_bytes)
|
|
* memset(sq_sum_left, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(sq_sum_right, 0, n_bytes)
|
|
* memset(sq_sum_total, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_sq_sum_left, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":695
|
|
* memset(mean_total, 0, n_bytes)
|
|
* memset(sq_sum_left, 0, n_bytes)
|
|
* memset(sq_sum_right, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(sq_sum_total, 0, n_bytes)
|
|
* memset(var_left, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_sq_sum_right, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":696
|
|
* memset(sq_sum_left, 0, n_bytes)
|
|
* memset(sq_sum_right, 0, n_bytes)
|
|
* memset(sq_sum_total, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(var_left, 0, n_bytes)
|
|
* memset(var_right, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_sq_sum_total, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":697
|
|
* memset(sq_sum_right, 0, n_bytes)
|
|
* memset(sq_sum_total, 0, n_bytes)
|
|
* memset(var_left, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(var_right, 0, n_bytes)
|
|
* memset(sum_left, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_var_left, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":698
|
|
* memset(sq_sum_total, 0, n_bytes)
|
|
* memset(var_left, 0, n_bytes)
|
|
* memset(var_right, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(sum_left, 0, n_bytes)
|
|
* memset(sum_right, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_var_right, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":699
|
|
* memset(var_left, 0, n_bytes)
|
|
* memset(var_right, 0, n_bytes)
|
|
* memset(sum_left, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(sum_right, 0, n_bytes)
|
|
* memset(sum_total, 0, n_bytes)
|
|
*/
|
|
memset(__pyx_v_sum_left, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":700
|
|
* memset(var_right, 0, n_bytes)
|
|
* memset(sum_left, 0, n_bytes)
|
|
* memset(sum_right, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
* memset(sum_total, 0, n_bytes)
|
|
*
|
|
*/
|
|
memset(__pyx_v_sum_right, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":701
|
|
* memset(sum_left, 0, n_bytes)
|
|
* memset(sum_right, 0, n_bytes)
|
|
* memset(sum_total, 0, n_bytes) # <<<<<<<<<<<<<<
|
|
*
|
|
* for p in range(start, end):
|
|
*/
|
|
memset(__pyx_v_sum_total, 0, __pyx_v_n_bytes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":703
|
|
* memset(sum_total, 0, n_bytes)
|
|
*
|
|
* for p in range(start, end): # <<<<<<<<<<<<<<
|
|
* i = samples[p]
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_end;
|
|
for (__pyx_t_3 = __pyx_v_start; __pyx_t_3 < __pyx_t_1; __pyx_t_3+=1) {
|
|
__pyx_v_p = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":704
|
|
*
|
|
* for p in range(start, end):
|
|
* i = samples[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if sample_weight != NULL:
|
|
*/
|
|
__pyx_v_i = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":706
|
|
* i = samples[p]
|
|
*
|
|
* if sample_weight != NULL: # <<<<<<<<<<<<<<
|
|
* w = sample_weight[i]
|
|
*
|
|
*/
|
|
__pyx_t_4 = ((__pyx_v_sample_weight != NULL) != 0);
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":707
|
|
*
|
|
* if sample_weight != NULL:
|
|
* w = sample_weight[i] # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_w = (__pyx_v_sample_weight[__pyx_v_i]);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":709
|
|
* w = sample_weight[i]
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* y_ik = y[i * y_stride + k]
|
|
* w_y_ik = w * y_ik
|
|
*/
|
|
__pyx_t_5 = __pyx_v_n_outputs;
|
|
for (__pyx_t_6 = 0; __pyx_t_6 < __pyx_t_5; __pyx_t_6+=1) {
|
|
__pyx_v_k = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":710
|
|
*
|
|
* for k in range(n_outputs):
|
|
* y_ik = y[i * y_stride + k] # <<<<<<<<<<<<<<
|
|
* w_y_ik = w * y_ik
|
|
* sum_total[k] += w_y_ik
|
|
*/
|
|
__pyx_v_y_ik = (__pyx_v_y[((__pyx_v_i * __pyx_v_y_stride) + __pyx_v_k)]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":711
|
|
* for k in range(n_outputs):
|
|
* y_ik = y[i * y_stride + k]
|
|
* w_y_ik = w * y_ik # <<<<<<<<<<<<<<
|
|
* sum_total[k] += w_y_ik
|
|
* sq_sum_total[k] += w_y_ik * y_ik
|
|
*/
|
|
__pyx_v_w_y_ik = (__pyx_v_w * __pyx_v_y_ik);
|
|
|
|
/* "sklearn/tree/_tree.pyx":712
|
|
* y_ik = y[i * y_stride + k]
|
|
* w_y_ik = w * y_ik
|
|
* sum_total[k] += w_y_ik # <<<<<<<<<<<<<<
|
|
* sq_sum_total[k] += w_y_ik * y_ik
|
|
*
|
|
*/
|
|
__pyx_t_7 = __pyx_v_k;
|
|
(__pyx_v_sum_total[__pyx_t_7]) = ((__pyx_v_sum_total[__pyx_t_7]) + __pyx_v_w_y_ik);
|
|
|
|
/* "sklearn/tree/_tree.pyx":713
|
|
* w_y_ik = w * y_ik
|
|
* sum_total[k] += w_y_ik
|
|
* sq_sum_total[k] += w_y_ik * y_ik # <<<<<<<<<<<<<<
|
|
*
|
|
* weighted_n_node_samples += w
|
|
*/
|
|
__pyx_t_7 = __pyx_v_k;
|
|
(__pyx_v_sq_sum_total[__pyx_t_7]) = ((__pyx_v_sq_sum_total[__pyx_t_7]) + (__pyx_v_w_y_ik * __pyx_v_y_ik));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":715
|
|
* sq_sum_total[k] += w_y_ik * y_ik
|
|
*
|
|
* weighted_n_node_samples += w # <<<<<<<<<<<<<<
|
|
*
|
|
* self.weighted_n_node_samples = weighted_n_node_samples
|
|
*/
|
|
__pyx_v_weighted_n_node_samples = (__pyx_v_weighted_n_node_samples + __pyx_v_w);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":717
|
|
* weighted_n_node_samples += w
|
|
*
|
|
* self.weighted_n_node_samples = weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_node_samples = __pyx_v_weighted_n_node_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":719
|
|
* self.weighted_n_node_samples = weighted_n_node_samples
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* mean_total[k] = sum_total[k] / weighted_n_node_samples
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_1; __pyx_t_3+=1) {
|
|
__pyx_v_k = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":720
|
|
*
|
|
* for k in range(n_outputs):
|
|
* mean_total[k] = sum_total[k] / weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reset to pos=start
|
|
*/
|
|
(__pyx_v_mean_total[__pyx_v_k]) = ((__pyx_v_sum_total[__pyx_v_k]) / __pyx_v_weighted_n_node_samples);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":723
|
|
*
|
|
* # Reset to pos=start
|
|
* self.reset() # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void reset(self) nogil:
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.reset(((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self));
|
|
|
|
/* "sklearn/tree/_tree.pyx":653
|
|
* pass
|
|
*
|
|
* cdef void init(self, DOUBLE_t* y, SIZE_t y_stride, DOUBLE_t* sample_weight, # <<<<<<<<<<<<<<
|
|
* double weighted_n_samples, SIZE_t* samples, SIZE_t start,
|
|
* SIZE_t end) nogil:
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":725
|
|
* self.reset()
|
|
*
|
|
* cdef void reset(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Reset the criterion at pos=start."""
|
|
* self.pos = self.start
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_reset(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
double *__pyx_v_mean_left;
|
|
double *__pyx_v_mean_right;
|
|
double *__pyx_v_mean_total;
|
|
double *__pyx_v_sq_sum_left;
|
|
double *__pyx_v_sq_sum_right;
|
|
double *__pyx_v_sq_sum_total;
|
|
double *__pyx_v_var_left;
|
|
double *__pyx_v_var_right;
|
|
double __pyx_v_weighted_n_node_samples;
|
|
double *__pyx_v_sum_left;
|
|
double *__pyx_v_sum_right;
|
|
double *__pyx_v_sum_total;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
double __pyx_t_2;
|
|
double *__pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":727
|
|
* cdef void reset(self) nogil:
|
|
* """Reset the criterion at pos=start."""
|
|
* self.pos = self.start # <<<<<<<<<<<<<<
|
|
*
|
|
* self.weighted_n_left = 0.0
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.start;
|
|
__pyx_v_self->__pyx_base.pos = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":729
|
|
* self.pos = self.start
|
|
*
|
|
* self.weighted_n_left = 0.0 # <<<<<<<<<<<<<<
|
|
* self.weighted_n_right = self.weighted_n_node_samples
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":730
|
|
*
|
|
* self.weighted_n_left = 0.0
|
|
* self.weighted_n_right = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_self->__pyx_base.weighted_n_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":732
|
|
* self.weighted_n_right = self.weighted_n_node_samples
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":733
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* mean_left = self.mean_left # <<<<<<<<<<<<<<
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* mean_total = self.mean_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->mean_left;
|
|
__pyx_v_mean_left = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":734
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right # <<<<<<<<<<<<<<
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->mean_right;
|
|
__pyx_v_mean_right = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":735
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* mean_total = self.mean_total # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->mean_total;
|
|
__pyx_v_mean_total = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":736
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double* sq_sum_left = self.sq_sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->sq_sum_left;
|
|
__pyx_v_sq_sum_left = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":737
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* var_left = self.var_left
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->sq_sum_right;
|
|
__pyx_v_sq_sum_right = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":738
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* sq_sum_total = self.sq_sum_total # <<<<<<<<<<<<<<
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->sq_sum_total;
|
|
__pyx_v_sq_sum_total = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":739
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* var_left = self.var_left # <<<<<<<<<<<<<<
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->var_left;
|
|
__pyx_v_var_left = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":740
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double* sum_left = self.sum_left
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->var_right;
|
|
__pyx_v_var_right = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":741
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_weighted_n_node_samples = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":742
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double* sum_left = self.sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double* sum_total = self.sum_total
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->sum_left;
|
|
__pyx_v_sum_left = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":743
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right # <<<<<<<<<<<<<<
|
|
* cdef double* sum_total = self.sum_total
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->sum_right;
|
|
__pyx_v_sum_right = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":744
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double* sum_total = self.sum_total # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t k = 0
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->sum_total;
|
|
__pyx_v_sum_total = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":746
|
|
* cdef double* sum_total = self.sum_total
|
|
*
|
|
* cdef SIZE_t k = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_k = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":748
|
|
* cdef SIZE_t k = 0
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* mean_right[k] = mean_total[k]
|
|
* mean_left[k] = 0.0
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_4 = 0; __pyx_t_4 < __pyx_t_1; __pyx_t_4+=1) {
|
|
__pyx_v_k = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":749
|
|
*
|
|
* for k in range(n_outputs):
|
|
* mean_right[k] = mean_total[k] # <<<<<<<<<<<<<<
|
|
* mean_left[k] = 0.0
|
|
* sq_sum_right[k] = sq_sum_total[k]
|
|
*/
|
|
(__pyx_v_mean_right[__pyx_v_k]) = (__pyx_v_mean_total[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":750
|
|
* for k in range(n_outputs):
|
|
* mean_right[k] = mean_total[k]
|
|
* mean_left[k] = 0.0 # <<<<<<<<<<<<<<
|
|
* sq_sum_right[k] = sq_sum_total[k]
|
|
* sq_sum_left[k] = 0.0
|
|
*/
|
|
(__pyx_v_mean_left[__pyx_v_k]) = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":751
|
|
* mean_right[k] = mean_total[k]
|
|
* mean_left[k] = 0.0
|
|
* sq_sum_right[k] = sq_sum_total[k] # <<<<<<<<<<<<<<
|
|
* sq_sum_left[k] = 0.0
|
|
* var_right[k] = (sq_sum_right[k] / weighted_n_node_samples -
|
|
*/
|
|
(__pyx_v_sq_sum_right[__pyx_v_k]) = (__pyx_v_sq_sum_total[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":752
|
|
* mean_left[k] = 0.0
|
|
* sq_sum_right[k] = sq_sum_total[k]
|
|
* sq_sum_left[k] = 0.0 # <<<<<<<<<<<<<<
|
|
* var_right[k] = (sq_sum_right[k] / weighted_n_node_samples -
|
|
* mean_right[k] * mean_right[k])
|
|
*/
|
|
(__pyx_v_sq_sum_left[__pyx_v_k]) = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":753
|
|
* sq_sum_right[k] = sq_sum_total[k]
|
|
* sq_sum_left[k] = 0.0
|
|
* var_right[k] = (sq_sum_right[k] / weighted_n_node_samples - # <<<<<<<<<<<<<<
|
|
* mean_right[k] * mean_right[k])
|
|
* var_left[k] = 0.0
|
|
*/
|
|
(__pyx_v_var_right[__pyx_v_k]) = (((__pyx_v_sq_sum_right[__pyx_v_k]) / __pyx_v_weighted_n_node_samples) - ((__pyx_v_mean_right[__pyx_v_k]) * (__pyx_v_mean_right[__pyx_v_k])));
|
|
|
|
/* "sklearn/tree/_tree.pyx":755
|
|
* var_right[k] = (sq_sum_right[k] / weighted_n_node_samples -
|
|
* mean_right[k] * mean_right[k])
|
|
* var_left[k] = 0.0 # <<<<<<<<<<<<<<
|
|
* sum_right[k] = sum_total[k]
|
|
* sum_left[k] = 0.0
|
|
*/
|
|
(__pyx_v_var_left[__pyx_v_k]) = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":756
|
|
* mean_right[k] * mean_right[k])
|
|
* var_left[k] = 0.0
|
|
* sum_right[k] = sum_total[k] # <<<<<<<<<<<<<<
|
|
* sum_left[k] = 0.0
|
|
*
|
|
*/
|
|
(__pyx_v_sum_right[__pyx_v_k]) = (__pyx_v_sum_total[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":757
|
|
* var_left[k] = 0.0
|
|
* sum_right[k] = sum_total[k]
|
|
* sum_left[k] = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil:
|
|
*/
|
|
(__pyx_v_sum_left[__pyx_v_k]) = 0.0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":725
|
|
* self.reset()
|
|
*
|
|
* cdef void reset(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Reset the criterion at pos=start."""
|
|
* self.pos = self.start
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":759
|
|
* sum_left[k] = 0.0
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil: # <<<<<<<<<<<<<<
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_update(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_new_pos) {
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_y;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_y_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_pos;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
double *__pyx_v_mean_left;
|
|
double *__pyx_v_mean_right;
|
|
double *__pyx_v_sq_sum_left;
|
|
double *__pyx_v_sq_sum_right;
|
|
double *__pyx_v_var_left;
|
|
double *__pyx_v_var_right;
|
|
double *__pyx_v_sum_left;
|
|
double *__pyx_v_sum_right;
|
|
double __pyx_v_weighted_n_left;
|
|
double __pyx_v_weighted_n_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_w;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_diff_w;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_y_ik;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t __pyx_v_w_y_ik;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_3;
|
|
double *__pyx_t_4;
|
|
double __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
int __pyx_t_7;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":762
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
* cdef DOUBLE_t* y = self.y # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t y_stride = self.y_stride
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.y;
|
|
__pyx_v_y = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":763
|
|
* the right child to the left child."""
|
|
* cdef DOUBLE_t* y = self.y
|
|
* cdef SIZE_t y_stride = self.y_stride # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.y_stride;
|
|
__pyx_v_y_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":764
|
|
* cdef DOUBLE_t* y = self.y
|
|
* cdef SIZE_t y_stride = self.y_stride
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.sample_weight;
|
|
__pyx_v_sample_weight = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":766
|
|
* cdef DOUBLE_t* sample_weight = self.sample_weight
|
|
*
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t pos = self.pos
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":767
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t pos = self.pos # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.pos;
|
|
__pyx_v_pos = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":769
|
|
* cdef SIZE_t pos = self.pos
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":770
|
|
*
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* mean_left = self.mean_left # <<<<<<<<<<<<<<
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->mean_left;
|
|
__pyx_v_mean_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":771
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->mean_right;
|
|
__pyx_v_mean_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":772
|
|
* cdef double* mean_left = self.mean_left
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* sq_sum_left = self.sq_sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* var_left = self.var_left
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->sq_sum_left;
|
|
__pyx_v_sq_sum_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":773
|
|
* cdef double* mean_right = self.mean_right
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right # <<<<<<<<<<<<<<
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->sq_sum_right;
|
|
__pyx_v_sq_sum_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":774
|
|
* cdef double* sq_sum_left = self.sq_sum_left
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* var_left = self.var_left # <<<<<<<<<<<<<<
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double* sum_left = self.sum_left
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->var_left;
|
|
__pyx_v_var_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":775
|
|
* cdef double* sq_sum_right = self.sq_sum_right
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right # <<<<<<<<<<<<<<
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->var_right;
|
|
__pyx_v_var_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":776
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double* sum_left = self.sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sum_right = self.sum_right
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->sum_left;
|
|
__pyx_v_sum_left = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":777
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->sum_right;
|
|
__pyx_v_sum_right = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":779
|
|
* cdef double* sum_right = self.sum_right
|
|
*
|
|
* cdef double weighted_n_left = self.weighted_n_left # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_self->__pyx_base.weighted_n_left;
|
|
__pyx_v_weighted_n_left = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":780
|
|
*
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t i
|
|
*/
|
|
__pyx_t_5 = __pyx_v_self->__pyx_base.weighted_n_right;
|
|
__pyx_v_weighted_n_right = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":785
|
|
* cdef SIZE_t p
|
|
* cdef SIZE_t k
|
|
* cdef DOUBLE_t w = 1.0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t diff_w = 0.0
|
|
* cdef DOUBLE_t y_ik, w_y_ik
|
|
*/
|
|
__pyx_v_w = 1.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":786
|
|
* cdef SIZE_t k
|
|
* cdef DOUBLE_t w = 1.0
|
|
* cdef DOUBLE_t diff_w = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t y_ik, w_y_ik
|
|
*
|
|
*/
|
|
__pyx_v_diff_w = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":790
|
|
*
|
|
* # Note: We assume start <= pos < new_pos <= end
|
|
* for p in range(pos, new_pos): # <<<<<<<<<<<<<<
|
|
* i = samples[p]
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_new_pos;
|
|
for (__pyx_t_6 = __pyx_v_pos; __pyx_t_6 < __pyx_t_2; __pyx_t_6+=1) {
|
|
__pyx_v_p = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":791
|
|
* # Note: We assume start <= pos < new_pos <= end
|
|
* for p in range(pos, new_pos):
|
|
* i = samples[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if sample_weight != NULL:
|
|
*/
|
|
__pyx_v_i = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":793
|
|
* i = samples[p]
|
|
*
|
|
* if sample_weight != NULL: # <<<<<<<<<<<<<<
|
|
* w = sample_weight[i]
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_sample_weight != NULL) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":794
|
|
*
|
|
* if sample_weight != NULL:
|
|
* w = sample_weight[i] # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_w = (__pyx_v_sample_weight[__pyx_v_i]);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":796
|
|
* w = sample_weight[i]
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* y_ik = y[i * y_stride + k]
|
|
* w_y_ik = w * y_ik
|
|
*/
|
|
__pyx_t_8 = __pyx_v_n_outputs;
|
|
for (__pyx_t_9 = 0; __pyx_t_9 < __pyx_t_8; __pyx_t_9+=1) {
|
|
__pyx_v_k = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":797
|
|
*
|
|
* for k in range(n_outputs):
|
|
* y_ik = y[i * y_stride + k] # <<<<<<<<<<<<<<
|
|
* w_y_ik = w * y_ik
|
|
*
|
|
*/
|
|
__pyx_v_y_ik = (__pyx_v_y[((__pyx_v_i * __pyx_v_y_stride) + __pyx_v_k)]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":798
|
|
* for k in range(n_outputs):
|
|
* y_ik = y[i * y_stride + k]
|
|
* w_y_ik = w * y_ik # <<<<<<<<<<<<<<
|
|
*
|
|
* sum_left[k] += w_y_ik
|
|
*/
|
|
__pyx_v_w_y_ik = (__pyx_v_w * __pyx_v_y_ik);
|
|
|
|
/* "sklearn/tree/_tree.pyx":800
|
|
* w_y_ik = w * y_ik
|
|
*
|
|
* sum_left[k] += w_y_ik # <<<<<<<<<<<<<<
|
|
* sum_right[k] -= w_y_ik
|
|
*
|
|
*/
|
|
__pyx_t_10 = __pyx_v_k;
|
|
(__pyx_v_sum_left[__pyx_t_10]) = ((__pyx_v_sum_left[__pyx_t_10]) + __pyx_v_w_y_ik);
|
|
|
|
/* "sklearn/tree/_tree.pyx":801
|
|
*
|
|
* sum_left[k] += w_y_ik
|
|
* sum_right[k] -= w_y_ik # <<<<<<<<<<<<<<
|
|
*
|
|
* sq_sum_left[k] += w_y_ik * y_ik
|
|
*/
|
|
__pyx_t_10 = __pyx_v_k;
|
|
(__pyx_v_sum_right[__pyx_t_10]) = ((__pyx_v_sum_right[__pyx_t_10]) - __pyx_v_w_y_ik);
|
|
|
|
/* "sklearn/tree/_tree.pyx":803
|
|
* sum_right[k] -= w_y_ik
|
|
*
|
|
* sq_sum_left[k] += w_y_ik * y_ik # <<<<<<<<<<<<<<
|
|
* sq_sum_right[k] -= w_y_ik * y_ik
|
|
*
|
|
*/
|
|
__pyx_t_10 = __pyx_v_k;
|
|
(__pyx_v_sq_sum_left[__pyx_t_10]) = ((__pyx_v_sq_sum_left[__pyx_t_10]) + (__pyx_v_w_y_ik * __pyx_v_y_ik));
|
|
|
|
/* "sklearn/tree/_tree.pyx":804
|
|
*
|
|
* sq_sum_left[k] += w_y_ik * y_ik
|
|
* sq_sum_right[k] -= w_y_ik * y_ik # <<<<<<<<<<<<<<
|
|
*
|
|
* diff_w += w
|
|
*/
|
|
__pyx_t_10 = __pyx_v_k;
|
|
(__pyx_v_sq_sum_right[__pyx_t_10]) = ((__pyx_v_sq_sum_right[__pyx_t_10]) - (__pyx_v_w_y_ik * __pyx_v_y_ik));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":806
|
|
* sq_sum_right[k] -= w_y_ik * y_ik
|
|
*
|
|
* diff_w += w # <<<<<<<<<<<<<<
|
|
*
|
|
* weighted_n_left += diff_w
|
|
*/
|
|
__pyx_v_diff_w = (__pyx_v_diff_w + __pyx_v_w);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":808
|
|
* diff_w += w
|
|
*
|
|
* weighted_n_left += diff_w # <<<<<<<<<<<<<<
|
|
* weighted_n_right -= diff_w
|
|
*
|
|
*/
|
|
__pyx_v_weighted_n_left = (__pyx_v_weighted_n_left + __pyx_v_diff_w);
|
|
|
|
/* "sklearn/tree/_tree.pyx":809
|
|
*
|
|
* weighted_n_left += diff_w
|
|
* weighted_n_right -= diff_w # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(n_outputs):
|
|
*/
|
|
__pyx_v_weighted_n_right = (__pyx_v_weighted_n_right - __pyx_v_diff_w);
|
|
|
|
/* "sklearn/tree/_tree.pyx":811
|
|
* weighted_n_right -= diff_w
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* mean_left[k] = sum_left[k] / weighted_n_left
|
|
* mean_right[k] = sum_right[k] / weighted_n_right
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_outputs;
|
|
for (__pyx_t_6 = 0; __pyx_t_6 < __pyx_t_2; __pyx_t_6+=1) {
|
|
__pyx_v_k = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":812
|
|
*
|
|
* for k in range(n_outputs):
|
|
* mean_left[k] = sum_left[k] / weighted_n_left # <<<<<<<<<<<<<<
|
|
* mean_right[k] = sum_right[k] / weighted_n_right
|
|
* var_left[k] = (sq_sum_left[k] / weighted_n_left -
|
|
*/
|
|
(__pyx_v_mean_left[__pyx_v_k]) = ((__pyx_v_sum_left[__pyx_v_k]) / __pyx_v_weighted_n_left);
|
|
|
|
/* "sklearn/tree/_tree.pyx":813
|
|
* for k in range(n_outputs):
|
|
* mean_left[k] = sum_left[k] / weighted_n_left
|
|
* mean_right[k] = sum_right[k] / weighted_n_right # <<<<<<<<<<<<<<
|
|
* var_left[k] = (sq_sum_left[k] / weighted_n_left -
|
|
* mean_left[k] * mean_left[k])
|
|
*/
|
|
(__pyx_v_mean_right[__pyx_v_k]) = ((__pyx_v_sum_right[__pyx_v_k]) / __pyx_v_weighted_n_right);
|
|
|
|
/* "sklearn/tree/_tree.pyx":814
|
|
* mean_left[k] = sum_left[k] / weighted_n_left
|
|
* mean_right[k] = sum_right[k] / weighted_n_right
|
|
* var_left[k] = (sq_sum_left[k] / weighted_n_left - # <<<<<<<<<<<<<<
|
|
* mean_left[k] * mean_left[k])
|
|
* var_right[k] = (sq_sum_right[k] / weighted_n_right -
|
|
*/
|
|
(__pyx_v_var_left[__pyx_v_k]) = (((__pyx_v_sq_sum_left[__pyx_v_k]) / __pyx_v_weighted_n_left) - ((__pyx_v_mean_left[__pyx_v_k]) * (__pyx_v_mean_left[__pyx_v_k])));
|
|
|
|
/* "sklearn/tree/_tree.pyx":816
|
|
* var_left[k] = (sq_sum_left[k] / weighted_n_left -
|
|
* mean_left[k] * mean_left[k])
|
|
* var_right[k] = (sq_sum_right[k] / weighted_n_right - # <<<<<<<<<<<<<<
|
|
* mean_right[k] * mean_right[k])
|
|
*
|
|
*/
|
|
(__pyx_v_var_right[__pyx_v_k]) = (((__pyx_v_sq_sum_right[__pyx_v_k]) / __pyx_v_weighted_n_right) - ((__pyx_v_mean_right[__pyx_v_k]) * (__pyx_v_mean_right[__pyx_v_k])));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":819
|
|
* mean_right[k] * mean_right[k])
|
|
*
|
|
* self.weighted_n_left = weighted_n_left # <<<<<<<<<<<<<<
|
|
* self.weighted_n_right = weighted_n_right
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_left = __pyx_v_weighted_n_left;
|
|
|
|
/* "sklearn/tree/_tree.pyx":820
|
|
*
|
|
* self.weighted_n_left = weighted_n_left
|
|
* self.weighted_n_right = weighted_n_right # <<<<<<<<<<<<<<
|
|
*
|
|
* self.pos = new_pos
|
|
*/
|
|
__pyx_v_self->__pyx_base.weighted_n_right = __pyx_v_weighted_n_right;
|
|
|
|
/* "sklearn/tree/_tree.pyx":822
|
|
* self.weighted_n_right = weighted_n_right
|
|
*
|
|
* self.pos = new_pos # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double node_impurity(self) nogil:
|
|
*/
|
|
__pyx_v_self->__pyx_base.pos = __pyx_v_new_pos;
|
|
|
|
/* "sklearn/tree/_tree.pyx":759
|
|
* sum_left[k] = 0.0
|
|
*
|
|
* cdef void update(self, SIZE_t new_pos) nogil: # <<<<<<<<<<<<<<
|
|
* """Update the collected statistics by moving samples[pos:new_pos] from
|
|
* the right child to the left child."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":824
|
|
* self.pos = new_pos
|
|
*
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* pass
|
|
*
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_node_impurity(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
|
|
double __pyx_r;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":827
|
|
* pass
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* pass
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_children_impurity(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, CYTHON_UNUSED double *__pyx_v_impurity_left, CYTHON_UNUSED double *__pyx_v_impurity_right) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":831
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* memcpy(dest, self.mean_total, self.n_outputs * sizeof(double))
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_node_value(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, double *__pyx_v_dest) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":833
|
|
* cdef void node_value(self, double* dest) nogil:
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* memcpy(dest, self.mean_total, self.n_outputs * sizeof(double)) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
memcpy(__pyx_v_dest, __pyx_v_self->mean_total, (__pyx_v_self->__pyx_base.n_outputs * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":831
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Compute the node value of samples[start:end] into dest."""
|
|
* memcpy(dest, self.mean_total, self.n_outputs * sizeof(double))
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":841
|
|
* MSE = var_left + var_right
|
|
* """
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_3MSE_node_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_MSE *__pyx_v_self) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
double *__pyx_v_sq_sum_total;
|
|
double *__pyx_v_mean_total;
|
|
double __pyx_v_weighted_n_node_samples;
|
|
double __pyx_v_total;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
double __pyx_r;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
double *__pyx_t_2;
|
|
double __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":844
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* mean_total = self.mean_total
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":845
|
|
* samples[start:end]."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* sq_sum_total = self.sq_sum_total # <<<<<<<<<<<<<<
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.sq_sum_total;
|
|
__pyx_v_sq_sum_total = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":846
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* mean_total = self.mean_total # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double total = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.mean_total;
|
|
__pyx_v_mean_total = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":847
|
|
* cdef double* sq_sum_total = self.sq_sum_total
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
* cdef double total = 0.0
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.__pyx_base.weighted_n_node_samples;
|
|
__pyx_v_weighted_n_node_samples = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":848
|
|
* cdef double* mean_total = self.mean_total
|
|
* cdef double weighted_n_node_samples = self.weighted_n_node_samples
|
|
* cdef double total = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t k
|
|
*
|
|
*/
|
|
__pyx_v_total = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":851
|
|
* cdef SIZE_t k
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* total += (sq_sum_total[k] / weighted_n_node_samples -
|
|
* mean_total[k] * mean_total[k])
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_4 = 0; __pyx_t_4 < __pyx_t_1; __pyx_t_4+=1) {
|
|
__pyx_v_k = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":852
|
|
*
|
|
* for k in range(n_outputs):
|
|
* total += (sq_sum_total[k] / weighted_n_node_samples - # <<<<<<<<<<<<<<
|
|
* mean_total[k] * mean_total[k])
|
|
*
|
|
*/
|
|
__pyx_v_total = (__pyx_v_total + (((__pyx_v_sq_sum_total[__pyx_v_k]) / __pyx_v_weighted_n_node_samples) - ((__pyx_v_mean_total[__pyx_v_k]) * (__pyx_v_mean_total[__pyx_v_k]))));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":855
|
|
* mean_total[k] * mean_total[k])
|
|
*
|
|
* return total / n_outputs # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left,
|
|
*/
|
|
__pyx_r = (__pyx_v_total / __pyx_v_n_outputs);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":841
|
|
* MSE = var_left + var_right
|
|
* """
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Evaluate the impurity of the current node, i.e. the impurity of
|
|
* samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":857
|
|
* return total / n_outputs
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of the
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_3MSE_children_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_MSE *__pyx_v_self, double *__pyx_v_impurity_left, double *__pyx_v_impurity_right) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
double *__pyx_v_var_left;
|
|
double *__pyx_v_var_right;
|
|
double __pyx_v_total_left;
|
|
double __pyx_v_total_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
double *__pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":862
|
|
* left child (samples[start:pos]) and the impurity the right child
|
|
* (samples[pos:end])."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":863
|
|
* (samples[pos:end])."""
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* var_left = self.var_left # <<<<<<<<<<<<<<
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double total_left = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.var_left;
|
|
__pyx_v_var_left = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":864
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right # <<<<<<<<<<<<<<
|
|
* cdef double total_left = 0.0
|
|
* cdef double total_right = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.var_right;
|
|
__pyx_v_var_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":865
|
|
* cdef double* var_left = self.var_left
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double total_left = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total_right = 0.0
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_total_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":866
|
|
* cdef double* var_right = self.var_right
|
|
* cdef double total_left = 0.0
|
|
* cdef double total_right = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t k
|
|
*
|
|
*/
|
|
__pyx_v_total_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":869
|
|
* cdef SIZE_t k
|
|
*
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* total_left += var_left[k]
|
|
* total_right += var_right[k]
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_1; __pyx_t_3+=1) {
|
|
__pyx_v_k = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":870
|
|
*
|
|
* for k in range(n_outputs):
|
|
* total_left += var_left[k] # <<<<<<<<<<<<<<
|
|
* total_right += var_right[k]
|
|
*
|
|
*/
|
|
__pyx_v_total_left = (__pyx_v_total_left + (__pyx_v_var_left[__pyx_v_k]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":871
|
|
* for k in range(n_outputs):
|
|
* total_left += var_left[k]
|
|
* total_right += var_right[k] # <<<<<<<<<<<<<<
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs
|
|
*/
|
|
__pyx_v_total_right = (__pyx_v_total_right + (__pyx_v_var_right[__pyx_v_k]));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":873
|
|
* total_right += var_right[k]
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs # <<<<<<<<<<<<<<
|
|
* impurity_right[0] = total_right / n_outputs
|
|
*
|
|
*/
|
|
(__pyx_v_impurity_left[0]) = (__pyx_v_total_left / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":874
|
|
*
|
|
* impurity_left[0] = total_left / n_outputs
|
|
* impurity_right[0] = total_right / n_outputs # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_impurity_right[0]) = (__pyx_v_total_right / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":857
|
|
* return total / n_outputs
|
|
*
|
|
* cdef void children_impurity(self, double* impurity_left, # <<<<<<<<<<<<<<
|
|
* double* impurity_right) nogil:
|
|
* """Evaluate the impurity in children nodes, i.e. the impurity of the
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":886
|
|
* """
|
|
*
|
|
* cdef double impurity_improvement(self, double impurity) nogil: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t k
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_11FriedmanMSE_impurity_improvement(struct __pyx_obj_7sklearn_4tree_5_tree_FriedmanMSE *__pyx_v_self, CYTHON_UNUSED double __pyx_v_impurity) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_outputs;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
double *__pyx_v_sum_left;
|
|
double *__pyx_v_sum_right;
|
|
double __pyx_v_total_sum_left;
|
|
double __pyx_v_total_sum_right;
|
|
double __pyx_v_weighted_n_left;
|
|
double __pyx_v_weighted_n_right;
|
|
double __pyx_v_diff;
|
|
double __pyx_r;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
double *__pyx_t_2;
|
|
double __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":887
|
|
*
|
|
* cdef double impurity_improvement(self, double impurity) nogil:
|
|
* cdef SIZE_t n_outputs = self.n_outputs # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t k
|
|
* cdef double* sum_left = self.sum_left
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.__pyx_base.n_outputs;
|
|
__pyx_v_n_outputs = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":889
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t k
|
|
* cdef double* sum_left = self.sum_left # <<<<<<<<<<<<<<
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double total_sum_left = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.sum_left;
|
|
__pyx_v_sum_left = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":890
|
|
* cdef SIZE_t k
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right # <<<<<<<<<<<<<<
|
|
* cdef double total_sum_left = 0.0
|
|
* cdef double total_sum_right = 0.0
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.sum_right;
|
|
__pyx_v_sum_right = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":891
|
|
* cdef double* sum_left = self.sum_left
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double total_sum_left = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double total_sum_right = 0.0
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
*/
|
|
__pyx_v_total_sum_left = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":892
|
|
* cdef double* sum_right = self.sum_right
|
|
* cdef double total_sum_left = 0.0
|
|
* cdef double total_sum_right = 0.0 # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
*/
|
|
__pyx_v_total_sum_right = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":893
|
|
* cdef double total_sum_left = 0.0
|
|
* cdef double total_sum_right = 0.0
|
|
* cdef double weighted_n_left = self.weighted_n_left # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
* cdef double diff = 0.0
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.__pyx_base.__pyx_base.weighted_n_left;
|
|
__pyx_v_weighted_n_left = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":894
|
|
* cdef double total_sum_right = 0.0
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right # <<<<<<<<<<<<<<
|
|
* cdef double diff = 0.0
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.__pyx_base.__pyx_base.weighted_n_right;
|
|
__pyx_v_weighted_n_right = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":895
|
|
* cdef double weighted_n_left = self.weighted_n_left
|
|
* cdef double weighted_n_right = self.weighted_n_right
|
|
* cdef double diff = 0.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for k from 0 <= k < n_outputs:
|
|
*/
|
|
__pyx_v_diff = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":897
|
|
* cdef double diff = 0.0
|
|
*
|
|
* for k from 0 <= k < n_outputs: # <<<<<<<<<<<<<<
|
|
* total_sum_left += sum_left[k]
|
|
* total_sum_right += sum_right[k]
|
|
*/
|
|
__pyx_t_1 = __pyx_v_n_outputs;
|
|
for (__pyx_v_k = 0; __pyx_v_k < __pyx_t_1; __pyx_v_k++) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":898
|
|
*
|
|
* for k from 0 <= k < n_outputs:
|
|
* total_sum_left += sum_left[k] # <<<<<<<<<<<<<<
|
|
* total_sum_right += sum_right[k]
|
|
*
|
|
*/
|
|
__pyx_v_total_sum_left = (__pyx_v_total_sum_left + (__pyx_v_sum_left[__pyx_v_k]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":899
|
|
* for k from 0 <= k < n_outputs:
|
|
* total_sum_left += sum_left[k]
|
|
* total_sum_right += sum_right[k] # <<<<<<<<<<<<<<
|
|
*
|
|
* total_sum_left = total_sum_left / n_outputs
|
|
*/
|
|
__pyx_v_total_sum_right = (__pyx_v_total_sum_right + (__pyx_v_sum_right[__pyx_v_k]));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":901
|
|
* total_sum_right += sum_right[k]
|
|
*
|
|
* total_sum_left = total_sum_left / n_outputs # <<<<<<<<<<<<<<
|
|
* total_sum_right = total_sum_right / n_outputs
|
|
* diff = ((total_sum_left / weighted_n_left) -
|
|
*/
|
|
__pyx_v_total_sum_left = (__pyx_v_total_sum_left / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":902
|
|
*
|
|
* total_sum_left = total_sum_left / n_outputs
|
|
* total_sum_right = total_sum_right / n_outputs # <<<<<<<<<<<<<<
|
|
* diff = ((total_sum_left / weighted_n_left) -
|
|
* (total_sum_right / weighted_n_right))
|
|
*/
|
|
__pyx_v_total_sum_right = (__pyx_v_total_sum_right / __pyx_v_n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":903
|
|
* total_sum_left = total_sum_left / n_outputs
|
|
* total_sum_right = total_sum_right / n_outputs
|
|
* diff = ((total_sum_left / weighted_n_left) - # <<<<<<<<<<<<<<
|
|
* (total_sum_right / weighted_n_right))
|
|
*
|
|
*/
|
|
__pyx_v_diff = ((__pyx_v_total_sum_left / __pyx_v_weighted_n_left) - (__pyx_v_total_sum_right / __pyx_v_weighted_n_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":906
|
|
* (total_sum_right / weighted_n_right))
|
|
*
|
|
* return (weighted_n_left * weighted_n_right * diff * diff / # <<<<<<<<<<<<<<
|
|
* (weighted_n_left + weighted_n_right))
|
|
*
|
|
*/
|
|
__pyx_r = ((((__pyx_v_weighted_n_left * __pyx_v_weighted_n_right) * __pyx_v_diff) * __pyx_v_diff) / (__pyx_v_weighted_n_left + __pyx_v_weighted_n_right));
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":886
|
|
* """
|
|
*
|
|
* cdef double impurity_improvement(self, double impurity) nogil: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_outputs = self.n_outputs
|
|
* cdef SIZE_t k
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":913
|
|
* # =============================================================================
|
|
*
|
|
* cdef inline void _init_split(SplitRecord* self, SIZE_t start_pos) nogil: # <<<<<<<<<<<<<<
|
|
* self.impurity_left = INFINITY
|
|
* self.impurity_right = INFINITY
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree__init_split(struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start_pos) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":914
|
|
*
|
|
* cdef inline void _init_split(SplitRecord* self, SIZE_t start_pos) nogil:
|
|
* self.impurity_left = INFINITY # <<<<<<<<<<<<<<
|
|
* self.impurity_right = INFINITY
|
|
* self.pos = start_pos
|
|
*/
|
|
__pyx_v_self->impurity_left = __pyx_v_7sklearn_4tree_5_tree_INFINITY;
|
|
|
|
/* "sklearn/tree/_tree.pyx":915
|
|
* cdef inline void _init_split(SplitRecord* self, SIZE_t start_pos) nogil:
|
|
* self.impurity_left = INFINITY
|
|
* self.impurity_right = INFINITY # <<<<<<<<<<<<<<
|
|
* self.pos = start_pos
|
|
* self.feature = 0
|
|
*/
|
|
__pyx_v_self->impurity_right = __pyx_v_7sklearn_4tree_5_tree_INFINITY;
|
|
|
|
/* "sklearn/tree/_tree.pyx":916
|
|
* self.impurity_left = INFINITY
|
|
* self.impurity_right = INFINITY
|
|
* self.pos = start_pos # <<<<<<<<<<<<<<
|
|
* self.feature = 0
|
|
* self.threshold = 0.
|
|
*/
|
|
__pyx_v_self->pos = __pyx_v_start_pos;
|
|
|
|
/* "sklearn/tree/_tree.pyx":917
|
|
* self.impurity_right = INFINITY
|
|
* self.pos = start_pos
|
|
* self.feature = 0 # <<<<<<<<<<<<<<
|
|
* self.threshold = 0.
|
|
* self.improvement = -INFINITY
|
|
*/
|
|
__pyx_v_self->feature = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":918
|
|
* self.pos = start_pos
|
|
* self.feature = 0
|
|
* self.threshold = 0. # <<<<<<<<<<<<<<
|
|
* self.improvement = -INFINITY
|
|
*
|
|
*/
|
|
__pyx_v_self->threshold = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":919
|
|
* self.feature = 0
|
|
* self.threshold = 0.
|
|
* self.improvement = -INFINITY # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_v_self->improvement = (-__pyx_v_7sklearn_4tree_5_tree_INFINITY);
|
|
|
|
/* "sklearn/tree/_tree.pyx":913
|
|
* # =============================================================================
|
|
*
|
|
* cdef inline void _init_split(SplitRecord* self, SIZE_t start_pos) nogil: # <<<<<<<<<<<<<<
|
|
* self.impurity_left = INFINITY
|
|
* self.impurity_right = INFINITY
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":923
|
|
*
|
|
* cdef class Splitter:
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
PyObject *__pyx_v_random_state = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_criterion,&__pyx_n_s_max_features,&__pyx_n_s_min_samples_leaf,&__pyx_n_s_min_weight_leaf,&__pyx_n_s_random_state,0};
|
|
PyObject* values[5] = {0,0,0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_criterion)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_features)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (likely((values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_weight_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 4:
|
|
if (likely((values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_random_state)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 5) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
}
|
|
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[0]);
|
|
__pyx_v_max_features = __Pyx_PyInt_As_Py_intptr_t(values[1]); if (unlikely((__pyx_v_max_features == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_samples_leaf = __Pyx_PyInt_As_Py_intptr_t(values[2]); if (unlikely((__pyx_v_min_samples_leaf == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 924; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_weight_leaf = __pyx_PyFloat_AsDouble(values[3]); if (unlikely((__pyx_v_min_weight_leaf == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 924; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_random_state = values[4];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_criterion), __pyx_ptype_7sklearn_4tree_5_tree_Criterion, 1, "criterion", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 923; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), __pyx_v_criterion, __pyx_v_max_features, __pyx_v_min_samples_leaf, __pyx_v_min_weight_leaf, __pyx_v_random_state);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, double __pyx_v_min_weight_leaf, PyObject *__pyx_v_random_state) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":926
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
* self.criterion = criterion # <<<<<<<<<<<<<<
|
|
*
|
|
* self.samples = NULL
|
|
*/
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_criterion));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_criterion));
|
|
__Pyx_GOTREF(__pyx_v_self->criterion);
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_self->criterion));
|
|
__pyx_v_self->criterion = __pyx_v_criterion;
|
|
|
|
/* "sklearn/tree/_tree.pyx":928
|
|
* self.criterion = criterion
|
|
*
|
|
* self.samples = NULL # <<<<<<<<<<<<<<
|
|
* self.n_samples = 0
|
|
* self.features = NULL
|
|
*/
|
|
__pyx_v_self->samples = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":929
|
|
*
|
|
* self.samples = NULL
|
|
* self.n_samples = 0 # <<<<<<<<<<<<<<
|
|
* self.features = NULL
|
|
* self.n_features = 0
|
|
*/
|
|
__pyx_v_self->n_samples = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":930
|
|
* self.samples = NULL
|
|
* self.n_samples = 0
|
|
* self.features = NULL # <<<<<<<<<<<<<<
|
|
* self.n_features = 0
|
|
* self.feature_values = NULL
|
|
*/
|
|
__pyx_v_self->features = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":931
|
|
* self.n_samples = 0
|
|
* self.features = NULL
|
|
* self.n_features = 0 # <<<<<<<<<<<<<<
|
|
* self.feature_values = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->n_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":932
|
|
* self.features = NULL
|
|
* self.n_features = 0
|
|
* self.feature_values = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* self.y = NULL
|
|
*/
|
|
__pyx_v_self->feature_values = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":934
|
|
* self.feature_values = NULL
|
|
*
|
|
* self.y = NULL # <<<<<<<<<<<<<<
|
|
* self.y_stride = 0
|
|
* self.sample_weight = NULL
|
|
*/
|
|
__pyx_v_self->y = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":935
|
|
*
|
|
* self.y = NULL
|
|
* self.y_stride = 0 # <<<<<<<<<<<<<<
|
|
* self.sample_weight = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->y_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":936
|
|
* self.y = NULL
|
|
* self.y_stride = 0
|
|
* self.sample_weight = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* self.max_features = max_features
|
|
*/
|
|
__pyx_v_self->sample_weight = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":938
|
|
* self.sample_weight = NULL
|
|
*
|
|
* self.max_features = max_features # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
*/
|
|
__pyx_v_self->max_features = __pyx_v_max_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":939
|
|
*
|
|
* self.max_features = max_features
|
|
* self.min_samples_leaf = min_samples_leaf # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.random_state = random_state
|
|
*/
|
|
__pyx_v_self->min_samples_leaf = __pyx_v_min_samples_leaf;
|
|
|
|
/* "sklearn/tree/_tree.pyx":940
|
|
* self.max_features = max_features
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf # <<<<<<<<<<<<<<
|
|
* self.random_state = random_state
|
|
*
|
|
*/
|
|
__pyx_v_self->min_weight_leaf = __pyx_v_min_weight_leaf;
|
|
|
|
/* "sklearn/tree/_tree.pyx":941
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.random_state = random_state # <<<<<<<<<<<<<<
|
|
*
|
|
* def __dealloc__(self):
|
|
*/
|
|
__Pyx_INCREF(__pyx_v_random_state);
|
|
__Pyx_GIVEREF(__pyx_v_random_state);
|
|
__Pyx_GOTREF(__pyx_v_self->random_state);
|
|
__Pyx_DECREF(__pyx_v_self->random_state);
|
|
__pyx_v_self->random_state = __pyx_v_random_state;
|
|
|
|
/* "sklearn/tree/_tree.pyx":923
|
|
*
|
|
* cdef class Splitter:
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":943
|
|
* self.random_state = random_state
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.samples)
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_8Splitter_3__dealloc__(PyObject *__pyx_v_self); /*proto*/
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_8Splitter_3__dealloc__(PyObject *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0);
|
|
__pyx_pf_7sklearn_4tree_5_tree_8Splitter_2__dealloc__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_8Splitter_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":945
|
|
* def __dealloc__(self):
|
|
* """Destructor."""
|
|
* free(self.samples) # <<<<<<<<<<<<<<
|
|
* free(self.features)
|
|
* free(self.constant_features)
|
|
*/
|
|
free(__pyx_v_self->samples);
|
|
|
|
/* "sklearn/tree/_tree.pyx":946
|
|
* """Destructor."""
|
|
* free(self.samples)
|
|
* free(self.features) # <<<<<<<<<<<<<<
|
|
* free(self.constant_features)
|
|
* free(self.feature_values)
|
|
*/
|
|
free(__pyx_v_self->features);
|
|
|
|
/* "sklearn/tree/_tree.pyx":947
|
|
* free(self.samples)
|
|
* free(self.features)
|
|
* free(self.constant_features) # <<<<<<<<<<<<<<
|
|
* free(self.feature_values)
|
|
*
|
|
*/
|
|
free(__pyx_v_self->constant_features);
|
|
|
|
/* "sklearn/tree/_tree.pyx":948
|
|
* free(self.features)
|
|
* free(self.constant_features)
|
|
* free(self.feature_values) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __getstate__(self):
|
|
*/
|
|
free(__pyx_v_self->feature_values);
|
|
|
|
/* "sklearn/tree/_tree.pyx":943
|
|
* self.random_state = random_state
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.samples)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":950
|
|
* free(self.feature_values)
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* return {}
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_5__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_5__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__getstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_4__getstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_4__getstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
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("__getstate__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":951
|
|
*
|
|
* def __getstate__(self):
|
|
* return {} # <<<<<<<<<<<<<<
|
|
*
|
|
* def __setstate__(self, d):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 951; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":950
|
|
* free(self.feature_values)
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* return {}
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.__getstate__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":953
|
|
* return {}
|
|
*
|
|
* def __setstate__(self, d): # <<<<<<<<<<<<<<
|
|
* pass
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_7__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_7__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_6__setstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), ((PyObject *)__pyx_v_d));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_6__setstate__(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__", 0);
|
|
|
|
/* function exit code */
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":956
|
|
* pass
|
|
*
|
|
* cdef void init(self, # <<<<<<<<<<<<<<
|
|
* object X,
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_8Splitter_init(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_j;
|
|
double __pyx_v_weighted_n_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_features;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_y;
|
|
__Pyx_Buffer __pyx_pybuffer_y;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t __pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_11;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("init", 0);
|
|
__pyx_pybuffer_y.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_y.refcount = 0;
|
|
__pyx_pybuffernd_y.data = NULL;
|
|
__pyx_pybuffernd_y.rcbuffer = &__pyx_pybuffer_y;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_y.rcbuffer->pybuffer, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DOUBLE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 956; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_y.diminfo[0].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_y.diminfo[0].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[0]; __pyx_pybuffernd_y.diminfo[1].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[1]; __pyx_pybuffernd_y.diminfo[1].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[1];
|
|
|
|
/* "sklearn/tree/_tree.pyx":962
|
|
* """Initialize the splitter."""
|
|
* # Reset random state
|
|
* self.rand_r_state = self.random_state.randint(0, RAND_R_MAX) # <<<<<<<<<<<<<<
|
|
* # Initialize samples and features structures
|
|
* cdef SIZE_t n_samples = X.shape[0]
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_self->random_state, __pyx_n_s_randint); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 962; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyInt_From_int(__pyx_e_7sklearn_4tree_5_tree_RAND_R_MAX); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 962; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = PyTuple_New(2); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 962; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_int_0);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_int_0);
|
|
__Pyx_GIVEREF(__pyx_int_0);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_3, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 962; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_4 = __Pyx_PyInt_As_npy_uint32(__pyx_t_2); if (unlikely((__pyx_t_4 == (npy_uint32)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 962; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_v_self->rand_r_state = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":964
|
|
* self.rand_r_state = self.random_state.randint(0, RAND_R_MAX)
|
|
* # Initialize samples and features structures
|
|
* cdef SIZE_t n_samples = X.shape[0] # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* samples = safe_realloc(&self.samples, n_samples)
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 964; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = __Pyx_GetItemInt(__pyx_t_2, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_3 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 964; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_3); if (unlikely((__pyx_t_5 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 964; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_v_n_samples = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":965
|
|
* # Initialize samples and features structures
|
|
* cdef SIZE_t n_samples = X.shape[0]
|
|
* cdef SIZE_t* samples = safe_realloc(&self.samples, n_samples) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t i, j
|
|
*/
|
|
__pyx_t_6 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->samples), __pyx_v_n_samples); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 965; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_samples = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":968
|
|
*
|
|
* cdef SIZE_t i, j
|
|
* cdef double weighted_n_samples = 0.0 # <<<<<<<<<<<<<<
|
|
* j = 0
|
|
*
|
|
*/
|
|
__pyx_v_weighted_n_samples = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":969
|
|
* cdef SIZE_t i, j
|
|
* cdef double weighted_n_samples = 0.0
|
|
* j = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* for i in range(n_samples):
|
|
*/
|
|
__pyx_v_j = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":971
|
|
* j = 0
|
|
*
|
|
* for i in range(n_samples): # <<<<<<<<<<<<<<
|
|
* # Only work with positively weighted samples
|
|
* if sample_weight == NULL or sample_weight[i] != 0.0:
|
|
*/
|
|
__pyx_t_5 = __pyx_v_n_samples;
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_5; __pyx_t_7+=1) {
|
|
__pyx_v_i = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":973
|
|
* for i in range(n_samples):
|
|
* # Only work with positively weighted samples
|
|
* if sample_weight == NULL or sample_weight[i] != 0.0: # <<<<<<<<<<<<<<
|
|
* samples[j] = i
|
|
* j += 1
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_sample_weight == NULL) != 0);
|
|
if (!__pyx_t_8) {
|
|
__pyx_t_9 = (((__pyx_v_sample_weight[__pyx_v_i]) != 0.0) != 0);
|
|
__pyx_t_10 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_10 = __pyx_t_8;
|
|
}
|
|
if (__pyx_t_10) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":974
|
|
* # Only work with positively weighted samples
|
|
* if sample_weight == NULL or sample_weight[i] != 0.0:
|
|
* samples[j] = i # <<<<<<<<<<<<<<
|
|
* j += 1
|
|
*
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_j]) = __pyx_v_i;
|
|
|
|
/* "sklearn/tree/_tree.pyx":975
|
|
* if sample_weight == NULL or sample_weight[i] != 0.0:
|
|
* samples[j] = i
|
|
* j += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* if sample_weight != NULL:
|
|
*/
|
|
__pyx_v_j = (__pyx_v_j + 1);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":977
|
|
* j += 1
|
|
*
|
|
* if sample_weight != NULL: # <<<<<<<<<<<<<<
|
|
* weighted_n_samples += sample_weight[i]
|
|
* else:
|
|
*/
|
|
__pyx_t_10 = ((__pyx_v_sample_weight != NULL) != 0);
|
|
if (__pyx_t_10) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":978
|
|
*
|
|
* if sample_weight != NULL:
|
|
* weighted_n_samples += sample_weight[i] # <<<<<<<<<<<<<<
|
|
* else:
|
|
* weighted_n_samples += 1.0
|
|
*/
|
|
__pyx_v_weighted_n_samples = (__pyx_v_weighted_n_samples + (__pyx_v_sample_weight[__pyx_v_i]));
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":980
|
|
* weighted_n_samples += sample_weight[i]
|
|
* else:
|
|
* weighted_n_samples += 1.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* self.n_samples = j
|
|
*/
|
|
__pyx_v_weighted_n_samples = (__pyx_v_weighted_n_samples + 1.0);
|
|
}
|
|
__pyx_L6:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":982
|
|
* weighted_n_samples += 1.0
|
|
*
|
|
* self.n_samples = j # <<<<<<<<<<<<<<
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
*
|
|
*/
|
|
__pyx_v_self->n_samples = __pyx_v_j;
|
|
|
|
/* "sklearn/tree/_tree.pyx":983
|
|
*
|
|
* self.n_samples = j
|
|
* self.weighted_n_samples = weighted_n_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_features = X.shape[1]
|
|
*/
|
|
__pyx_v_self->weighted_n_samples = __pyx_v_weighted_n_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":985
|
|
* self.weighted_n_samples = weighted_n_samples
|
|
*
|
|
* cdef SIZE_t n_features = X.shape[1] # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* features = safe_realloc(&self.features, n_features)
|
|
*
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 985; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_GetItemInt(__pyx_t_3, 1, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 985; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_2); if (unlikely((__pyx_t_5 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 985; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_v_n_features = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":986
|
|
*
|
|
* cdef SIZE_t n_features = X.shape[1]
|
|
* cdef SIZE_t* features = safe_realloc(&self.features, n_features) # <<<<<<<<<<<<<<
|
|
*
|
|
* for i in range(n_features):
|
|
*/
|
|
__pyx_t_6 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->features), __pyx_v_n_features); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 986; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_features = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":988
|
|
* cdef SIZE_t* features = safe_realloc(&self.features, n_features)
|
|
*
|
|
* for i in range(n_features): # <<<<<<<<<<<<<<
|
|
* features[i] = i
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_n_features;
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_5; __pyx_t_7+=1) {
|
|
__pyx_v_i = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":989
|
|
*
|
|
* for i in range(n_features):
|
|
* features[i] = i # <<<<<<<<<<<<<<
|
|
*
|
|
* self.n_features = n_features
|
|
*/
|
|
(__pyx_v_features[__pyx_v_i]) = __pyx_v_i;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":991
|
|
* features[i] = i
|
|
*
|
|
* self.n_features = n_features # <<<<<<<<<<<<<<
|
|
*
|
|
* safe_realloc(&self.feature_values, n_samples)
|
|
*/
|
|
__pyx_v_self->n_features = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":993
|
|
* self.n_features = n_features
|
|
*
|
|
* safe_realloc(&self.feature_values, n_samples) # <<<<<<<<<<<<<<
|
|
* safe_realloc(&self.constant_features, n_features)
|
|
*
|
|
*/
|
|
__pyx_t_11 = __pyx_fuse_0__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->feature_values), __pyx_v_n_samples); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 993; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":994
|
|
*
|
|
* safe_realloc(&self.feature_values, n_samples)
|
|
* safe_realloc(&self.constant_features, n_features) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize y, sample_weight
|
|
*/
|
|
__pyx_t_6 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->constant_features), __pyx_v_n_features); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 994; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":997
|
|
*
|
|
* # Initialize y, sample_weight
|
|
* self.y = <DOUBLE_t*> y.data # <<<<<<<<<<<<<<
|
|
* self.y_stride = <SIZE_t> y.strides[0] / <SIZE_t> y.itemsize
|
|
* self.sample_weight = sample_weight
|
|
*/
|
|
__pyx_v_self->y = ((__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *)__pyx_v_y->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":998
|
|
* # Initialize y, sample_weight
|
|
* self.y = <DOUBLE_t*> y.data
|
|
* self.y_stride = <SIZE_t> y.strides[0] / <SIZE_t> y.itemsize # <<<<<<<<<<<<<<
|
|
* self.sample_weight = sample_weight
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_y), __pyx_n_s_itemsize); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 998; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_5 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_2); if (unlikely((__pyx_t_5 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 998; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_v_self->y_stride = (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_y->strides[0])) / ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_5));
|
|
|
|
/* "sklearn/tree/_tree.pyx":999
|
|
* self.y = <DOUBLE_t*> y.data
|
|
* self.y_stride = <SIZE_t> y.strides[0] / <SIZE_t> y.itemsize
|
|
* self.sample_weight = sample_weight # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_reset(self, SIZE_t start, SIZE_t end,
|
|
*/
|
|
__pyx_v_self->sample_weight = __pyx_v_sample_weight;
|
|
|
|
/* "sklearn/tree/_tree.pyx":956
|
|
* pass
|
|
*
|
|
* cdef void init(self, # <<<<<<<<<<<<<<
|
|
* object X,
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.tree._tree.Splitter.init", __pyx_clineno, __pyx_lineno, __pyx_filename, 0);
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1001
|
|
* self.sample_weight = sample_weight
|
|
*
|
|
* cdef void node_reset(self, SIZE_t start, SIZE_t end, # <<<<<<<<<<<<<<
|
|
* double* weighted_n_node_samples) nogil:
|
|
* """Reset splitter on node samples[start:end]."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_8Splitter_node_reset(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end, double *__pyx_v_weighted_n_node_samples) {
|
|
double __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1004
|
|
* double* weighted_n_node_samples) nogil:
|
|
* """Reset splitter on node samples[start:end]."""
|
|
* self.start = start # <<<<<<<<<<<<<<
|
|
* self.end = end
|
|
*
|
|
*/
|
|
__pyx_v_self->start = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1005
|
|
* """Reset splitter on node samples[start:end]."""
|
|
* self.start = start
|
|
* self.end = end # <<<<<<<<<<<<<<
|
|
*
|
|
* self.criterion.init(self.y,
|
|
*/
|
|
__pyx_v_self->end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1007
|
|
* self.end = end
|
|
*
|
|
* self.criterion.init(self.y, # <<<<<<<<<<<<<<
|
|
* self.y_stride,
|
|
* self.sample_weight,
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->criterion->__pyx_vtab)->init(__pyx_v_self->criterion, __pyx_v_self->y, __pyx_v_self->y_stride, __pyx_v_self->sample_weight, __pyx_v_self->weighted_n_samples, __pyx_v_self->samples, __pyx_v_start, __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1015
|
|
* end)
|
|
*
|
|
* weighted_n_node_samples[0] = self.criterion.weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split,
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->criterion->weighted_n_node_samples;
|
|
(__pyx_v_weighted_n_node_samples[0]) = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1001
|
|
* self.sample_weight = sample_weight
|
|
*
|
|
* cdef void node_reset(self, SIZE_t start, SIZE_t end, # <<<<<<<<<<<<<<
|
|
* double* weighted_n_node_samples) nogil:
|
|
* """Reset splitter on node samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1017
|
|
* weighted_n_node_samples[0] = self.criterion.weighted_n_node_samples
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find a split on node samples[start:end]."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_8Splitter_node_split(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, CYTHON_UNUSED double __pyx_v_impurity, CYTHON_UNUSED struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_split, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_constant_features) {
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1022
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Copy the value of node samples[start:end] into dest."""
|
|
* self.criterion.node_value(dest)
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_8Splitter_node_value(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, double *__pyx_v_dest) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1024
|
|
* cdef void node_value(self, double* dest) nogil:
|
|
* """Copy the value of node samples[start:end] into dest."""
|
|
* self.criterion.node_value(dest) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double node_impurity(self) nogil:
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->criterion->__pyx_vtab)->node_value(__pyx_v_self->criterion, __pyx_v_dest);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1022
|
|
* pass
|
|
*
|
|
* cdef void node_value(self, double* dest) nogil: # <<<<<<<<<<<<<<
|
|
* """Copy the value of node samples[start:end] into dest."""
|
|
* self.criterion.node_value(dest)
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1026
|
|
* self.criterion.node_value(dest)
|
|
*
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Copy the impurity of node samples[start:end."""
|
|
* return self.criterion.node_impurity()
|
|
*/
|
|
|
|
static double __pyx_f_7sklearn_4tree_5_tree_8Splitter_node_impurity(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
double __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1028
|
|
* cdef double node_impurity(self) nogil:
|
|
* """Copy the impurity of node samples[start:end."""
|
|
* return self.criterion.node_impurity() # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->criterion->__pyx_vtab)->node_impurity(__pyx_v_self->criterion);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1026
|
|
* self.criterion.node_value(dest)
|
|
*
|
|
* cdef double node_impurity(self) nogil: # <<<<<<<<<<<<<<
|
|
* """Copy the impurity of node samples[start:end."""
|
|
* return self.criterion.node_impurity()
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":86
|
|
*
|
|
* # Internal structures
|
|
* cdef public Criterion criterion # Impurity criterion # <<<<<<<<<<<<<<
|
|
* cdef public SIZE_t max_features # Number of features to test
|
|
* cdef public SIZE_t min_samples_leaf # Min samples in a leaf
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->criterion));
|
|
__pyx_r = ((PyObject *)__pyx_v_self->criterion);
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
if (!(likely(((__pyx_v_value) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_value, __pyx_ptype_7sklearn_4tree_5_tree_Criterion))))) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 86; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_1 = __pyx_v_value;
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__Pyx_GOTREF(__pyx_v_self->criterion);
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_self->criterion));
|
|
__pyx_v_self->criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.criterion.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_5__del__(PyObject *__pyx_v_self); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_5__del__(PyObject *__pyx_v_self) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__del__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion_4__del__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_9criterion_4__del__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__del__", 0);
|
|
__Pyx_INCREF(Py_None);
|
|
__Pyx_GIVEREF(Py_None);
|
|
__Pyx_GOTREF(__pyx_v_self->criterion);
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_self->criterion));
|
|
__pyx_v_self->criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)Py_None);
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":87
|
|
* # Internal structures
|
|
* cdef public Criterion criterion # Impurity criterion
|
|
* cdef public SIZE_t max_features # Number of features to test # <<<<<<<<<<<<<<
|
|
* cdef public SIZE_t min_samples_leaf # Min samples in a leaf
|
|
* cdef public double min_weight_leaf # Minimum weight in a leaf
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_12max_features_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_12max_features_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_12max_features___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_12max_features___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->max_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 87; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.max_features.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_12max_features_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_12max_features_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_12max_features_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_12max_features_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 87; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->max_features = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.max_features.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":88
|
|
* cdef public Criterion criterion # Impurity criterion
|
|
* cdef public SIZE_t max_features # Number of features to test
|
|
* cdef public SIZE_t min_samples_leaf # Min samples in a leaf # <<<<<<<<<<<<<<
|
|
* cdef public double min_weight_leaf # Minimum weight in a leaf
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->min_samples_leaf); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 88; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.min_samples_leaf.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 88; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->min_samples_leaf = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.min_samples_leaf.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":89
|
|
* cdef public SIZE_t max_features # Number of features to test
|
|
* cdef public SIZE_t min_samples_leaf # Min samples in a leaf
|
|
* cdef public double min_weight_leaf # Minimum weight in a leaf # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef object random_state # Random state
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyFloat_FromDouble(__pyx_v_self->min_weight_leaf); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 89; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.min_weight_leaf.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
double __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __pyx_PyFloat_AsDouble(__pyx_v_value); if (unlikely((__pyx_t_1 == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 89; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->min_weight_leaf = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Splitter.min_weight_leaf.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1036
|
|
* cdef SIZE_t X_fx_stride
|
|
*
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_17BaseDenseSplitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_17BaseDenseSplitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
CYTHON_UNUSED double __pyx_v_min_weight_leaf;
|
|
CYTHON_UNUSED PyObject *__pyx_v_random_state = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_criterion,&__pyx_n_s_max_features,&__pyx_n_s_min_samples_leaf,&__pyx_n_s_min_weight_leaf,&__pyx_n_s_random_state,0};
|
|
PyObject* values[5] = {0,0,0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_criterion)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_features)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (likely((values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_weight_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 4:
|
|
if (likely((values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_random_state)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 5) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
}
|
|
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[0]);
|
|
__pyx_v_max_features = __Pyx_PyInt_As_Py_intptr_t(values[1]); if (unlikely((__pyx_v_max_features == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_samples_leaf = __Pyx_PyInt_As_Py_intptr_t(values[2]); if (unlikely((__pyx_v_min_samples_leaf == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1037; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_weight_leaf = __pyx_PyFloat_AsDouble(values[3]); if (unlikely((__pyx_v_min_weight_leaf == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1037; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_random_state = values[4];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BaseDenseSplitter.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_criterion), __pyx_ptype_7sklearn_4tree_5_tree_Criterion, 1, "criterion", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1036; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_17BaseDenseSplitter___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter *)__pyx_v_self), __pyx_v_criterion, __pyx_v_max_features, __pyx_v_min_samples_leaf, __pyx_v_min_weight_leaf, __pyx_v_random_state);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_17BaseDenseSplitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, CYTHON_UNUSED double __pyx_v_min_weight_leaf, CYTHON_UNUSED PyObject *__pyx_v_random_state) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1041
|
|
* # Parent __cinit__ is automatically called
|
|
*
|
|
* self.X = NULL # <<<<<<<<<<<<<<
|
|
* self.X_sample_stride = 0
|
|
* self.X_fx_stride = 0
|
|
*/
|
|
__pyx_v_self->X = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1042
|
|
*
|
|
* self.X = NULL
|
|
* self.X_sample_stride = 0 # <<<<<<<<<<<<<<
|
|
* self.X_fx_stride = 0
|
|
*
|
|
*/
|
|
__pyx_v_self->X_sample_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1043
|
|
* self.X = NULL
|
|
* self.X_sample_stride = 0
|
|
* self.X_fx_stride = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void init(self,
|
|
*/
|
|
__pyx_v_self->X_fx_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1036
|
|
* cdef SIZE_t X_fx_stride
|
|
*
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1045
|
|
* self.X_fx_stride = 0
|
|
*
|
|
* cdef void init(self, # <<<<<<<<<<<<<<
|
|
* object X,
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_17BaseDenseSplitter_init(struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter *__pyx_v_self, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight) {
|
|
PyArrayObject *__pyx_v_X_ndarray = 0;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_y;
|
|
__Pyx_Buffer __pyx_pybuffer_y;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("init", 0);
|
|
__pyx_pybuffer_y.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_y.refcount = 0;
|
|
__pyx_pybuffernd_y.data = NULL;
|
|
__pyx_pybuffernd_y.rcbuffer = &__pyx_pybuffer_y;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_y.rcbuffer->pybuffer, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DOUBLE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1045; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_y.diminfo[0].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_y.diminfo[0].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[0]; __pyx_pybuffernd_y.diminfo[1].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[1]; __pyx_pybuffernd_y.diminfo[1].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[1];
|
|
|
|
/* "sklearn/tree/_tree.pyx":1052
|
|
*
|
|
* # Call parent init
|
|
* Splitter.init(self, X, y, sample_weight) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize X
|
|
*/
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Splitter->init(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), __pyx_v_X, ((PyArrayObject *)__pyx_v_y), __pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1055
|
|
*
|
|
* # Initialize X
|
|
* cdef np.ndarray X_ndarray = X # <<<<<<<<<<<<<<
|
|
*
|
|
* self.X = <DTYPE_t*> X_ndarray.data
|
|
*/
|
|
if (!(likely(((__pyx_v_X) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_X, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1055; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_1 = __pyx_v_X;
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__pyx_v_X_ndarray = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1057
|
|
* cdef np.ndarray X_ndarray = X
|
|
*
|
|
* self.X = <DTYPE_t*> X_ndarray.data # <<<<<<<<<<<<<<
|
|
* self.X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize
|
|
* self.X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize
|
|
*/
|
|
__pyx_v_self->X = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_X_ndarray->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1058
|
|
*
|
|
* self.X = <DTYPE_t*> X_ndarray.data
|
|
* self.X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize # <<<<<<<<<<<<<<
|
|
* self.X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_strides); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1058; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_GetItemInt(__pyx_t_1, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1058; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_3 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_2); if (unlikely((__pyx_t_3 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1058; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_itemsize); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1058; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_4 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_2); if (unlikely((__pyx_t_4 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1058; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_v_self->X_sample_stride = (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_3) / ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_4));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1059
|
|
* self.X = <DTYPE_t*> X_ndarray.data
|
|
* self.X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize
|
|
* self.X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_strides); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = __Pyx_GetItemInt(__pyx_t_2, 1, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_1 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1059; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_4 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_1); if (unlikely((__pyx_t_4 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_itemsize); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_1); if (unlikely((__pyx_t_3 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_v_self->X_fx_stride = (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_4) / ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_3));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1045
|
|
* self.X_fx_stride = 0
|
|
*
|
|
* cdef void init(self, # <<<<<<<<<<<<<<
|
|
* object X,
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.tree._tree.BaseDenseSplitter.init", __pyx_clineno, __pyx_lineno, __pyx_filename, 0);
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_X_ndarray);
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1064
|
|
* cdef class BestSplitter(BaseDenseSplitter):
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (BestSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_12BestSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_12BestSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_12BestSplitter___reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_12BestSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1065
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self):
|
|
* return (BestSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1066
|
|
* def __reduce__(self):
|
|
* return (BestSplitter, (self.criterion,
|
|
* self.max_features, # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.max_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1066; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1067
|
|
* return (BestSplitter, (self.criterion,
|
|
* self.max_features,
|
|
* self.min_samples_leaf, # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__())
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1067; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1068
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf, # <<<<<<<<<<<<<<
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyFloat_FromDouble(__pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1068; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1065
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self):
|
|
* return (BestSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1065; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 3, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 4, __pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__Pyx_GIVEREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1069
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split,
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1069; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1069; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1065
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self):
|
|
* return (BestSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1065; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_BestSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_BestSplitter)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_BestSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1064
|
|
* cdef class BestSplitter(BaseDenseSplitter):
|
|
* """Splitter for finding the best split."""
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (BestSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestSplitter.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1071
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best split on node samples[start:end]."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_12BestSplitter_node_split(struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter *__pyx_v_self, double __pyx_v_impurity, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_constant_features) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_constant_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_sample_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_fx_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_best;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_current;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_j;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_visited_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_found_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_drawn_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_known_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_partition_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_3;
|
|
double __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1075
|
|
* """Find the best split on node samples[start:end]."""
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1076
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.start;
|
|
__pyx_v_start = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1077
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.end;
|
|
__pyx_v_end = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1079
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
* cdef SIZE_t* features = self.features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.features;
|
|
__pyx_v_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1080
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.constant_features;
|
|
__pyx_v_constant_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1081
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t* X = self.X
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_features;
|
|
__pyx_v_n_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1083
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
* cdef DTYPE_t* X = self.X # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X;
|
|
__pyx_v_X = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1084
|
|
*
|
|
* cdef DTYPE_t* X = self.X
|
|
* cdef DTYPE_t* Xf = self.feature_values # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.__pyx_base.feature_values;
|
|
__pyx_v_Xf = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1085
|
|
* cdef DTYPE_t* X = self.X
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef SIZE_t max_features = self.max_features
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.X_sample_stride;
|
|
__pyx_v_X_sample_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1086
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.X_fx_stride;
|
|
__pyx_v_X_fx_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1087
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef SIZE_t max_features = self.max_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.max_features;
|
|
__pyx_v_max_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1088
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1089
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1090
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SplitRecord best, current
|
|
*/
|
|
__pyx_v_random_state = (&__pyx_v_self->__pyx_base.__pyx_base.rand_r_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1094
|
|
* cdef SplitRecord best, current
|
|
*
|
|
* cdef SIZE_t f_i = n_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* cdef SIZE_t n_visited_features = 0
|
|
*/
|
|
__pyx_v_f_i = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1096
|
|
* cdef SIZE_t f_i = n_features
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* cdef SIZE_t n_visited_features = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0
|
|
*/
|
|
__pyx_v_n_visited_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1098
|
|
* cdef SIZE_t n_visited_features = 0
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
*/
|
|
__pyx_v_n_found_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1100
|
|
* cdef SIZE_t n_found_constants = 0
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
*/
|
|
__pyx_v_n_drawn_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1101
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0] # <<<<<<<<<<<<<<
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
*/
|
|
__pyx_v_n_known_constants = (__pyx_v_n_constant_features[0]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1103
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t current_feature_value
|
|
* cdef SIZE_t partition_end
|
|
*/
|
|
__pyx_v_n_total_constants = __pyx_v_n_known_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1107
|
|
* cdef SIZE_t partition_end
|
|
*
|
|
* _init_split(&best, end) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Sample up to max_features without replacement using a
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree__init_split((&__pyx_v_best), __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1118
|
|
* # newly discovered constant features to spare computation on descendant
|
|
* # nodes.
|
|
* while (f_i > n_total_constants and # Stop early if remaining features # <<<<<<<<<<<<<<
|
|
* # are constant
|
|
* (n_visited_features < max_features or
|
|
*/
|
|
while (1) {
|
|
__pyx_t_5 = ((__pyx_v_f_i > __pyx_v_n_total_constants) != 0);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1120
|
|
* while (f_i > n_total_constants and # Stop early if remaining features
|
|
* # are constant
|
|
* (n_visited_features < max_features or # <<<<<<<<<<<<<<
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_n_visited_features < __pyx_v_max_features) != 0);
|
|
if (!__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1122
|
|
* (n_visited_features < max_features or
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)): # <<<<<<<<<<<<<<
|
|
*
|
|
* n_visited_features += 1
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_n_visited_features <= (__pyx_v_n_found_constants + __pyx_v_n_drawn_constants)) != 0);
|
|
__pyx_t_8 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_6;
|
|
}
|
|
__pyx_t_6 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_5;
|
|
}
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1124
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*
|
|
* n_visited_features += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Loop invariant: elements of features in
|
|
*/
|
|
__pyx_v_n_visited_features = (__pyx_v_n_visited_features + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1138
|
|
*
|
|
* # Draw a feature at random
|
|
* f_j = rand_int(n_drawn_constants, f_i - n_found_constants, # <<<<<<<<<<<<<<
|
|
* random_state)
|
|
*
|
|
*/
|
|
__pyx_v_f_j = __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_v_n_drawn_constants, (__pyx_v_f_i - __pyx_v_n_found_constants), __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1141
|
|
* random_state)
|
|
*
|
|
* if f_j < n_known_constants: # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_f_j < __pyx_v_n_known_constants) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1143
|
|
* if f_j < n_known_constants:
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j] # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1144
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants] # <<<<<<<<<<<<<<
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_drawn_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1145
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* n_drawn_constants += 1
|
|
*/
|
|
(__pyx_v_features[__pyx_v_n_drawn_constants]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1147
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
* n_drawn_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_drawn_constants = (__pyx_v_n_drawn_constants + 1);
|
|
goto __pyx_L5;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1151
|
|
* else:
|
|
* # f_j in the interval [n_known_constants, f_i - n_found_constants[
|
|
* f_j += n_found_constants # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
*/
|
|
__pyx_v_f_j = (__pyx_v_f_j + __pyx_v_n_found_constants);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1154
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
* current.feature = features[f_j] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Sort samples along that feature; first copy the feature
|
|
*/
|
|
__pyx_v_current.feature = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1160
|
|
* # Xf[i] == X[samples[i], j], so the sort uses the cache more
|
|
* # effectively.
|
|
* for p in range(start, end): # <<<<<<<<<<<<<<
|
|
* Xf[p] = X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * current.feature]
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end;
|
|
for (__pyx_t_9 = __pyx_v_start; __pyx_t_9 < __pyx_t_2; __pyx_t_9+=1) {
|
|
__pyx_v_p = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1161
|
|
* # effectively.
|
|
* for p in range(start, end):
|
|
* Xf[p] = X[X_sample_stride * samples[p] + # <<<<<<<<<<<<<<
|
|
* X_fx_stride * current.feature]
|
|
*
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_p]) = (__pyx_v_X[((__pyx_v_X_sample_stride * (__pyx_v_samples[__pyx_v_p])) + (__pyx_v_X_fx_stride * __pyx_v_current.feature))]);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1164
|
|
* X_fx_stride * current.feature]
|
|
*
|
|
* sort(Xf + start, samples + start, end - start) # <<<<<<<<<<<<<<
|
|
*
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sort((__pyx_v_Xf + __pyx_v_start), (__pyx_v_samples + __pyx_v_start), (__pyx_v_end - __pyx_v_start));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1166
|
|
* sort(Xf + start, samples + start, end - start)
|
|
*
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD: # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_Xf[(__pyx_v_end - 1)]) <= ((__pyx_v_Xf[__pyx_v_start]) + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1167
|
|
*
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants] # <<<<<<<<<<<<<<
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_total_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1168
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature # <<<<<<<<<<<<<<
|
|
*
|
|
* n_found_constants += 1
|
|
*/
|
|
__pyx_t_2 = __pyx_v_current.feature;
|
|
(__pyx_v_features[__pyx_v_n_total_constants]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1170
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
* n_found_constants += 1 # <<<<<<<<<<<<<<
|
|
* n_total_constants += 1
|
|
*
|
|
*/
|
|
__pyx_v_n_found_constants = (__pyx_v_n_found_constants + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1171
|
|
*
|
|
* n_found_constants += 1
|
|
* n_total_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_total_constants = (__pyx_v_n_total_constants + 1);
|
|
goto __pyx_L8;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1174
|
|
*
|
|
* else:
|
|
* f_i -= 1 # <<<<<<<<<<<<<<
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i]
|
|
*
|
|
*/
|
|
__pyx_v_f_i = (__pyx_v_f_i - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1175
|
|
* else:
|
|
* f_i -= 1
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Evaluate all splits
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_features[__pyx_v_f_j]);
|
|
__pyx_t_9 = (__pyx_v_features[__pyx_v_f_i]);
|
|
(__pyx_v_features[__pyx_v_f_i]) = __pyx_t_2;
|
|
(__pyx_v_features[__pyx_v_f_j]) = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1178
|
|
*
|
|
* # Evaluate all splits
|
|
* self.criterion.reset() # <<<<<<<<<<<<<<
|
|
* p = start
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->reset(__pyx_v_self->__pyx_base.__pyx_base.criterion);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1179
|
|
* # Evaluate all splits
|
|
* self.criterion.reset()
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < end:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1181
|
|
* p = start
|
|
*
|
|
* while p < end: # <<<<<<<<<<<<<<
|
|
* while (p + 1 < end and
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD):
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1182
|
|
*
|
|
* while p < end:
|
|
* while (p + 1 < end and # <<<<<<<<<<<<<<
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p += 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = (((__pyx_v_p + 1) < __pyx_v_end) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1183
|
|
* while p < end:
|
|
* while (p + 1 < end and
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD): # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_t_5 = (((__pyx_v_Xf[(__pyx_v_p + 1)]) <= ((__pyx_v_Xf[__pyx_v_p]) + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
__pyx_t_8 = __pyx_t_5;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_6;
|
|
}
|
|
if (!__pyx_t_8) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1184
|
|
* while (p + 1 < end and
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # (p + 1 >= end) or (X[samples[p + 1], current.feature] >
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1188
|
|
* # (p + 1 >= end) or (X[samples[p + 1], current.feature] >
|
|
* # X[samples[p], current.feature])
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
* # (p >= end) or (X[samples[p], current.feature] >
|
|
* # X[samples[p - 1], current.feature])
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1192
|
|
* # X[samples[p - 1], current.feature])
|
|
*
|
|
* if p < end: # <<<<<<<<<<<<<<
|
|
* current.pos = p
|
|
*
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1193
|
|
*
|
|
* if p < end:
|
|
* current.pos = p # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
*/
|
|
__pyx_v_current.pos = __pyx_v_p;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1196
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_8 = (((__pyx_v_current.pos - __pyx_v_start) < __pyx_v_min_samples_leaf) != 0);
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1197
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_end - __pyx_v_current.pos) < __pyx_v_min_samples_leaf) != 0);
|
|
__pyx_t_5 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_5 = __pyx_t_8;
|
|
}
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1198
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* self.criterion.update(current.pos)
|
|
*/
|
|
goto __pyx_L9_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1200
|
|
* continue
|
|
*
|
|
* self.criterion.update(current.pos) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->update(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_current.pos);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1203
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or # <<<<<<<<<<<<<<
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_5 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_left < __pyx_v_min_weight_leaf) != 0);
|
|
if (!__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1204
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_right < __pyx_v_min_weight_leaf) != 0);
|
|
__pyx_t_6 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_5;
|
|
}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1205
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*/
|
|
goto __pyx_L9_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1207
|
|
* continue
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity) # <<<<<<<<<<<<<<
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
*/
|
|
__pyx_v_current.improvement = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->impurity_improvement(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_impurity);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1209
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*
|
|
* if current.improvement > best.improvement: # <<<<<<<<<<<<<<
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current.improvement > __pyx_v_best.improvement) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1210
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
* self.criterion.children_impurity(¤t.impurity_left, # <<<<<<<<<<<<<<
|
|
* ¤t.impurity_right)
|
|
* current.threshold = (Xf[p - 1] + Xf[p]) / 2.0
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->children_impurity(__pyx_v_self->__pyx_base.__pyx_base.criterion, (&__pyx_v_current.impurity_left), (&__pyx_v_current.impurity_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1212
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
* current.threshold = (Xf[p - 1] + Xf[p]) / 2.0 # <<<<<<<<<<<<<<
|
|
*
|
|
* if current.threshold == Xf[p]:
|
|
*/
|
|
__pyx_v_current.threshold = (((__pyx_v_Xf[(__pyx_v_p - 1)]) + (__pyx_v_Xf[__pyx_v_p])) / 2.0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1214
|
|
* current.threshold = (Xf[p - 1] + Xf[p]) / 2.0
|
|
*
|
|
* if current.threshold == Xf[p]: # <<<<<<<<<<<<<<
|
|
* current.threshold = Xf[p - 1]
|
|
*
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current.threshold == (__pyx_v_Xf[__pyx_v_p])) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1215
|
|
*
|
|
* if current.threshold == Xf[p]:
|
|
* current.threshold = Xf[p - 1] # <<<<<<<<<<<<<<
|
|
*
|
|
* best = current # copy
|
|
*/
|
|
__pyx_v_current.threshold = (__pyx_v_Xf[(__pyx_v_p - 1)]);
|
|
goto __pyx_L17;
|
|
}
|
|
__pyx_L17:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1217
|
|
* current.threshold = Xf[p - 1]
|
|
*
|
|
* best = current # copy # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
*/
|
|
__pyx_v_best = __pyx_v_current;
|
|
goto __pyx_L16;
|
|
}
|
|
__pyx_L16:;
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
__pyx_L9_continue:;
|
|
}
|
|
}
|
|
__pyx_L8:;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1220
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end: # <<<<<<<<<<<<<<
|
|
* partition_end = end
|
|
* p = start
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_best.pos < __pyx_v_end) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1221
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end:
|
|
* partition_end = end # <<<<<<<<<<<<<<
|
|
* p = start
|
|
*
|
|
*/
|
|
__pyx_v_partition_end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1222
|
|
* if best.pos < end:
|
|
* partition_end = end
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < partition_end:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1224
|
|
* p = start
|
|
*
|
|
* while p < partition_end: # <<<<<<<<<<<<<<
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = ((__pyx_v_p < __pyx_v_partition_end) != 0);
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1226
|
|
* while p < partition_end:
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold: # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_X[((__pyx_v_X_sample_stride * (__pyx_v_samples[__pyx_v_p])) + (__pyx_v_X_fx_stride * __pyx_v_best.feature))]) <= __pyx_v_best.threshold) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1227
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold:
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
goto __pyx_L21;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1230
|
|
*
|
|
* else:
|
|
* partition_end -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* tmp = samples[partition_end]
|
|
*/
|
|
__pyx_v_partition_end = (__pyx_v_partition_end - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1232
|
|
* partition_end -= 1
|
|
*
|
|
* tmp = samples[partition_end] # <<<<<<<<<<<<<<
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_samples[__pyx_v_partition_end]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1233
|
|
*
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p] # <<<<<<<<<<<<<<
|
|
* samples[p] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_partition_end]) = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1234
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* # Respect invariant for constant features: the original order of
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_p]) = __pyx_v_tmp;
|
|
}
|
|
__pyx_L21:;
|
|
}
|
|
goto __pyx_L18;
|
|
}
|
|
__pyx_L18:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1239
|
|
* # element in features[:n_known_constants] must be preserved for sibling
|
|
* # and child nodes
|
|
* memcpy(features, constant_features, sizeof(SIZE_t) * n_known_constants) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Copy newly found constant features
|
|
*/
|
|
memcpy(__pyx_v_features, __pyx_v_constant_features, ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_known_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1242
|
|
*
|
|
* # Copy newly found constant features
|
|
* memcpy(constant_features + n_known_constants, # <<<<<<<<<<<<<<
|
|
* features + n_known_constants,
|
|
* sizeof(SIZE_t) * n_found_constants)
|
|
*/
|
|
memcpy((__pyx_v_constant_features + __pyx_v_n_known_constants), (__pyx_v_features + __pyx_v_n_known_constants), ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_found_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1247
|
|
*
|
|
* # Return values
|
|
* split[0] = best # <<<<<<<<<<<<<<
|
|
* n_constant_features[0] = n_total_constants
|
|
*
|
|
*/
|
|
(__pyx_v_split[0]) = __pyx_v_best;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1248
|
|
* # Return values
|
|
* split[0] = best
|
|
* n_constant_features[0] = n_total_constants # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_n_constant_features[0]) = __pyx_v_n_total_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1071
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best split on node samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1253
|
|
* # Sort n-element arrays pointed to by Xf and samples, simultaneously,
|
|
* # by the values in Xf. Algorithm: Introsort (Musser, SP&E, 1997).
|
|
* cdef inline void sort(DTYPE_t* Xf, SIZE_t* samples, SIZE_t n) nogil: # <<<<<<<<<<<<<<
|
|
* cdef int maxd = 2 * <int>log(n)
|
|
* introsort(Xf, samples, n, maxd)
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_sort(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n) {
|
|
int __pyx_v_maxd;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1254
|
|
* # by the values in Xf. Algorithm: Introsort (Musser, SP&E, 1997).
|
|
* cdef inline void sort(DTYPE_t* Xf, SIZE_t* samples, SIZE_t n) nogil:
|
|
* cdef int maxd = 2 * <int>log(n) # <<<<<<<<<<<<<<
|
|
* introsort(Xf, samples, n, maxd)
|
|
*
|
|
*/
|
|
__pyx_v_maxd = (2 * ((int)__pyx_f_7sklearn_4tree_5_tree_log(__pyx_v_n)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1255
|
|
* cdef inline void sort(DTYPE_t* Xf, SIZE_t* samples, SIZE_t n) nogil:
|
|
* cdef int maxd = 2 * <int>log(n)
|
|
* introsort(Xf, samples, n, maxd) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_introsort(__pyx_v_Xf, __pyx_v_samples, __pyx_v_n, __pyx_v_maxd);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1253
|
|
* # Sort n-element arrays pointed to by Xf and samples, simultaneously,
|
|
* # by the values in Xf. Algorithm: Introsort (Musser, SP&E, 1997).
|
|
* cdef inline void sort(DTYPE_t* Xf, SIZE_t* samples, SIZE_t n) nogil: # <<<<<<<<<<<<<<
|
|
* cdef int maxd = 2 * <int>log(n)
|
|
* introsort(Xf, samples, n, maxd)
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1258
|
|
*
|
|
*
|
|
* cdef inline void swap(DTYPE_t* Xf, SIZE_t* samples, SIZE_t i, SIZE_t j) nogil: # <<<<<<<<<<<<<<
|
|
* # Helper for sort
|
|
* Xf[i], Xf[j] = Xf[j], Xf[i]
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_swap(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_j) {
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1260
|
|
* cdef inline void swap(DTYPE_t* Xf, SIZE_t* samples, SIZE_t i, SIZE_t j) nogil:
|
|
* # Helper for sort
|
|
* Xf[i], Xf[j] = Xf[j], Xf[i] # <<<<<<<<<<<<<<
|
|
* samples[i], samples[j] = samples[j], samples[i]
|
|
*
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_Xf[__pyx_v_j]);
|
|
__pyx_t_2 = (__pyx_v_Xf[__pyx_v_i]);
|
|
(__pyx_v_Xf[__pyx_v_i]) = __pyx_t_1;
|
|
(__pyx_v_Xf[__pyx_v_j]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1261
|
|
* # Helper for sort
|
|
* Xf[i], Xf[j] = Xf[j], Xf[i]
|
|
* samples[i], samples[j] = samples[j], samples[i] # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_t_3 = (__pyx_v_samples[__pyx_v_j]);
|
|
__pyx_t_4 = (__pyx_v_samples[__pyx_v_i]);
|
|
(__pyx_v_samples[__pyx_v_i]) = __pyx_t_3;
|
|
(__pyx_v_samples[__pyx_v_j]) = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1258
|
|
*
|
|
*
|
|
* cdef inline void swap(DTYPE_t* Xf, SIZE_t* samples, SIZE_t i, SIZE_t j) nogil: # <<<<<<<<<<<<<<
|
|
* # Helper for sort
|
|
* Xf[i], Xf[j] = Xf[j], Xf[i]
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1264
|
|
*
|
|
*
|
|
* cdef inline DTYPE_t median3(DTYPE_t* Xf, SIZE_t n) nogil: # <<<<<<<<<<<<<<
|
|
* # Median of three pivot selection, after Bentley and McIlroy (1993).
|
|
* # Engineering a sort function. SP&E. Requires 8/3 comparisons on average.
|
|
*/
|
|
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_f_7sklearn_4tree_5_tree_median3(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n) {
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_a;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_b;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_c;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_r;
|
|
int __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1267
|
|
* # Median of three pivot selection, after Bentley and McIlroy (1993).
|
|
* # Engineering a sort function. SP&E. Requires 8/3 comparisons on average.
|
|
* cdef DTYPE_t a = Xf[0], b = Xf[n / 2], c = Xf[n - 1] # <<<<<<<<<<<<<<
|
|
* if a < b:
|
|
* if b < c:
|
|
*/
|
|
__pyx_v_a = (__pyx_v_Xf[0]);
|
|
__pyx_v_b = (__pyx_v_Xf[(__pyx_v_n / 2)]);
|
|
__pyx_v_c = (__pyx_v_Xf[(__pyx_v_n - 1)]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1268
|
|
* # Engineering a sort function. SP&E. Requires 8/3 comparisons on average.
|
|
* cdef DTYPE_t a = Xf[0], b = Xf[n / 2], c = Xf[n - 1]
|
|
* if a < b: # <<<<<<<<<<<<<<
|
|
* if b < c:
|
|
* return b
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_a < __pyx_v_b) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1269
|
|
* cdef DTYPE_t a = Xf[0], b = Xf[n / 2], c = Xf[n - 1]
|
|
* if a < b:
|
|
* if b < c: # <<<<<<<<<<<<<<
|
|
* return b
|
|
* elif a < c:
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_b < __pyx_v_c) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1270
|
|
* if a < b:
|
|
* if b < c:
|
|
* return b # <<<<<<<<<<<<<<
|
|
* elif a < c:
|
|
* return c
|
|
*/
|
|
__pyx_r = __pyx_v_b;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1271
|
|
* if b < c:
|
|
* return b
|
|
* elif a < c: # <<<<<<<<<<<<<<
|
|
* return c
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_a < __pyx_v_c) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1272
|
|
* return b
|
|
* elif a < c:
|
|
* return c # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return a
|
|
*/
|
|
__pyx_r = __pyx_v_c;
|
|
goto __pyx_L0;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1274
|
|
* return c
|
|
* else:
|
|
* return a # <<<<<<<<<<<<<<
|
|
* elif b < c:
|
|
* if a < c:
|
|
*/
|
|
__pyx_r = __pyx_v_a;
|
|
goto __pyx_L0;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1275
|
|
* else:
|
|
* return a
|
|
* elif b < c: # <<<<<<<<<<<<<<
|
|
* if a < c:
|
|
* return a
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_b < __pyx_v_c) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1276
|
|
* return a
|
|
* elif b < c:
|
|
* if a < c: # <<<<<<<<<<<<<<
|
|
* return a
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_a < __pyx_v_c) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1277
|
|
* elif b < c:
|
|
* if a < c:
|
|
* return a # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return c
|
|
*/
|
|
__pyx_r = __pyx_v_a;
|
|
goto __pyx_L0;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1279
|
|
* return a
|
|
* else:
|
|
* return c # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return b
|
|
*/
|
|
__pyx_r = __pyx_v_c;
|
|
goto __pyx_L0;
|
|
}
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1281
|
|
* return c
|
|
* else:
|
|
* return b # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_b;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1264
|
|
*
|
|
*
|
|
* cdef inline DTYPE_t median3(DTYPE_t* Xf, SIZE_t n) nogil: # <<<<<<<<<<<<<<
|
|
* # Median of three pivot selection, after Bentley and McIlroy (1993).
|
|
* # Engineering a sort function. SP&E. Requires 8/3 comparisons on average.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1286
|
|
* # Introsort with median of 3 pivot selection and 3-way partition function
|
|
* # (robust to repeated elements, e.g. lots of zero features).
|
|
* cdef void introsort(DTYPE_t* Xf, SIZE_t *samples, SIZE_t n, int maxd) nogil: # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t pivot
|
|
* cdef SIZE_t i, l, r
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_introsort(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n, int __pyx_v_maxd) {
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_pivot;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_l;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_r;
|
|
int __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1290
|
|
* cdef SIZE_t i, l, r
|
|
*
|
|
* while n > 1: # <<<<<<<<<<<<<<
|
|
* if maxd <= 0: # max depth limit exceeded ("gone quadratic")
|
|
* heapsort(Xf, samples, n)
|
|
*/
|
|
while (1) {
|
|
__pyx_t_1 = ((__pyx_v_n > 1) != 0);
|
|
if (!__pyx_t_1) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1291
|
|
*
|
|
* while n > 1:
|
|
* if maxd <= 0: # max depth limit exceeded ("gone quadratic") # <<<<<<<<<<<<<<
|
|
* heapsort(Xf, samples, n)
|
|
* return
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_maxd <= 0) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1292
|
|
* while n > 1:
|
|
* if maxd <= 0: # max depth limit exceeded ("gone quadratic")
|
|
* heapsort(Xf, samples, n) # <<<<<<<<<<<<<<
|
|
* return
|
|
* maxd -= 1
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_heapsort(__pyx_v_Xf, __pyx_v_samples, __pyx_v_n);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1293
|
|
* if maxd <= 0: # max depth limit exceeded ("gone quadratic")
|
|
* heapsort(Xf, samples, n)
|
|
* return # <<<<<<<<<<<<<<
|
|
* maxd -= 1
|
|
*
|
|
*/
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1294
|
|
* heapsort(Xf, samples, n)
|
|
* return
|
|
* maxd -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* pivot = median3(Xf, n)
|
|
*/
|
|
__pyx_v_maxd = (__pyx_v_maxd - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1296
|
|
* maxd -= 1
|
|
*
|
|
* pivot = median3(Xf, n) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Three-way partition.
|
|
*/
|
|
__pyx_v_pivot = __pyx_f_7sklearn_4tree_5_tree_median3(__pyx_v_Xf, __pyx_v_n);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1299
|
|
*
|
|
* # Three-way partition.
|
|
* i = l = 0 # <<<<<<<<<<<<<<
|
|
* r = n
|
|
* while i < r:
|
|
*/
|
|
__pyx_v_i = 0;
|
|
__pyx_v_l = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1300
|
|
* # Three-way partition.
|
|
* i = l = 0
|
|
* r = n # <<<<<<<<<<<<<<
|
|
* while i < r:
|
|
* if Xf[i] < pivot:
|
|
*/
|
|
__pyx_v_r = __pyx_v_n;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1301
|
|
* i = l = 0
|
|
* r = n
|
|
* while i < r: # <<<<<<<<<<<<<<
|
|
* if Xf[i] < pivot:
|
|
* swap(Xf, samples, i, l)
|
|
*/
|
|
while (1) {
|
|
__pyx_t_1 = ((__pyx_v_i < __pyx_v_r) != 0);
|
|
if (!__pyx_t_1) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1302
|
|
* r = n
|
|
* while i < r:
|
|
* if Xf[i] < pivot: # <<<<<<<<<<<<<<
|
|
* swap(Xf, samples, i, l)
|
|
* i += 1
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_Xf[__pyx_v_i]) < __pyx_v_pivot) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1303
|
|
* while i < r:
|
|
* if Xf[i] < pivot:
|
|
* swap(Xf, samples, i, l) # <<<<<<<<<<<<<<
|
|
* i += 1
|
|
* l += 1
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_swap(__pyx_v_Xf, __pyx_v_samples, __pyx_v_i, __pyx_v_l);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1304
|
|
* if Xf[i] < pivot:
|
|
* swap(Xf, samples, i, l)
|
|
* i += 1 # <<<<<<<<<<<<<<
|
|
* l += 1
|
|
* elif Xf[i] > pivot:
|
|
*/
|
|
__pyx_v_i = (__pyx_v_i + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1305
|
|
* swap(Xf, samples, i, l)
|
|
* i += 1
|
|
* l += 1 # <<<<<<<<<<<<<<
|
|
* elif Xf[i] > pivot:
|
|
* r -= 1
|
|
*/
|
|
__pyx_v_l = (__pyx_v_l + 1);
|
|
goto __pyx_L8;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1306
|
|
* i += 1
|
|
* l += 1
|
|
* elif Xf[i] > pivot: # <<<<<<<<<<<<<<
|
|
* r -= 1
|
|
* swap(Xf, samples, i, r)
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_Xf[__pyx_v_i]) > __pyx_v_pivot) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1307
|
|
* l += 1
|
|
* elif Xf[i] > pivot:
|
|
* r -= 1 # <<<<<<<<<<<<<<
|
|
* swap(Xf, samples, i, r)
|
|
* else:
|
|
*/
|
|
__pyx_v_r = (__pyx_v_r - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1308
|
|
* elif Xf[i] > pivot:
|
|
* r -= 1
|
|
* swap(Xf, samples, i, r) # <<<<<<<<<<<<<<
|
|
* else:
|
|
* i += 1
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_swap(__pyx_v_Xf, __pyx_v_samples, __pyx_v_i, __pyx_v_r);
|
|
goto __pyx_L8;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1310
|
|
* swap(Xf, samples, i, r)
|
|
* else:
|
|
* i += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* introsort(Xf, samples, l, maxd)
|
|
*/
|
|
__pyx_v_i = (__pyx_v_i + 1);
|
|
}
|
|
__pyx_L8:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1312
|
|
* i += 1
|
|
*
|
|
* introsort(Xf, samples, l, maxd) # <<<<<<<<<<<<<<
|
|
* Xf += r
|
|
* samples += r
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_introsort(__pyx_v_Xf, __pyx_v_samples, __pyx_v_l, __pyx_v_maxd);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1313
|
|
*
|
|
* introsort(Xf, samples, l, maxd)
|
|
* Xf += r # <<<<<<<<<<<<<<
|
|
* samples += r
|
|
* n -= r
|
|
*/
|
|
__pyx_v_Xf = (__pyx_v_Xf + __pyx_v_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1314
|
|
* introsort(Xf, samples, l, maxd)
|
|
* Xf += r
|
|
* samples += r # <<<<<<<<<<<<<<
|
|
* n -= r
|
|
*
|
|
*/
|
|
__pyx_v_samples = (__pyx_v_samples + __pyx_v_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1315
|
|
* Xf += r
|
|
* samples += r
|
|
* n -= r # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_v_n = (__pyx_v_n - __pyx_v_r);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1286
|
|
* # Introsort with median of 3 pivot selection and 3-way partition function
|
|
* # (robust to repeated elements, e.g. lots of zero features).
|
|
* cdef void introsort(DTYPE_t* Xf, SIZE_t *samples, SIZE_t n, int maxd) nogil: # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t pivot
|
|
* cdef SIZE_t i, l, r
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1318
|
|
*
|
|
*
|
|
* cdef inline void sift_down(DTYPE_t* Xf, SIZE_t* samples, # <<<<<<<<<<<<<<
|
|
* SIZE_t start, SIZE_t end) nogil:
|
|
* # Restore heap order in Xf[start:end] by moving the max element to start.
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_sift_down(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_child;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_maxind;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_root;
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1323
|
|
* cdef SIZE_t child, maxind, root
|
|
*
|
|
* root = start # <<<<<<<<<<<<<<
|
|
* while True:
|
|
* child = root * 2 + 1
|
|
*/
|
|
__pyx_v_root = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1324
|
|
*
|
|
* root = start
|
|
* while True: # <<<<<<<<<<<<<<
|
|
* child = root * 2 + 1
|
|
*
|
|
*/
|
|
while (1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1325
|
|
* root = start
|
|
* while True:
|
|
* child = root * 2 + 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # find max of root, left child, right child
|
|
*/
|
|
__pyx_v_child = ((__pyx_v_root * 2) + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1328
|
|
*
|
|
* # find max of root, left child, right child
|
|
* maxind = root # <<<<<<<<<<<<<<
|
|
* if child < end and Xf[maxind] < Xf[child]:
|
|
* maxind = child
|
|
*/
|
|
__pyx_v_maxind = __pyx_v_root;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1329
|
|
* # find max of root, left child, right child
|
|
* maxind = root
|
|
* if child < end and Xf[maxind] < Xf[child]: # <<<<<<<<<<<<<<
|
|
* maxind = child
|
|
* if child + 1 < end and Xf[maxind] < Xf[child + 1]:
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_child < __pyx_v_end) != 0);
|
|
if (__pyx_t_1) {
|
|
__pyx_t_2 = (((__pyx_v_Xf[__pyx_v_maxind]) < (__pyx_v_Xf[__pyx_v_child])) != 0);
|
|
__pyx_t_3 = __pyx_t_2;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_1;
|
|
}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1330
|
|
* maxind = root
|
|
* if child < end and Xf[maxind] < Xf[child]:
|
|
* maxind = child # <<<<<<<<<<<<<<
|
|
* if child + 1 < end and Xf[maxind] < Xf[child + 1]:
|
|
* maxind = child + 1
|
|
*/
|
|
__pyx_v_maxind = __pyx_v_child;
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1331
|
|
* if child < end and Xf[maxind] < Xf[child]:
|
|
* maxind = child
|
|
* if child + 1 < end and Xf[maxind] < Xf[child + 1]: # <<<<<<<<<<<<<<
|
|
* maxind = child + 1
|
|
*
|
|
*/
|
|
__pyx_t_3 = (((__pyx_v_child + 1) < __pyx_v_end) != 0);
|
|
if (__pyx_t_3) {
|
|
__pyx_t_1 = (((__pyx_v_Xf[__pyx_v_maxind]) < (__pyx_v_Xf[(__pyx_v_child + 1)])) != 0);
|
|
__pyx_t_2 = __pyx_t_1;
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_3;
|
|
}
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1332
|
|
* maxind = child
|
|
* if child + 1 < end and Xf[maxind] < Xf[child + 1]:
|
|
* maxind = child + 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* if maxind == root:
|
|
*/
|
|
__pyx_v_maxind = (__pyx_v_child + 1);
|
|
goto __pyx_L6;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1334
|
|
* maxind = child + 1
|
|
*
|
|
* if maxind == root: # <<<<<<<<<<<<<<
|
|
* break
|
|
* else:
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_maxind == __pyx_v_root) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1335
|
|
*
|
|
* if maxind == root:
|
|
* break # <<<<<<<<<<<<<<
|
|
* else:
|
|
* swap(Xf, samples, root, maxind)
|
|
*/
|
|
goto __pyx_L4_break;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1337
|
|
* break
|
|
* else:
|
|
* swap(Xf, samples, root, maxind) # <<<<<<<<<<<<<<
|
|
* root = maxind
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_swap(__pyx_v_Xf, __pyx_v_samples, __pyx_v_root, __pyx_v_maxind);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1338
|
|
* else:
|
|
* swap(Xf, samples, root, maxind)
|
|
* root = maxind # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_v_root = __pyx_v_maxind;
|
|
}
|
|
}
|
|
__pyx_L4_break:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1318
|
|
*
|
|
*
|
|
* cdef inline void sift_down(DTYPE_t* Xf, SIZE_t* samples, # <<<<<<<<<<<<<<
|
|
* SIZE_t start, SIZE_t end) nogil:
|
|
* # Restore heap order in Xf[start:end] by moving the max element to start.
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1341
|
|
*
|
|
*
|
|
* cdef void heapsort(DTYPE_t* Xf, SIZE_t* samples, SIZE_t n) nogil: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start, end
|
|
*
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_heapsort(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
int __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1345
|
|
*
|
|
* # heapify
|
|
* start = (n - 2) / 2 # <<<<<<<<<<<<<<
|
|
* end = n
|
|
* while True:
|
|
*/
|
|
__pyx_v_start = ((__pyx_v_n - 2) / 2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1346
|
|
* # heapify
|
|
* start = (n - 2) / 2
|
|
* end = n # <<<<<<<<<<<<<<
|
|
* while True:
|
|
* sift_down(Xf, samples, start, end)
|
|
*/
|
|
__pyx_v_end = __pyx_v_n;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1347
|
|
* start = (n - 2) / 2
|
|
* end = n
|
|
* while True: # <<<<<<<<<<<<<<
|
|
* sift_down(Xf, samples, start, end)
|
|
* if start == 0:
|
|
*/
|
|
while (1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1348
|
|
* end = n
|
|
* while True:
|
|
* sift_down(Xf, samples, start, end) # <<<<<<<<<<<<<<
|
|
* if start == 0:
|
|
* break
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sift_down(__pyx_v_Xf, __pyx_v_samples, __pyx_v_start, __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1349
|
|
* while True:
|
|
* sift_down(Xf, samples, start, end)
|
|
* if start == 0: # <<<<<<<<<<<<<<
|
|
* break
|
|
* start -= 1
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_start == 0) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1350
|
|
* sift_down(Xf, samples, start, end)
|
|
* if start == 0:
|
|
* break # <<<<<<<<<<<<<<
|
|
* start -= 1
|
|
*
|
|
*/
|
|
goto __pyx_L4_break;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1351
|
|
* if start == 0:
|
|
* break
|
|
* start -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # sort by shrinking the heap, putting the max element immediately after it
|
|
*/
|
|
__pyx_v_start = (__pyx_v_start - 1);
|
|
}
|
|
__pyx_L4_break:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1354
|
|
*
|
|
* # sort by shrinking the heap, putting the max element immediately after it
|
|
* end = n - 1 # <<<<<<<<<<<<<<
|
|
* while end > 0:
|
|
* swap(Xf, samples, 0, end)
|
|
*/
|
|
__pyx_v_end = (__pyx_v_n - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1355
|
|
* # sort by shrinking the heap, putting the max element immediately after it
|
|
* end = n - 1
|
|
* while end > 0: # <<<<<<<<<<<<<<
|
|
* swap(Xf, samples, 0, end)
|
|
* sift_down(Xf, samples, 0, end)
|
|
*/
|
|
while (1) {
|
|
__pyx_t_1 = ((__pyx_v_end > 0) != 0);
|
|
if (!__pyx_t_1) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1356
|
|
* end = n - 1
|
|
* while end > 0:
|
|
* swap(Xf, samples, 0, end) # <<<<<<<<<<<<<<
|
|
* sift_down(Xf, samples, 0, end)
|
|
* end = end - 1
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_swap(__pyx_v_Xf, __pyx_v_samples, 0, __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1357
|
|
* while end > 0:
|
|
* swap(Xf, samples, 0, end)
|
|
* sift_down(Xf, samples, 0, end) # <<<<<<<<<<<<<<
|
|
* end = end - 1
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sift_down(__pyx_v_Xf, __pyx_v_samples, 0, __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1358
|
|
* swap(Xf, samples, 0, end)
|
|
* sift_down(Xf, samples, 0, end)
|
|
* end = end - 1 # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_v_end = (__pyx_v_end - 1);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1341
|
|
*
|
|
*
|
|
* cdef void heapsort(DTYPE_t* Xf, SIZE_t* samples, SIZE_t n) nogil: # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start, end
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1363
|
|
* cdef class RandomSplitter(BaseDenseSplitter):
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (RandomSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_14RandomSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_14RandomSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_14RandomSplitter___reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_14RandomSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1364
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self):
|
|
* return (RandomSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1365
|
|
* def __reduce__(self):
|
|
* return (RandomSplitter, (self.criterion,
|
|
* self.max_features, # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.max_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1365; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1366
|
|
* return (RandomSplitter, (self.criterion,
|
|
* self.max_features,
|
|
* self.min_samples_leaf, # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__())
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1366; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1367
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf, # <<<<<<<<<<<<<<
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyFloat_FromDouble(__pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1367; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1364
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self):
|
|
* return (RandomSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1364; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 3, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 4, __pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__Pyx_GIVEREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1368
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split,
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1364
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self):
|
|
* return (RandomSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1364; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RandomSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RandomSplitter)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RandomSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1363
|
|
* cdef class RandomSplitter(BaseDenseSplitter):
|
|
* """Splitter for finding the best random split."""
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (RandomSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.RandomSplitter.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1370
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best random split on node samples[start:end]."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_14RandomSplitter_node_split(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter *__pyx_v_self, double __pyx_v_impurity, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_constant_features) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_constant_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_sample_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_fx_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_best;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_current;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_j;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_found_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_drawn_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_known_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_visited_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_min_feature_value;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_max_feature_value;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_current_feature_value;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_partition_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_3;
|
|
double __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1374
|
|
* """Find the best random split on node samples[start:end]."""
|
|
* # Draw random splits and pick the best
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1375
|
|
* # Draw random splits and pick the best
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.start;
|
|
__pyx_v_start = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1376
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.end;
|
|
__pyx_v_end = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1378
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
* cdef SIZE_t* features = self.features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.features;
|
|
__pyx_v_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1379
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.constant_features;
|
|
__pyx_v_constant_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1380
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t* X = self.X
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_features;
|
|
__pyx_v_n_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1382
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
* cdef DTYPE_t* X = self.X # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X;
|
|
__pyx_v_X = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1383
|
|
*
|
|
* cdef DTYPE_t* X = self.X
|
|
* cdef DTYPE_t* Xf = self.feature_values # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.__pyx_base.feature_values;
|
|
__pyx_v_Xf = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1384
|
|
* cdef DTYPE_t* X = self.X
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef SIZE_t max_features = self.max_features
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.X_sample_stride;
|
|
__pyx_v_X_sample_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1385
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.X_fx_stride;
|
|
__pyx_v_X_fx_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1386
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef SIZE_t max_features = self.max_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.max_features;
|
|
__pyx_v_max_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1387
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1388
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1389
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SplitRecord best, current
|
|
*/
|
|
__pyx_v_random_state = (&__pyx_v_self->__pyx_base.__pyx_base.rand_r_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1393
|
|
* cdef SplitRecord best, current
|
|
*
|
|
* cdef SIZE_t f_i = n_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* # Number of features discovered to be constant during the split search
|
|
*/
|
|
__pyx_v_f_i = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1396
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
*/
|
|
__pyx_v_n_found_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1398
|
|
* cdef SIZE_t n_found_constants = 0
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
*/
|
|
__pyx_v_n_drawn_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1399
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0] # <<<<<<<<<<<<<<
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
*/
|
|
__pyx_v_n_known_constants = (__pyx_v_n_constant_features[0]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1401
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_visited_features = 0
|
|
* cdef DTYPE_t min_feature_value
|
|
*/
|
|
__pyx_v_n_total_constants = __pyx_v_n_known_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1402
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
* cdef SIZE_t n_visited_features = 0 # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t min_feature_value
|
|
* cdef DTYPE_t max_feature_value
|
|
*/
|
|
__pyx_v_n_visited_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1408
|
|
* cdef SIZE_t partition_end
|
|
*
|
|
* _init_split(&best, end) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Sample up to max_features without replacement using a
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree__init_split((&__pyx_v_best), __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1419
|
|
* # newly discovered constant features to spare computation on descendant
|
|
* # nodes.
|
|
* while (f_i > n_total_constants and # Stop early if remaining features # <<<<<<<<<<<<<<
|
|
* # are constant
|
|
* (n_visited_features < max_features or
|
|
*/
|
|
while (1) {
|
|
__pyx_t_5 = ((__pyx_v_f_i > __pyx_v_n_total_constants) != 0);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1421
|
|
* while (f_i > n_total_constants and # Stop early if remaining features
|
|
* # are constant
|
|
* (n_visited_features < max_features or # <<<<<<<<<<<<<<
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_n_visited_features < __pyx_v_max_features) != 0);
|
|
if (!__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1423
|
|
* (n_visited_features < max_features or
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)): # <<<<<<<<<<<<<<
|
|
* n_visited_features += 1
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_n_visited_features <= (__pyx_v_n_found_constants + __pyx_v_n_drawn_constants)) != 0);
|
|
__pyx_t_8 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_6;
|
|
}
|
|
__pyx_t_6 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_5;
|
|
}
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1424
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
* n_visited_features += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Loop invariant: elements of features in
|
|
*/
|
|
__pyx_v_n_visited_features = (__pyx_v_n_visited_features + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1438
|
|
*
|
|
* # Draw a feature at random
|
|
* f_j = rand_int(n_drawn_constants, f_i - n_found_constants, # <<<<<<<<<<<<<<
|
|
* random_state)
|
|
*
|
|
*/
|
|
__pyx_v_f_j = __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_v_n_drawn_constants, (__pyx_v_f_i - __pyx_v_n_found_constants), __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1441
|
|
* random_state)
|
|
*
|
|
* if f_j < n_known_constants: # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_f_j < __pyx_v_n_known_constants) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1443
|
|
* if f_j < n_known_constants:
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j] # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1444
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants] # <<<<<<<<<<<<<<
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_drawn_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1445
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* n_drawn_constants += 1
|
|
*/
|
|
(__pyx_v_features[__pyx_v_n_drawn_constants]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1447
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
* n_drawn_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_drawn_constants = (__pyx_v_n_drawn_constants + 1);
|
|
goto __pyx_L5;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1451
|
|
* else:
|
|
* # f_j in the interval [n_known_constants, f_i - n_found_constants[
|
|
* f_j += n_found_constants # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
*/
|
|
__pyx_v_f_j = (__pyx_v_f_j + __pyx_v_n_found_constants);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1454
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
* current.feature = features[f_j] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Find min, max
|
|
*/
|
|
__pyx_v_current.feature = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1457
|
|
*
|
|
* # Find min, max
|
|
* min_feature_value = X[X_sample_stride * samples[start] + # <<<<<<<<<<<<<<
|
|
* X_fx_stride * current.feature]
|
|
* max_feature_value = min_feature_value
|
|
*/
|
|
__pyx_v_min_feature_value = (__pyx_v_X[((__pyx_v_X_sample_stride * (__pyx_v_samples[__pyx_v_start])) + (__pyx_v_X_fx_stride * __pyx_v_current.feature))]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1459
|
|
* min_feature_value = X[X_sample_stride * samples[start] +
|
|
* X_fx_stride * current.feature]
|
|
* max_feature_value = min_feature_value # <<<<<<<<<<<<<<
|
|
* Xf[start] = min_feature_value
|
|
*
|
|
*/
|
|
__pyx_v_max_feature_value = __pyx_v_min_feature_value;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1460
|
|
* X_fx_stride * current.feature]
|
|
* max_feature_value = min_feature_value
|
|
* Xf[start] = min_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* for p in range(start + 1, end):
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_start]) = __pyx_v_min_feature_value;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1462
|
|
* Xf[start] = min_feature_value
|
|
*
|
|
* for p in range(start + 1, end): # <<<<<<<<<<<<<<
|
|
* current_feature_value = X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * current.feature]
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end;
|
|
for (__pyx_t_9 = (__pyx_v_start + 1); __pyx_t_9 < __pyx_t_2; __pyx_t_9+=1) {
|
|
__pyx_v_p = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1463
|
|
*
|
|
* for p in range(start + 1, end):
|
|
* current_feature_value = X[X_sample_stride * samples[p] + # <<<<<<<<<<<<<<
|
|
* X_fx_stride * current.feature]
|
|
* Xf[p] = current_feature_value
|
|
*/
|
|
__pyx_v_current_feature_value = (__pyx_v_X[((__pyx_v_X_sample_stride * (__pyx_v_samples[__pyx_v_p])) + (__pyx_v_X_fx_stride * __pyx_v_current.feature))]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1465
|
|
* current_feature_value = X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * current.feature]
|
|
* Xf[p] = current_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* if current_feature_value < min_feature_value:
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_p]) = __pyx_v_current_feature_value;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1467
|
|
* Xf[p] = current_feature_value
|
|
*
|
|
* if current_feature_value < min_feature_value: # <<<<<<<<<<<<<<
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value:
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current_feature_value < __pyx_v_min_feature_value) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1468
|
|
*
|
|
* if current_feature_value < min_feature_value:
|
|
* min_feature_value = current_feature_value # <<<<<<<<<<<<<<
|
|
* elif current_feature_value > max_feature_value:
|
|
* max_feature_value = current_feature_value
|
|
*/
|
|
__pyx_v_min_feature_value = __pyx_v_current_feature_value;
|
|
goto __pyx_L8;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1469
|
|
* if current_feature_value < min_feature_value:
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value: # <<<<<<<<<<<<<<
|
|
* max_feature_value = current_feature_value
|
|
*
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current_feature_value > __pyx_v_max_feature_value) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1470
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value:
|
|
* max_feature_value = current_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD:
|
|
*/
|
|
__pyx_v_max_feature_value = __pyx_v_current_feature_value;
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1472
|
|
* max_feature_value = current_feature_value
|
|
*
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD: # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_max_feature_value <= (__pyx_v_min_feature_value + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1473
|
|
*
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants] # <<<<<<<<<<<<<<
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_total_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1474
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature # <<<<<<<<<<<<<<
|
|
*
|
|
* n_found_constants += 1
|
|
*/
|
|
__pyx_t_2 = __pyx_v_current.feature;
|
|
(__pyx_v_features[__pyx_v_n_total_constants]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1476
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
* n_found_constants += 1 # <<<<<<<<<<<<<<
|
|
* n_total_constants += 1
|
|
*
|
|
*/
|
|
__pyx_v_n_found_constants = (__pyx_v_n_found_constants + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1477
|
|
*
|
|
* n_found_constants += 1
|
|
* n_total_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_total_constants = (__pyx_v_n_total_constants + 1);
|
|
goto __pyx_L9;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1480
|
|
*
|
|
* else:
|
|
* f_i -= 1 # <<<<<<<<<<<<<<
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i]
|
|
*
|
|
*/
|
|
__pyx_v_f_i = (__pyx_v_f_i - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1481
|
|
* else:
|
|
* f_i -= 1
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Draw a random threshold
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_features[__pyx_v_f_j]);
|
|
__pyx_t_9 = (__pyx_v_features[__pyx_v_f_i]);
|
|
(__pyx_v_features[__pyx_v_f_i]) = __pyx_t_2;
|
|
(__pyx_v_features[__pyx_v_f_j]) = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1484
|
|
*
|
|
* # Draw a random threshold
|
|
* current.threshold = rand_uniform(min_feature_value, # <<<<<<<<<<<<<<
|
|
* max_feature_value,
|
|
* random_state)
|
|
*/
|
|
__pyx_v_current.threshold = __pyx_f_7sklearn_4tree_5_tree_rand_uniform(__pyx_v_min_feature_value, __pyx_v_max_feature_value, __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1488
|
|
* random_state)
|
|
*
|
|
* if current.threshold == max_feature_value: # <<<<<<<<<<<<<<
|
|
* current.threshold = min_feature_value
|
|
*
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current.threshold == __pyx_v_max_feature_value) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1489
|
|
*
|
|
* if current.threshold == max_feature_value:
|
|
* current.threshold = min_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* # Partition
|
|
*/
|
|
__pyx_v_current.threshold = __pyx_v_min_feature_value;
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1492
|
|
*
|
|
* # Partition
|
|
* partition_end = end # <<<<<<<<<<<<<<
|
|
* p = start
|
|
* while p < partition_end:
|
|
*/
|
|
__pyx_v_partition_end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1493
|
|
* # Partition
|
|
* partition_end = end
|
|
* p = start # <<<<<<<<<<<<<<
|
|
* while p < partition_end:
|
|
* current_feature_value = Xf[p]
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1494
|
|
* partition_end = end
|
|
* p = start
|
|
* while p < partition_end: # <<<<<<<<<<<<<<
|
|
* current_feature_value = Xf[p]
|
|
* if current_feature_value <= current.threshold:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = ((__pyx_v_p < __pyx_v_partition_end) != 0);
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1495
|
|
* p = start
|
|
* while p < partition_end:
|
|
* current_feature_value = Xf[p] # <<<<<<<<<<<<<<
|
|
* if current_feature_value <= current.threshold:
|
|
* p += 1
|
|
*/
|
|
__pyx_v_current_feature_value = (__pyx_v_Xf[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1496
|
|
* while p < partition_end:
|
|
* current_feature_value = Xf[p]
|
|
* if current_feature_value <= current.threshold: # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
* else:
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current_feature_value <= __pyx_v_current.threshold) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1497
|
|
* current_feature_value = Xf[p]
|
|
* if current_feature_value <= current.threshold:
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* partition_end -= 1
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
goto __pyx_L13;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1499
|
|
* p += 1
|
|
* else:
|
|
* partition_end -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* Xf[p] = Xf[partition_end]
|
|
*/
|
|
__pyx_v_partition_end = (__pyx_v_partition_end - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1501
|
|
* partition_end -= 1
|
|
*
|
|
* Xf[p] = Xf[partition_end] # <<<<<<<<<<<<<<
|
|
* Xf[partition_end] = current_feature_value
|
|
*
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_p]) = (__pyx_v_Xf[__pyx_v_partition_end]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1502
|
|
*
|
|
* Xf[p] = Xf[partition_end]
|
|
* Xf[partition_end] = current_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* tmp = samples[partition_end]
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_partition_end]) = __pyx_v_current_feature_value;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1504
|
|
* Xf[partition_end] = current_feature_value
|
|
*
|
|
* tmp = samples[partition_end] # <<<<<<<<<<<<<<
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_samples[__pyx_v_partition_end]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1505
|
|
*
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p] # <<<<<<<<<<<<<<
|
|
* samples[p] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_partition_end]) = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1506
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* current.pos = partition_end
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_p]) = __pyx_v_tmp;
|
|
}
|
|
__pyx_L13:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1508
|
|
* samples[p] = tmp
|
|
*
|
|
* current.pos = partition_end # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
*/
|
|
__pyx_v_current.pos = __pyx_v_partition_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1511
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_current.pos - __pyx_v_start) < __pyx_v_min_samples_leaf) != 0);
|
|
if (!__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1512
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_5 = (((__pyx_v_end - __pyx_v_current.pos) < __pyx_v_min_samples_leaf) != 0);
|
|
__pyx_t_8 = __pyx_t_5;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_6;
|
|
}
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1513
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* # Evaluate split
|
|
*/
|
|
goto __pyx_L3_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1516
|
|
*
|
|
* # Evaluate split
|
|
* self.criterion.reset() # <<<<<<<<<<<<<<
|
|
* self.criterion.update(current.pos)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->reset(__pyx_v_self->__pyx_base.__pyx_base.criterion);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1517
|
|
* # Evaluate split
|
|
* self.criterion.reset()
|
|
* self.criterion.update(current.pos) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->update(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_current.pos);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1520
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or # <<<<<<<<<<<<<<
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_left < __pyx_v_min_weight_leaf) != 0);
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1521
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_right < __pyx_v_min_weight_leaf) != 0);
|
|
__pyx_t_5 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_5 = __pyx_t_8;
|
|
}
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1522
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*/
|
|
goto __pyx_L3_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1524
|
|
* continue
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity) # <<<<<<<<<<<<<<
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
*/
|
|
__pyx_v_current.improvement = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->impurity_improvement(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_impurity);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1526
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*
|
|
* if current.improvement > best.improvement: # <<<<<<<<<<<<<<
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
*/
|
|
__pyx_t_5 = ((__pyx_v_current.improvement > __pyx_v_best.improvement) != 0);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1527
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
* self.criterion.children_impurity(¤t.impurity_left, # <<<<<<<<<<<<<<
|
|
* ¤t.impurity_right)
|
|
* best = current # copy
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->children_impurity(__pyx_v_self->__pyx_base.__pyx_base.criterion, (&__pyx_v_current.impurity_left), (&__pyx_v_current.impurity_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1529
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
* best = current # copy # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
*/
|
|
__pyx_v_best = __pyx_v_current;
|
|
goto __pyx_L16;
|
|
}
|
|
__pyx_L16:;
|
|
}
|
|
__pyx_L9:;
|
|
}
|
|
__pyx_L5:;
|
|
__pyx_L3_continue:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1532
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end and current.feature != best.feature: # <<<<<<<<<<<<<<
|
|
* partition_end = end
|
|
* p = start
|
|
*/
|
|
__pyx_t_5 = ((__pyx_v_best.pos < __pyx_v_end) != 0);
|
|
if (__pyx_t_5) {
|
|
__pyx_t_8 = ((__pyx_v_current.feature != __pyx_v_best.feature) != 0);
|
|
__pyx_t_6 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_5;
|
|
}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1533
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end and current.feature != best.feature:
|
|
* partition_end = end # <<<<<<<<<<<<<<
|
|
* p = start
|
|
*
|
|
*/
|
|
__pyx_v_partition_end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1534
|
|
* if best.pos < end and current.feature != best.feature:
|
|
* partition_end = end
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < partition_end:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1536
|
|
* p = start
|
|
*
|
|
* while p < partition_end: # <<<<<<<<<<<<<<
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = ((__pyx_v_p < __pyx_v_partition_end) != 0);
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1538
|
|
* while p < partition_end:
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold: # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_X[((__pyx_v_X_sample_stride * (__pyx_v_samples[__pyx_v_p])) + (__pyx_v_X_fx_stride * __pyx_v_best.feature))]) <= __pyx_v_best.threshold) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1539
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold:
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
goto __pyx_L20;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1542
|
|
*
|
|
* else:
|
|
* partition_end -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* tmp = samples[partition_end]
|
|
*/
|
|
__pyx_v_partition_end = (__pyx_v_partition_end - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1544
|
|
* partition_end -= 1
|
|
*
|
|
* tmp = samples[partition_end] # <<<<<<<<<<<<<<
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_samples[__pyx_v_partition_end]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1545
|
|
*
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p] # <<<<<<<<<<<<<<
|
|
* samples[p] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_partition_end]) = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1546
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* # Respect invariant for constant features: the original order of
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_p]) = __pyx_v_tmp;
|
|
}
|
|
__pyx_L20:;
|
|
}
|
|
goto __pyx_L17;
|
|
}
|
|
__pyx_L17:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1551
|
|
* # element in features[:n_known_constants] must be preserved for sibling
|
|
* # and child nodes
|
|
* memcpy(features, constant_features, sizeof(SIZE_t) * n_known_constants) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Copy newly found constant features
|
|
*/
|
|
memcpy(__pyx_v_features, __pyx_v_constant_features, ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_known_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1554
|
|
*
|
|
* # Copy newly found constant features
|
|
* memcpy(constant_features + n_known_constants, # <<<<<<<<<<<<<<
|
|
* features + n_known_constants,
|
|
* sizeof(SIZE_t) * n_found_constants)
|
|
*/
|
|
memcpy((__pyx_v_constant_features + __pyx_v_n_known_constants), (__pyx_v_features + __pyx_v_n_known_constants), ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_found_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1559
|
|
*
|
|
* # Return values
|
|
* split[0] = best # <<<<<<<<<<<<<<
|
|
* n_constant_features[0] = n_total_constants
|
|
*
|
|
*/
|
|
(__pyx_v_split[0]) = __pyx_v_best;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1560
|
|
* # Return values
|
|
* split[0] = best
|
|
* n_constant_features[0] = n_total_constants # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_n_constant_features[0]) = __pyx_v_n_total_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1370
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best random split on node samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1573
|
|
* cdef unsigned char* sample_mask
|
|
*
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf,
|
|
* double min_weight_leaf,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
CYTHON_UNUSED double __pyx_v_min_weight_leaf;
|
|
CYTHON_UNUSED PyObject *__pyx_v_random_state = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_criterion,&__pyx_n_s_max_features,&__pyx_n_s_min_samples_leaf,&__pyx_n_s_min_weight_leaf,&__pyx_n_s_random_state,0};
|
|
PyObject* values[5] = {0,0,0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_criterion)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_features)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (likely((values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_weight_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 4:
|
|
if (likely((values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_random_state)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 5) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
}
|
|
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[0]);
|
|
__pyx_v_max_features = __Pyx_PyInt_As_Py_intptr_t(values[1]); if (unlikely((__pyx_v_max_features == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_samples_leaf = __Pyx_PyInt_As_Py_intptr_t(values[2]); if (unlikely((__pyx_v_min_samples_leaf == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1574; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_weight_leaf = __pyx_PyFloat_AsDouble(values[3]); if (unlikely((__pyx_v_min_weight_leaf == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1575; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_random_state = values[4];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.PresortBestSplitter.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_criterion), __pyx_ptype_7sklearn_4tree_5_tree_Criterion, 1, "criterion", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1573; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)__pyx_v_self), __pyx_v_criterion, __pyx_v_max_features, __pyx_v_min_samples_leaf, __pyx_v_min_weight_leaf, __pyx_v_random_state);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, CYTHON_UNUSED double __pyx_v_min_weight_leaf, CYTHON_UNUSED PyObject *__pyx_v_random_state) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1578
|
|
* object random_state):
|
|
* # Initialize pointers
|
|
* self.X_old = NULL # <<<<<<<<<<<<<<
|
|
* self.X_argsorted_ptr = NULL
|
|
* self.X_argsorted_stride = 0
|
|
*/
|
|
__pyx_v_self->X_old = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1579
|
|
* # Initialize pointers
|
|
* self.X_old = NULL
|
|
* self.X_argsorted_ptr = NULL # <<<<<<<<<<<<<<
|
|
* self.X_argsorted_stride = 0
|
|
* self.sample_mask = NULL
|
|
*/
|
|
__pyx_v_self->X_argsorted_ptr = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1580
|
|
* self.X_old = NULL
|
|
* self.X_argsorted_ptr = NULL
|
|
* self.X_argsorted_stride = 0 # <<<<<<<<<<<<<<
|
|
* self.sample_mask = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->X_argsorted_stride = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1581
|
|
* self.X_argsorted_ptr = NULL
|
|
* self.X_argsorted_stride = 0
|
|
* self.sample_mask = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* def __dealloc__(self):
|
|
*/
|
|
__pyx_v_self->sample_mask = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1573
|
|
* cdef unsigned char* sample_mask
|
|
*
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf,
|
|
* double min_weight_leaf,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1583
|
|
* self.sample_mask = NULL
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.sample_mask)
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_3__dealloc__(PyObject *__pyx_v_self); /*proto*/
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_3__dealloc__(PyObject *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0);
|
|
__pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter_2__dealloc__(((struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1585
|
|
* def __dealloc__(self):
|
|
* """Destructor."""
|
|
* free(self.sample_mask) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __reduce__(self):
|
|
*/
|
|
free(__pyx_v_self->sample_mask);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1583
|
|
* self.sample_mask = NULL
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* free(self.sample_mask)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1587
|
|
* free(self.sample_mask)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (PresortBestSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter_4__reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19PresortBestSplitter_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1588
|
|
*
|
|
* def __reduce__(self):
|
|
* return (PresortBestSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1589
|
|
* def __reduce__(self):
|
|
* return (PresortBestSplitter, (self.criterion,
|
|
* self.max_features, # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.max_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1589; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1590
|
|
* return (PresortBestSplitter, (self.criterion,
|
|
* self.max_features,
|
|
* self.min_samples_leaf, # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__())
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1591
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf, # <<<<<<<<<<<<<<
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyFloat_FromDouble(__pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1591; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1588
|
|
*
|
|
* def __reduce__(self):
|
|
* return (PresortBestSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1588; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 3, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 4, __pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__Pyx_GIVEREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1592
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void init(self, object X,
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1588
|
|
*
|
|
* def __reduce__(self):
|
|
* return (PresortBestSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1588; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_PresortBestSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_PresortBestSplitter)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_PresortBestSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1587
|
|
* free(self.sample_mask)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (PresortBestSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.PresortBestSplitter.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1594
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void init(self, object X, # <<<<<<<<<<<<<<
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
* DOUBLE_t* sample_weight):
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19PresortBestSplitter_init(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight) {
|
|
void *__pyx_v_sample_mask;
|
|
PyArrayObject *__pyx_v_X_ndarray = 0;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_y;
|
|
__Pyx_Buffer __pyx_pybuffer_y;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_3;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyObject *__pyx_t_6 = NULL;
|
|
PyObject *__pyx_t_7 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_8;
|
|
unsigned char *__pyx_t_9;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("init", 0);
|
|
__pyx_pybuffer_y.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_y.refcount = 0;
|
|
__pyx_pybuffernd_y.data = NULL;
|
|
__pyx_pybuffernd_y.rcbuffer = &__pyx_pybuffer_y;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_y.rcbuffer->pybuffer, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DOUBLE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1594; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_y.diminfo[0].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_y.diminfo[0].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[0]; __pyx_pybuffernd_y.diminfo[1].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[1]; __pyx_pybuffernd_y.diminfo[1].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[1];
|
|
|
|
/* "sklearn/tree/_tree.pyx":1598
|
|
* DOUBLE_t* sample_weight):
|
|
*
|
|
* cdef void* sample_mask = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* # Call parent initializer
|
|
*/
|
|
__pyx_v_sample_mask = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1601
|
|
*
|
|
* # Call parent initializer
|
|
* BaseDenseSplitter.init(self, X, y, sample_weight) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef np.ndarray X_ndarray = X
|
|
*/
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter->__pyx_base.init(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), __pyx_v_X, ((PyArrayObject *)__pyx_v_y), __pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1603
|
|
* BaseDenseSplitter.init(self, X, y, sample_weight)
|
|
*
|
|
* cdef np.ndarray X_ndarray = X # <<<<<<<<<<<<<<
|
|
*
|
|
* # Pre-sort X
|
|
*/
|
|
if (!(likely(((__pyx_v_X) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_X, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1603; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_1 = __pyx_v_X;
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__pyx_v_X_ndarray = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1606
|
|
*
|
|
* # Pre-sort X
|
|
* if self.X_old != self.X: # <<<<<<<<<<<<<<
|
|
* self.X_old = self.X
|
|
* self.X_argsorted = np.asfortranarray(np.argsort(X_ndarray, axis=0),
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_self->X_old != __pyx_v_self->__pyx_base.X) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1607
|
|
* # Pre-sort X
|
|
* if self.X_old != self.X:
|
|
* self.X_old = self.X # <<<<<<<<<<<<<<
|
|
* self.X_argsorted = np.asfortranarray(np.argsort(X_ndarray, axis=0),
|
|
* dtype=np.int32)
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X;
|
|
__pyx_v_self->X_old = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1608
|
|
* if self.X_old != self.X:
|
|
* self.X_old = self.X
|
|
* self.X_argsorted = np.asfortranarray(np.argsort(X_ndarray, axis=0), # <<<<<<<<<<<<<<
|
|
* dtype=np.int32)
|
|
* self.X_argsorted_ptr = <INT32_t*> self.X_argsorted.data
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_asfortranarray); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_argsort); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_X_ndarray));
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, ((PyObject *)__pyx_v_X_ndarray));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_X_ndarray));
|
|
__pyx_t_6 = PyDict_New(); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
if (PyDict_SetItem(__pyx_t_6, __pyx_n_s_axis, __pyx_int_0) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_7 = __Pyx_PyObject_Call(__pyx_t_5, __pyx_t_1, __pyx_t_6); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_7);
|
|
__Pyx_GIVEREF(__pyx_t_7);
|
|
__pyx_t_7 = 0;
|
|
__pyx_t_7 = PyDict_New(); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1609
|
|
* self.X_old = self.X
|
|
* self.X_argsorted = np.asfortranarray(np.argsort(X_ndarray, axis=0),
|
|
* dtype=np.int32) # <<<<<<<<<<<<<<
|
|
* self.X_argsorted_ptr = <INT32_t*> self.X_argsorted.data
|
|
* self.X_argsorted_stride = (<SIZE_t> self.X_argsorted.strides[1] /
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1609; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_int32); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1609; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (PyDict_SetItem(__pyx_t_7, __pyx_n_s_dtype, __pyx_t_5) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1608
|
|
* if self.X_old != self.X:
|
|
* self.X_old = self.X
|
|
* self.X_argsorted = np.asfortranarray(np.argsort(X_ndarray, axis=0), # <<<<<<<<<<<<<<
|
|
* dtype=np.int32)
|
|
* self.X_argsorted_ptr = <INT32_t*> self.X_argsorted.data
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_Call(__pyx_t_4, __pyx_t_6, __pyx_t_7); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
if (!(likely(((__pyx_t_5) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_5, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GIVEREF(__pyx_t_5);
|
|
__Pyx_GOTREF(__pyx_v_self->X_argsorted);
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_self->X_argsorted));
|
|
__pyx_v_self->X_argsorted = ((PyArrayObject *)__pyx_t_5);
|
|
__pyx_t_5 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1610
|
|
* self.X_argsorted = np.asfortranarray(np.argsort(X_ndarray, axis=0),
|
|
* dtype=np.int32)
|
|
* self.X_argsorted_ptr = <INT32_t*> self.X_argsorted.data # <<<<<<<<<<<<<<
|
|
* self.X_argsorted_stride = (<SIZE_t> self.X_argsorted.strides[1] /
|
|
* <SIZE_t> self.X_argsorted.itemsize)
|
|
*/
|
|
__pyx_v_self->X_argsorted_ptr = ((__pyx_t_7sklearn_4tree_5_tree_INT32_t *)__pyx_v_self->X_argsorted->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1612
|
|
* self.X_argsorted_ptr = <INT32_t*> self.X_argsorted.data
|
|
* self.X_argsorted_stride = (<SIZE_t> self.X_argsorted.strides[1] /
|
|
* <SIZE_t> self.X_argsorted.itemsize) # <<<<<<<<<<<<<<
|
|
*
|
|
* self.n_total_samples = X.shape[0]
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self->X_argsorted), __pyx_n_s_itemsize); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1612; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_8 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_5); if (unlikely((__pyx_t_8 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1612; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1611
|
|
* dtype=np.int32)
|
|
* self.X_argsorted_ptr = <INT32_t*> self.X_argsorted.data
|
|
* self.X_argsorted_stride = (<SIZE_t> self.X_argsorted.strides[1] / # <<<<<<<<<<<<<<
|
|
* <SIZE_t> self.X_argsorted.itemsize)
|
|
*
|
|
*/
|
|
__pyx_v_self->X_argsorted_stride = (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_self->X_argsorted->strides[1])) / ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_8));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1614
|
|
* <SIZE_t> self.X_argsorted.itemsize)
|
|
*
|
|
* self.n_total_samples = X.shape[0] # <<<<<<<<<<<<<<
|
|
* sample_mask = safe_realloc(&self.sample_mask, self.n_total_samples)
|
|
* memset(sample_mask, 0, self.n_total_samples)
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1614; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_7 = __Pyx_GetItemInt(__pyx_t_5, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_7 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1614; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_8 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_7); if (unlikely((__pyx_t_8 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1614; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_v_self->n_total_samples = __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1615
|
|
*
|
|
* self.n_total_samples = X.shape[0]
|
|
* sample_mask = safe_realloc(&self.sample_mask, self.n_total_samples) # <<<<<<<<<<<<<<
|
|
* memset(sample_mask, 0, self.n_total_samples)
|
|
*
|
|
*/
|
|
__pyx_t_9 = __pyx_fuse_2__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->sample_mask), __pyx_v_self->n_total_samples); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1615; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_sample_mask = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1616
|
|
* self.n_total_samples = X.shape[0]
|
|
* sample_mask = safe_realloc(&self.sample_mask, self.n_total_samples)
|
|
* memset(sample_mask, 0, self.n_total_samples) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split,
|
|
*/
|
|
memset(__pyx_v_sample_mask, 0, __pyx_v_self->n_total_samples);
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1594
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void init(self, object X, # <<<<<<<<<<<<<<
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
* DOUBLE_t* sample_weight):
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_XDECREF(__pyx_t_6);
|
|
__Pyx_XDECREF(__pyx_t_7);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.tree._tree.PresortBestSplitter.init", __pyx_clineno, __pyx_lineno, __pyx_filename, 0);
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_X_ndarray);
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1618
|
|
* memset(sample_mask, 0, self.n_total_samples)
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best split on node samples[start:end]."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_19PresortBestSplitter_node_split(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *__pyx_v_self, double __pyx_v_impurity, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_constant_features) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_constant_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_sample_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_fx_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_argsorted;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_argsorted_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_samples;
|
|
unsigned char *__pyx_v_sample_mask;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_best;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_current;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_j;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_found_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_drawn_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_known_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_visited_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_partition_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_j;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_t_4;
|
|
unsigned char *__pyx_t_5;
|
|
double __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
int __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1622
|
|
* """Find the best split on node samples[start:end]."""
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1623
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.start;
|
|
__pyx_v_start = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1624
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.end;
|
|
__pyx_v_end = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1626
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
* cdef SIZE_t* features = self.features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.features;
|
|
__pyx_v_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1627
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.constant_features;
|
|
__pyx_v_constant_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1628
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t* X = self.X
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_features;
|
|
__pyx_v_n_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1630
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
* cdef DTYPE_t* X = self.X # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X;
|
|
__pyx_v_X = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1631
|
|
*
|
|
* cdef DTYPE_t* X = self.X
|
|
* cdef DTYPE_t* Xf = self.feature_values # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.__pyx_base.feature_values;
|
|
__pyx_v_Xf = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1632
|
|
* cdef DTYPE_t* X = self.X
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef INT32_t* X_argsorted = self.X_argsorted_ptr
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.X_sample_stride;
|
|
__pyx_v_X_sample_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1633
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride # <<<<<<<<<<<<<<
|
|
* cdef INT32_t* X_argsorted = self.X_argsorted_ptr
|
|
* cdef SIZE_t X_argsorted_stride = self.X_argsorted_stride
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.X_fx_stride;
|
|
__pyx_v_X_fx_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1634
|
|
* cdef SIZE_t X_sample_stride = self.X_sample_stride
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef INT32_t* X_argsorted = self.X_argsorted_ptr # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_argsorted_stride = self.X_argsorted_stride
|
|
* cdef SIZE_t n_total_samples = self.n_total_samples
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->X_argsorted_ptr;
|
|
__pyx_v_X_argsorted = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1635
|
|
* cdef SIZE_t X_fx_stride = self.X_fx_stride
|
|
* cdef INT32_t* X_argsorted = self.X_argsorted_ptr
|
|
* cdef SIZE_t X_argsorted_stride = self.X_argsorted_stride # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_total_samples = self.n_total_samples
|
|
* cdef unsigned char* sample_mask = self.sample_mask
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->X_argsorted_stride;
|
|
__pyx_v_X_argsorted_stride = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1636
|
|
* cdef INT32_t* X_argsorted = self.X_argsorted_ptr
|
|
* cdef SIZE_t X_argsorted_stride = self.X_argsorted_stride
|
|
* cdef SIZE_t n_total_samples = self.n_total_samples # <<<<<<<<<<<<<<
|
|
* cdef unsigned char* sample_mask = self.sample_mask
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->n_total_samples;
|
|
__pyx_v_n_total_samples = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1637
|
|
* cdef SIZE_t X_argsorted_stride = self.X_argsorted_stride
|
|
* cdef SIZE_t n_total_samples = self.n_total_samples
|
|
* cdef unsigned char* sample_mask = self.sample_mask # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t max_features = self.max_features
|
|
*/
|
|
__pyx_t_5 = __pyx_v_self->sample_mask;
|
|
__pyx_v_sample_mask = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1639
|
|
* cdef unsigned char* sample_mask = self.sample_mask
|
|
*
|
|
* cdef SIZE_t max_features = self.max_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.max_features;
|
|
__pyx_v_max_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1640
|
|
*
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1641
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*
|
|
*/
|
|
__pyx_t_6 = __pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1642
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SplitRecord best, current
|
|
*/
|
|
__pyx_v_random_state = (&__pyx_v_self->__pyx_base.__pyx_base.rand_r_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1646
|
|
* cdef SplitRecord best, current
|
|
*
|
|
* cdef SIZE_t f_i = n_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t f_j, p
|
|
* # Number of features discovered to be constant during the split search
|
|
*/
|
|
__pyx_v_f_i = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1649
|
|
* cdef SIZE_t f_j, p
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
*/
|
|
__pyx_v_n_found_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1651
|
|
* cdef SIZE_t n_found_constants = 0
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
*/
|
|
__pyx_v_n_drawn_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1652
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0] # <<<<<<<<<<<<<<
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
*/
|
|
__pyx_v_n_known_constants = (__pyx_v_n_constant_features[0]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1654
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_visited_features = 0
|
|
* cdef SIZE_t partition_end
|
|
*/
|
|
__pyx_v_n_total_constants = __pyx_v_n_known_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1655
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
* cdef SIZE_t n_visited_features = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t partition_end
|
|
* cdef SIZE_t i, j
|
|
*/
|
|
__pyx_v_n_visited_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1659
|
|
* cdef SIZE_t i, j
|
|
*
|
|
* _init_split(&best, end) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Set sample mask
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree__init_split((&__pyx_v_best), __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1662
|
|
*
|
|
* # Set sample mask
|
|
* for p in range(start, end): # <<<<<<<<<<<<<<
|
|
* sample_mask[samples[p]] = 1
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end;
|
|
for (__pyx_t_7 = __pyx_v_start; __pyx_t_7 < __pyx_t_2; __pyx_t_7+=1) {
|
|
__pyx_v_p = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1663
|
|
* # Set sample mask
|
|
* for p in range(start, end):
|
|
* sample_mask[samples[p]] = 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Sample up to max_features without replacement using a
|
|
*/
|
|
(__pyx_v_sample_mask[(__pyx_v_samples[__pyx_v_p])]) = 1;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1674
|
|
* # newly discovered constant features to spare computation on descendant
|
|
* # nodes.
|
|
* while (f_i > n_total_constants and # Stop early if remaining features # <<<<<<<<<<<<<<
|
|
* # are constant
|
|
* (n_visited_features < max_features or
|
|
*/
|
|
while (1) {
|
|
__pyx_t_8 = ((__pyx_v_f_i > __pyx_v_n_total_constants) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1676
|
|
* while (f_i > n_total_constants and # Stop early if remaining features
|
|
* # are constant
|
|
* (n_visited_features < max_features or # <<<<<<<<<<<<<<
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_n_visited_features < __pyx_v_max_features) != 0);
|
|
if (!__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1678
|
|
* (n_visited_features < max_features or
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)): # <<<<<<<<<<<<<<
|
|
* n_visited_features += 1
|
|
*
|
|
*/
|
|
__pyx_t_10 = ((__pyx_v_n_visited_features <= (__pyx_v_n_found_constants + __pyx_v_n_drawn_constants)) != 0);
|
|
__pyx_t_11 = __pyx_t_10;
|
|
} else {
|
|
__pyx_t_11 = __pyx_t_9;
|
|
}
|
|
__pyx_t_9 = __pyx_t_11;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_8;
|
|
}
|
|
if (!__pyx_t_9) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1679
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
* n_visited_features += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Loop invariant: elements of features in
|
|
*/
|
|
__pyx_v_n_visited_features = (__pyx_v_n_visited_features + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1693
|
|
*
|
|
* # Draw a feature at random
|
|
* f_j = rand_int(n_drawn_constants, f_i - n_found_constants, # <<<<<<<<<<<<<<
|
|
* random_state)
|
|
*
|
|
*/
|
|
__pyx_v_f_j = __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_v_n_drawn_constants, (__pyx_v_f_i - __pyx_v_n_found_constants), __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1696
|
|
* random_state)
|
|
*
|
|
* if f_j < n_known_constants: # <<<<<<<<<<<<<<
|
|
* # f_j is in [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_f_j < __pyx_v_n_known_constants) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1698
|
|
* if f_j < n_known_constants:
|
|
* # f_j is in [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j] # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1699
|
|
* # f_j is in [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants] # <<<<<<<<<<<<<<
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_drawn_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1700
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* n_drawn_constants += 1
|
|
*/
|
|
(__pyx_v_features[__pyx_v_n_drawn_constants]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1702
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
* n_drawn_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_drawn_constants = (__pyx_v_n_drawn_constants + 1);
|
|
goto __pyx_L7;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1706
|
|
* else:
|
|
* # f_j in the interval [n_known_constants, f_i - n_found_constants[
|
|
* f_j += n_found_constants # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
*/
|
|
__pyx_v_f_j = (__pyx_v_f_j + __pyx_v_n_found_constants);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1709
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
* current.feature = features[f_j] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Extract ordering from X_argsorted
|
|
*/
|
|
__pyx_v_current.feature = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1712
|
|
*
|
|
* # Extract ordering from X_argsorted
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* for i in range(n_total_samples):
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1714
|
|
* p = start
|
|
*
|
|
* for i in range(n_total_samples): # <<<<<<<<<<<<<<
|
|
* j = X_argsorted[X_argsorted_stride * current.feature + i]
|
|
* if sample_mask[j] == 1:
|
|
*/
|
|
__pyx_t_2 = __pyx_v_n_total_samples;
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_2; __pyx_t_7+=1) {
|
|
__pyx_v_i = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1715
|
|
*
|
|
* for i in range(n_total_samples):
|
|
* j = X_argsorted[X_argsorted_stride * current.feature + i] # <<<<<<<<<<<<<<
|
|
* if sample_mask[j] == 1:
|
|
* samples[p] = j
|
|
*/
|
|
__pyx_v_j = (__pyx_v_X_argsorted[((__pyx_v_X_argsorted_stride * __pyx_v_current.feature) + __pyx_v_i)]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1716
|
|
* for i in range(n_total_samples):
|
|
* j = X_argsorted[X_argsorted_stride * current.feature + i]
|
|
* if sample_mask[j] == 1: # <<<<<<<<<<<<<<
|
|
* samples[p] = j
|
|
* Xf[p] = X[X_sample_stride * j +
|
|
*/
|
|
__pyx_t_9 = (((__pyx_v_sample_mask[__pyx_v_j]) == 1) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1717
|
|
* j = X_argsorted[X_argsorted_stride * current.feature + i]
|
|
* if sample_mask[j] == 1:
|
|
* samples[p] = j # <<<<<<<<<<<<<<
|
|
* Xf[p] = X[X_sample_stride * j +
|
|
* X_fx_stride * current.feature]
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_p]) = __pyx_v_j;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1718
|
|
* if sample_mask[j] == 1:
|
|
* samples[p] = j
|
|
* Xf[p] = X[X_sample_stride * j + # <<<<<<<<<<<<<<
|
|
* X_fx_stride * current.feature]
|
|
* p += 1
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_p]) = (__pyx_v_X[((__pyx_v_X_sample_stride * __pyx_v_j) + (__pyx_v_X_fx_stride * __pyx_v_current.feature))]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1720
|
|
* Xf[p] = X[X_sample_stride * j +
|
|
* X_fx_stride * current.feature]
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Evaluate all splits
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1723
|
|
*
|
|
* # Evaluate all splits
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD: # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature
|
|
*/
|
|
__pyx_t_9 = (((__pyx_v_Xf[(__pyx_v_end - 1)]) <= ((__pyx_v_Xf[__pyx_v_start]) + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1724
|
|
* # Evaluate all splits
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants] # <<<<<<<<<<<<<<
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_total_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1725
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature # <<<<<<<<<<<<<<
|
|
*
|
|
* n_found_constants += 1
|
|
*/
|
|
__pyx_t_2 = __pyx_v_current.feature;
|
|
(__pyx_v_features[__pyx_v_n_total_constants]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1727
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
* n_found_constants += 1 # <<<<<<<<<<<<<<
|
|
* n_total_constants += 1
|
|
*
|
|
*/
|
|
__pyx_v_n_found_constants = (__pyx_v_n_found_constants + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1728
|
|
*
|
|
* n_found_constants += 1
|
|
* n_total_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_total_constants = (__pyx_v_n_total_constants + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1731
|
|
*
|
|
* else:
|
|
* f_i -= 1 # <<<<<<<<<<<<<<
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i]
|
|
*
|
|
*/
|
|
__pyx_v_f_i = (__pyx_v_f_i - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1732
|
|
* else:
|
|
* f_i -= 1
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i] # <<<<<<<<<<<<<<
|
|
*
|
|
* self.criterion.reset()
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_features[__pyx_v_f_j]);
|
|
__pyx_t_7 = (__pyx_v_features[__pyx_v_f_i]);
|
|
(__pyx_v_features[__pyx_v_f_i]) = __pyx_t_2;
|
|
(__pyx_v_features[__pyx_v_f_j]) = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1734
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i]
|
|
*
|
|
* self.criterion.reset() # <<<<<<<<<<<<<<
|
|
* p = start
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->reset(__pyx_v_self->__pyx_base.__pyx_base.criterion);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1735
|
|
*
|
|
* self.criterion.reset()
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < end:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1737
|
|
* p = start
|
|
*
|
|
* while p < end: # <<<<<<<<<<<<<<
|
|
* while (p + 1 < end and
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD):
|
|
*/
|
|
while (1) {
|
|
__pyx_t_9 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (!__pyx_t_9) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1738
|
|
*
|
|
* while p < end:
|
|
* while (p + 1 < end and # <<<<<<<<<<<<<<
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p += 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_9 = (((__pyx_v_p + 1) < __pyx_v_end) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1739
|
|
* while p < end:
|
|
* while (p + 1 < end and
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD): # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_t_8 = (((__pyx_v_Xf[(__pyx_v_p + 1)]) <= ((__pyx_v_Xf[__pyx_v_p]) + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
__pyx_t_11 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_11 = __pyx_t_9;
|
|
}
|
|
if (!__pyx_t_11) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1740
|
|
* while (p + 1 < end and
|
|
* Xf[p + 1] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # (p + 1 >= end) or (X[samples[p + 1], current.feature] >
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1744
|
|
* # (p + 1 >= end) or (X[samples[p + 1], current.feature] >
|
|
* # X[samples[p], current.feature])
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
* # (p >= end) or (X[samples[p], current.feature] >
|
|
* # X[samples[p - 1], current.feature])
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1748
|
|
* # X[samples[p - 1], current.feature])
|
|
*
|
|
* if p < end: # <<<<<<<<<<<<<<
|
|
* current.pos = p
|
|
*
|
|
*/
|
|
__pyx_t_11 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (__pyx_t_11) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1749
|
|
*
|
|
* if p < end:
|
|
* current.pos = p # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
*/
|
|
__pyx_v_current.pos = __pyx_v_p;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1752
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_11 = (((__pyx_v_current.pos - __pyx_v_start) < __pyx_v_min_samples_leaf) != 0);
|
|
if (!__pyx_t_11) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1753
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_9 = (((__pyx_v_end - __pyx_v_current.pos) < __pyx_v_min_samples_leaf) != 0);
|
|
__pyx_t_8 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_11;
|
|
}
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1754
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* self.criterion.update(current.pos)
|
|
*/
|
|
goto __pyx_L12_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1756
|
|
* continue
|
|
*
|
|
* self.criterion.update(current.pos) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->update(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_current.pos);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1759
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or # <<<<<<<<<<<<<<
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_left < __pyx_v_min_weight_leaf) != 0);
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1760
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_11 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_right < __pyx_v_min_weight_leaf) != 0);
|
|
__pyx_t_9 = __pyx_t_11;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_8;
|
|
}
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1761
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*/
|
|
goto __pyx_L12_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1763
|
|
* continue
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity) # <<<<<<<<<<<<<<
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
*/
|
|
__pyx_v_current.improvement = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->impurity_improvement(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_impurity);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1765
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*
|
|
* if current.improvement > best.improvement: # <<<<<<<<<<<<<<
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_current.improvement > __pyx_v_best.improvement) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1766
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
* self.criterion.children_impurity(¤t.impurity_left, # <<<<<<<<<<<<<<
|
|
* ¤t.impurity_right)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->children_impurity(__pyx_v_self->__pyx_base.__pyx_base.criterion, (&__pyx_v_current.impurity_left), (&__pyx_v_current.impurity_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1769
|
|
* ¤t.impurity_right)
|
|
*
|
|
* current.threshold = (Xf[p - 1] + Xf[p]) / 2.0 # <<<<<<<<<<<<<<
|
|
* if current.threshold == Xf[p]:
|
|
* current.threshold = Xf[p - 1]
|
|
*/
|
|
__pyx_v_current.threshold = (((__pyx_v_Xf[(__pyx_v_p - 1)]) + (__pyx_v_Xf[__pyx_v_p])) / 2.0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1770
|
|
*
|
|
* current.threshold = (Xf[p - 1] + Xf[p]) / 2.0
|
|
* if current.threshold == Xf[p]: # <<<<<<<<<<<<<<
|
|
* current.threshold = Xf[p - 1]
|
|
*
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_current.threshold == (__pyx_v_Xf[__pyx_v_p])) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1771
|
|
* current.threshold = (Xf[p - 1] + Xf[p]) / 2.0
|
|
* if current.threshold == Xf[p]:
|
|
* current.threshold = Xf[p - 1] # <<<<<<<<<<<<<<
|
|
*
|
|
* best = current # copy
|
|
*/
|
|
__pyx_v_current.threshold = (__pyx_v_Xf[(__pyx_v_p - 1)]);
|
|
goto __pyx_L20;
|
|
}
|
|
__pyx_L20:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1773
|
|
* current.threshold = Xf[p - 1]
|
|
*
|
|
* best = current # copy # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
*/
|
|
__pyx_v_best = __pyx_v_current;
|
|
goto __pyx_L19;
|
|
}
|
|
__pyx_L19:;
|
|
goto __pyx_L16;
|
|
}
|
|
__pyx_L16:;
|
|
__pyx_L12_continue:;
|
|
}
|
|
}
|
|
__pyx_L11:;
|
|
}
|
|
__pyx_L7:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1776
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end: # <<<<<<<<<<<<<<
|
|
* partition_end = end
|
|
* p = start
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_best.pos < __pyx_v_end) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1777
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end:
|
|
* partition_end = end # <<<<<<<<<<<<<<
|
|
* p = start
|
|
*
|
|
*/
|
|
__pyx_v_partition_end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1778
|
|
* if best.pos < end:
|
|
* partition_end = end
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < partition_end:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1780
|
|
* p = start
|
|
*
|
|
* while p < partition_end: # <<<<<<<<<<<<<<
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_9 = ((__pyx_v_p < __pyx_v_partition_end) != 0);
|
|
if (!__pyx_t_9) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1782
|
|
* while p < partition_end:
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold: # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_t_9 = (((__pyx_v_X[((__pyx_v_X_sample_stride * (__pyx_v_samples[__pyx_v_p])) + (__pyx_v_X_fx_stride * __pyx_v_best.feature))]) <= __pyx_v_best.threshold) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1783
|
|
* if X[X_sample_stride * samples[p] +
|
|
* X_fx_stride * best.feature] <= best.threshold:
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
goto __pyx_L24;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1786
|
|
*
|
|
* else:
|
|
* partition_end -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* tmp = samples[partition_end]
|
|
*/
|
|
__pyx_v_partition_end = (__pyx_v_partition_end - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1788
|
|
* partition_end -= 1
|
|
*
|
|
* tmp = samples[partition_end] # <<<<<<<<<<<<<<
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_samples[__pyx_v_partition_end]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1789
|
|
*
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p] # <<<<<<<<<<<<<<
|
|
* samples[p] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_partition_end]) = (__pyx_v_samples[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1790
|
|
* tmp = samples[partition_end]
|
|
* samples[partition_end] = samples[p]
|
|
* samples[p] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reset sample mask
|
|
*/
|
|
(__pyx_v_samples[__pyx_v_p]) = __pyx_v_tmp;
|
|
}
|
|
__pyx_L24:;
|
|
}
|
|
goto __pyx_L21;
|
|
}
|
|
__pyx_L21:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1793
|
|
*
|
|
* # Reset sample mask
|
|
* for p in range(start, end): # <<<<<<<<<<<<<<
|
|
* sample_mask[samples[p]] = 0
|
|
*
|
|
*/
|
|
__pyx_t_7 = __pyx_v_end;
|
|
for (__pyx_t_2 = __pyx_v_start; __pyx_t_2 < __pyx_t_7; __pyx_t_2+=1) {
|
|
__pyx_v_p = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1794
|
|
* # Reset sample mask
|
|
* for p in range(start, end):
|
|
* sample_mask[samples[p]] = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Respect invariant for constant features: the original order of
|
|
*/
|
|
(__pyx_v_sample_mask[(__pyx_v_samples[__pyx_v_p])]) = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1799
|
|
* # element in features[:n_known_constants] must be preserved for sibling
|
|
* # and child nodes
|
|
* memcpy(features, constant_features, sizeof(SIZE_t) * n_known_constants) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Copy newly found constant features
|
|
*/
|
|
memcpy(__pyx_v_features, __pyx_v_constant_features, ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_known_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1802
|
|
*
|
|
* # Copy newly found constant features
|
|
* memcpy(constant_features + n_known_constants, # <<<<<<<<<<<<<<
|
|
* features + n_known_constants,
|
|
* sizeof(SIZE_t) * n_found_constants)
|
|
*/
|
|
memcpy((__pyx_v_constant_features + __pyx_v_n_known_constants), (__pyx_v_features + __pyx_v_n_known_constants), ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_found_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1807
|
|
*
|
|
* # Return values
|
|
* split[0] = best # <<<<<<<<<<<<<<
|
|
* n_constant_features[0] = n_total_constants
|
|
*
|
|
*/
|
|
(__pyx_v_split[0]) = __pyx_v_best;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1808
|
|
* # Return values
|
|
* split[0] = best
|
|
* n_constant_features[0] = n_total_constants # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_n_constant_features[0]) = __pyx_v_n_total_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1618
|
|
* memset(sample_mask, 0, self.n_total_samples)
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best split on node samples[start:end]."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1822
|
|
* cdef SIZE_t* sorted_samples
|
|
*
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_18BaseSparseSplitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_18BaseSparseSplitter_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
CYTHON_UNUSED double __pyx_v_min_weight_leaf;
|
|
CYTHON_UNUSED PyObject *__pyx_v_random_state = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_criterion,&__pyx_n_s_max_features,&__pyx_n_s_min_samples_leaf,&__pyx_n_s_min_weight_leaf,&__pyx_n_s_random_state,0};
|
|
PyObject* values[5] = {0,0,0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_criterion)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_features)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (likely((values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_weight_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 4:
|
|
if (likely((values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_random_state)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 5) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
}
|
|
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[0]);
|
|
__pyx_v_max_features = __Pyx_PyInt_As_Py_intptr_t(values[1]); if (unlikely((__pyx_v_max_features == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_samples_leaf = __Pyx_PyInt_As_Py_intptr_t(values[2]); if (unlikely((__pyx_v_min_samples_leaf == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1823; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_weight_leaf = __pyx_PyFloat_AsDouble(values[3]); if (unlikely((__pyx_v_min_weight_leaf == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1823; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_random_state = values[4];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BaseSparseSplitter.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_criterion), __pyx_ptype_7sklearn_4tree_5_tree_Criterion, 1, "criterion", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_18BaseSparseSplitter___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_criterion, __pyx_v_max_features, __pyx_v_min_samples_leaf, __pyx_v_min_weight_leaf, __pyx_v_random_state);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_18BaseSparseSplitter___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features, CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, CYTHON_UNUSED double __pyx_v_min_weight_leaf, CYTHON_UNUSED PyObject *__pyx_v_random_state) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1827
|
|
* # Parent __cinit__ is automatically called
|
|
*
|
|
* self.X_data = NULL # <<<<<<<<<<<<<<
|
|
* self.X_indices = NULL
|
|
* self.X_indptr = NULL
|
|
*/
|
|
__pyx_v_self->X_data = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1828
|
|
*
|
|
* self.X_data = NULL
|
|
* self.X_indices = NULL # <<<<<<<<<<<<<<
|
|
* self.X_indptr = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->X_indices = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1829
|
|
* self.X_data = NULL
|
|
* self.X_indices = NULL
|
|
* self.X_indptr = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* self.n_total_samples = 0
|
|
*/
|
|
__pyx_v_self->X_indptr = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1831
|
|
* self.X_indptr = NULL
|
|
*
|
|
* self.n_total_samples = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* self.index_to_samples = NULL
|
|
*/
|
|
__pyx_v_self->n_total_samples = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1833
|
|
* self.n_total_samples = 0
|
|
*
|
|
* self.index_to_samples = NULL # <<<<<<<<<<<<<<
|
|
* self.sorted_samples = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->index_to_samples = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1834
|
|
*
|
|
* self.index_to_samples = NULL
|
|
* self.sorted_samples = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* def __dealloc__(self):
|
|
*/
|
|
__pyx_v_self->sorted_samples = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1822
|
|
* cdef SIZE_t* sorted_samples
|
|
*
|
|
* def __cinit__(self, Criterion criterion, SIZE_t max_features, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* object random_state):
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1836
|
|
* self.sorted_samples = NULL
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Deallocate memory"""
|
|
* free(self.index_to_samples)
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_18BaseSparseSplitter_3__dealloc__(PyObject *__pyx_v_self); /*proto*/
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_18BaseSparseSplitter_3__dealloc__(PyObject *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0);
|
|
__pyx_pf_7sklearn_4tree_5_tree_18BaseSparseSplitter_2__dealloc__(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_18BaseSparseSplitter_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1838
|
|
* def __dealloc__(self):
|
|
* """Deallocate memory"""
|
|
* free(self.index_to_samples) # <<<<<<<<<<<<<<
|
|
* free(self.sorted_samples)
|
|
*
|
|
*/
|
|
free(__pyx_v_self->index_to_samples);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1839
|
|
* """Deallocate memory"""
|
|
* free(self.index_to_samples)
|
|
* free(self.sorted_samples) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void init(self,
|
|
*/
|
|
free(__pyx_v_self->sorted_samples);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1836
|
|
* self.sorted_samples = NULL
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Deallocate memory"""
|
|
* free(self.index_to_samples)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1841
|
|
* free(self.sorted_samples)
|
|
*
|
|
* cdef void init(self, # <<<<<<<<<<<<<<
|
|
* object X,
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_init(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_samples;
|
|
PyArrayObject *__pyx_v_data = 0;
|
|
PyArrayObject *__pyx_v_indices = 0;
|
|
PyArrayObject *__pyx_v_indptr = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_data;
|
|
__Pyx_Buffer __pyx_pybuffer_data;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indices;
|
|
__Pyx_Buffer __pyx_pybuffer_indices;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_indptr;
|
|
__Pyx_Buffer __pyx_pybuffer_indptr;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_y;
|
|
__Pyx_Buffer __pyx_pybuffer_y;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_4;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
PyArrayObject *__pyx_t_6 = NULL;
|
|
PyArrayObject *__pyx_t_7 = NULL;
|
|
PyArrayObject *__pyx_t_8 = NULL;
|
|
PyObject *__pyx_t_9 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_10;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("init", 0);
|
|
__pyx_pybuffer_data.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_data.refcount = 0;
|
|
__pyx_pybuffernd_data.data = NULL;
|
|
__pyx_pybuffernd_data.rcbuffer = &__pyx_pybuffer_data;
|
|
__pyx_pybuffer_indices.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indices.refcount = 0;
|
|
__pyx_pybuffernd_indices.data = NULL;
|
|
__pyx_pybuffernd_indices.rcbuffer = &__pyx_pybuffer_indices;
|
|
__pyx_pybuffer_indptr.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_indptr.refcount = 0;
|
|
__pyx_pybuffernd_indptr.data = NULL;
|
|
__pyx_pybuffernd_indptr.rcbuffer = &__pyx_pybuffer_indptr;
|
|
__pyx_pybuffer_y.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_y.refcount = 0;
|
|
__pyx_pybuffernd_y.data = NULL;
|
|
__pyx_pybuffernd_y.rcbuffer = &__pyx_pybuffer_y;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_y.rcbuffer->pybuffer, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DOUBLE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_y.diminfo[0].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_y.diminfo[0].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[0]; __pyx_pybuffernd_y.diminfo[1].strides = __pyx_pybuffernd_y.rcbuffer->pybuffer.strides[1]; __pyx_pybuffernd_y.diminfo[1].shape = __pyx_pybuffernd_y.rcbuffer->pybuffer.shape[1];
|
|
|
|
/* "sklearn/tree/_tree.pyx":1848
|
|
*
|
|
* # Call parent init
|
|
* Splitter.init(self, X, y, sample_weight) # <<<<<<<<<<<<<<
|
|
*
|
|
* if not isinstance(X, csc_matrix):
|
|
*/
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Splitter->init(((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_v_self), __pyx_v_X, ((PyArrayObject *)__pyx_v_y), __pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1850
|
|
* Splitter.init(self, X, y, sample_weight)
|
|
*
|
|
* if not isinstance(X, csc_matrix): # <<<<<<<<<<<<<<
|
|
* raise ValueError("X should be in csc format")
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_csc_matrix); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1850; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_IsInstance(__pyx_v_X, __pyx_t_1); if (unlikely(__pyx_t_2 == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1850; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_3 = ((!(__pyx_t_2 != 0)) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1851
|
|
*
|
|
* if not isinstance(X, csc_matrix):
|
|
* raise ValueError("X should be in csc format") # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple_, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1851; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_Raise(__pyx_t_1, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 1851; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1853
|
|
* raise ValueError("X should be in csc format")
|
|
*
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_samples = self.n_samples
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1854
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t n_samples = self.n_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize X
|
|
*/
|
|
__pyx_t_5 = __pyx_v_self->__pyx_base.n_samples;
|
|
__pyx_v_n_samples = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1857
|
|
*
|
|
* # Initialize X
|
|
* cdef np.ndarray[dtype=DTYPE_t, ndim=1] data = X.data # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indices = X.indices
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indptr = X.indptr
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_data); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1857; __pyx_clineno = __LINE__; goto __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_filename = __pyx_f[0]; __pyx_lineno = 1857; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_6 = ((PyArrayObject *)__pyx_t_1);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_data.rcbuffer->pybuffer, (PyObject*)__pyx_t_6, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_data = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_data.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 1857; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_data.diminfo[0].strides = __pyx_pybuffernd_data.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_data.diminfo[0].shape = __pyx_pybuffernd_data.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_6 = 0;
|
|
__pyx_v_data = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1858
|
|
* # Initialize X
|
|
* cdef np.ndarray[dtype=DTYPE_t, ndim=1] data = X.data
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indices = X.indices # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indptr = X.indptr
|
|
* cdef SIZE_t n_total_samples = X.shape[0]
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_indices); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1858; __pyx_clineno = __LINE__; goto __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_filename = __pyx_f[0]; __pyx_lineno = 1858; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_7 = ((PyArrayObject *)__pyx_t_1);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indices.rcbuffer->pybuffer, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_INT32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_indices = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_indices.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 1858; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_indices.diminfo[0].strides = __pyx_pybuffernd_indices.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indices.diminfo[0].shape = __pyx_pybuffernd_indices.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_7 = 0;
|
|
__pyx_v_indices = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1859
|
|
* cdef np.ndarray[dtype=DTYPE_t, ndim=1] data = X.data
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indices = X.indices
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indptr = X.indptr # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_total_samples = X.shape[0]
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_indptr); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1859; __pyx_clineno = __LINE__; goto __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_filename = __pyx_f[0]; __pyx_lineno = 1859; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_8 = ((PyArrayObject *)__pyx_t_1);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer, (PyObject*)__pyx_t_8, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_INT32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_indptr = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_indptr.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 1859; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_indptr.diminfo[0].strides = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_indptr.diminfo[0].shape = __pyx_pybuffernd_indptr.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_8 = 0;
|
|
__pyx_v_indptr = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1860
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indices = X.indices
|
|
* cdef np.ndarray[dtype=INT32_t, ndim=1] indptr = X.indptr
|
|
* cdef SIZE_t n_total_samples = X.shape[0] # <<<<<<<<<<<<<<
|
|
*
|
|
* self.X_data = <DTYPE_t*> data.data
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1860; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_9 = __Pyx_GetItemInt(__pyx_t_1, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_9 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1860; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_9); if (unlikely((__pyx_t_5 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1860; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
|
|
__pyx_v_n_total_samples = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1862
|
|
* cdef SIZE_t n_total_samples = X.shape[0]
|
|
*
|
|
* self.X_data = <DTYPE_t*> data.data # <<<<<<<<<<<<<<
|
|
* self.X_indices = <INT32_t*> indices.data
|
|
* self.X_indptr = <INT32_t*> indptr.data
|
|
*/
|
|
__pyx_v_self->X_data = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_data->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1863
|
|
*
|
|
* self.X_data = <DTYPE_t*> data.data
|
|
* self.X_indices = <INT32_t*> indices.data # <<<<<<<<<<<<<<
|
|
* self.X_indptr = <INT32_t*> indptr.data
|
|
* self.n_total_samples = n_total_samples
|
|
*/
|
|
__pyx_v_self->X_indices = ((__pyx_t_7sklearn_4tree_5_tree_INT32_t *)__pyx_v_indices->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1864
|
|
* self.X_data = <DTYPE_t*> data.data
|
|
* self.X_indices = <INT32_t*> indices.data
|
|
* self.X_indptr = <INT32_t*> indptr.data # <<<<<<<<<<<<<<
|
|
* self.n_total_samples = n_total_samples
|
|
*
|
|
*/
|
|
__pyx_v_self->X_indptr = ((__pyx_t_7sklearn_4tree_5_tree_INT32_t *)__pyx_v_indptr->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1865
|
|
* self.X_indices = <INT32_t*> indices.data
|
|
* self.X_indptr = <INT32_t*> indptr.data
|
|
* self.n_total_samples = n_total_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize auxiliary array used to perform split
|
|
*/
|
|
__pyx_v_self->n_total_samples = __pyx_v_n_total_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1868
|
|
*
|
|
* # Initialize auxiliary array used to perform split
|
|
* safe_realloc(&self.index_to_samples, n_total_samples * sizeof(SIZE_t)) # <<<<<<<<<<<<<<
|
|
* safe_realloc(&self.sorted_samples, n_samples * sizeof(SIZE_t))
|
|
*
|
|
*/
|
|
__pyx_t_4 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->index_to_samples), (__pyx_v_n_total_samples * (sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)))); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1868; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1869
|
|
* # Initialize auxiliary array used to perform split
|
|
* safe_realloc(&self.index_to_samples, n_total_samples * sizeof(SIZE_t))
|
|
* safe_realloc(&self.sorted_samples, n_samples * sizeof(SIZE_t)) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
*/
|
|
__pyx_t_4 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->sorted_samples), (__pyx_v_n_samples * (sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)))); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1869; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1871
|
|
* safe_realloc(&self.sorted_samples, n_samples * sizeof(SIZE_t))
|
|
*
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t p
|
|
* for p in range(n_total_samples):
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->index_to_samples;
|
|
__pyx_v_index_to_samples = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1873
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t p
|
|
* for p in range(n_total_samples): # <<<<<<<<<<<<<<
|
|
* index_to_samples[p] = -1
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_n_total_samples;
|
|
for (__pyx_t_10 = 0; __pyx_t_10 < __pyx_t_5; __pyx_t_10+=1) {
|
|
__pyx_v_p = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1874
|
|
* cdef SIZE_t p
|
|
* for p in range(n_total_samples):
|
|
* index_to_samples[p] = -1 # <<<<<<<<<<<<<<
|
|
*
|
|
* for p in range(n_samples):
|
|
*/
|
|
(__pyx_v_index_to_samples[__pyx_v_p]) = -1;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1876
|
|
* index_to_samples[p] = -1
|
|
*
|
|
* for p in range(n_samples): # <<<<<<<<<<<<<<
|
|
* index_to_samples[samples[p]] = p
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_n_samples;
|
|
for (__pyx_t_10 = 0; __pyx_t_10 < __pyx_t_5; __pyx_t_10+=1) {
|
|
__pyx_v_p = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1877
|
|
*
|
|
* for p in range(n_samples):
|
|
* index_to_samples[samples[p]] = p # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline SIZE_t _partition(self, double threshold,
|
|
*/
|
|
(__pyx_v_index_to_samples[(__pyx_v_samples[__pyx_v_p])]) = __pyx_v_p;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1841
|
|
* free(self.sorted_samples)
|
|
*
|
|
* cdef void init(self, # <<<<<<<<<<<<<<
|
|
* object X,
|
|
* np.ndarray[DOUBLE_t, ndim=2, mode="c"] y,
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_9);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_WriteUnraisable("sklearn.tree._tree.BaseSparseSplitter.init", __pyx_clineno, __pyx_lineno, __pyx_filename, 0);
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_data.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indices.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_indptr.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_y.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_data);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_indices);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_indptr);
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1879
|
|
* index_to_samples[samples[p]] = p
|
|
*
|
|
* cdef inline SIZE_t _partition(self, double threshold, # <<<<<<<<<<<<<<
|
|
* SIZE_t end_negative, SIZE_t start_positive,
|
|
* SIZE_t zero_pos) nogil:
|
|
*/
|
|
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter__partition(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self, double __pyx_v_threshold, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end_negative, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start_positive, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_zero_pos) {
|
|
double __pyx_v_value;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_partition_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_r;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_2;
|
|
int __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1888
|
|
* cdef SIZE_t p
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.feature_values;
|
|
__pyx_v_Xf = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1889
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1890
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* if threshold < 0.:
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->index_to_samples;
|
|
__pyx_v_index_to_samples = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1892
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
*
|
|
* if threshold < 0.: # <<<<<<<<<<<<<<
|
|
* p = self.start
|
|
* partition_end = end_negative
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_threshold < 0.) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1893
|
|
*
|
|
* if threshold < 0.:
|
|
* p = self.start # <<<<<<<<<<<<<<
|
|
* partition_end = end_negative
|
|
* elif threshold > 0.:
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.start;
|
|
__pyx_v_p = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1894
|
|
* if threshold < 0.:
|
|
* p = self.start
|
|
* partition_end = end_negative # <<<<<<<<<<<<<<
|
|
* elif threshold > 0.:
|
|
* p = start_positive
|
|
*/
|
|
__pyx_v_partition_end = __pyx_v_end_negative;
|
|
goto __pyx_L3;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1895
|
|
* p = self.start
|
|
* partition_end = end_negative
|
|
* elif threshold > 0.: # <<<<<<<<<<<<<<
|
|
* p = start_positive
|
|
* partition_end = self.end
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_threshold > 0.) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1896
|
|
* partition_end = end_negative
|
|
* elif threshold > 0.:
|
|
* p = start_positive # <<<<<<<<<<<<<<
|
|
* partition_end = self.end
|
|
* else:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start_positive;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1897
|
|
* elif threshold > 0.:
|
|
* p = start_positive
|
|
* partition_end = self.end # <<<<<<<<<<<<<<
|
|
* else:
|
|
* # Data are already split
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.end;
|
|
__pyx_v_partition_end = __pyx_t_4;
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1900
|
|
* else:
|
|
* # Data are already split
|
|
* return zero_pos # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < partition_end:
|
|
*/
|
|
__pyx_r = __pyx_v_zero_pos;
|
|
goto __pyx_L0;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1902
|
|
* return zero_pos
|
|
*
|
|
* while p < partition_end: # <<<<<<<<<<<<<<
|
|
* value = Xf[p]
|
|
*
|
|
*/
|
|
while (1) {
|
|
__pyx_t_3 = ((__pyx_v_p < __pyx_v_partition_end) != 0);
|
|
if (!__pyx_t_3) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1903
|
|
*
|
|
* while p < partition_end:
|
|
* value = Xf[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if value <= threshold:
|
|
*/
|
|
__pyx_v_value = (__pyx_v_Xf[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1905
|
|
* value = Xf[p]
|
|
*
|
|
* if value <= threshold: # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_value <= __pyx_v_threshold) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1906
|
|
*
|
|
* if value <= threshold:
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1909
|
|
*
|
|
* else:
|
|
* partition_end -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* Xf[p] = Xf[partition_end]
|
|
*/
|
|
__pyx_v_partition_end = (__pyx_v_partition_end - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1911
|
|
* partition_end -= 1
|
|
*
|
|
* Xf[p] = Xf[partition_end] # <<<<<<<<<<<<<<
|
|
* Xf[partition_end] = value
|
|
* sparse_swap(index_to_samples, samples, p, partition_end)
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_p]) = (__pyx_v_Xf[__pyx_v_partition_end]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1912
|
|
*
|
|
* Xf[p] = Xf[partition_end]
|
|
* Xf[partition_end] = value # <<<<<<<<<<<<<<
|
|
* sparse_swap(index_to_samples, samples, p, partition_end)
|
|
*
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_partition_end]) = __pyx_v_value;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1913
|
|
* Xf[p] = Xf[partition_end]
|
|
* Xf[partition_end] = value
|
|
* sparse_swap(index_to_samples, samples, p, partition_end) # <<<<<<<<<<<<<<
|
|
*
|
|
* return partition_end
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_v_index_to_samples, __pyx_v_samples, __pyx_v_p, __pyx_v_partition_end);
|
|
}
|
|
__pyx_L6:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1915
|
|
* sparse_swap(index_to_samples, samples, p, partition_end)
|
|
*
|
|
* return partition_end # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline void extract_nnz(self, SIZE_t feature,
|
|
*/
|
|
__pyx_r = __pyx_v_partition_end;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1879
|
|
* index_to_samples[samples[p]] = p
|
|
*
|
|
* cdef inline SIZE_t _partition(self, double threshold, # <<<<<<<<<<<<<<
|
|
* SIZE_t end_negative, SIZE_t start_positive,
|
|
* SIZE_t zero_pos) nogil:
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1917
|
|
* return partition_end
|
|
*
|
|
* cdef inline void extract_nnz(self, SIZE_t feature, # <<<<<<<<<<<<<<
|
|
* SIZE_t* end_negative, SIZE_t* start_positive,
|
|
* bint* is_samples_sorted) nogil:
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_feature, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_end_negative, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_start_positive, int *__pyx_v_is_samples_sorted) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_indptr_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_indptr_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_indices;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_samples;
|
|
int __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1949
|
|
*
|
|
* """
|
|
* cdef SIZE_t indptr_start = self.X_indptr[feature], # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t indptr_end = self.X_indptr[feature + 1]
|
|
* cdef SIZE_t n_indices = <SIZE_t>(indptr_end - indptr_start)
|
|
*/
|
|
__pyx_v_indptr_start = (__pyx_v_self->X_indptr[__pyx_v_feature]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1950
|
|
* """
|
|
* cdef SIZE_t indptr_start = self.X_indptr[feature],
|
|
* cdef SIZE_t indptr_end = self.X_indptr[feature + 1] # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_indices = <SIZE_t>(indptr_end - indptr_start)
|
|
* cdef SIZE_t n_samples = self.end - self.start
|
|
*/
|
|
__pyx_v_indptr_end = (__pyx_v_self->X_indptr[(__pyx_v_feature + 1)]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1951
|
|
* cdef SIZE_t indptr_start = self.X_indptr[feature],
|
|
* cdef SIZE_t indptr_end = self.X_indptr[feature + 1]
|
|
* cdef SIZE_t n_indices = <SIZE_t>(indptr_end - indptr_start) # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_samples = self.end - self.start
|
|
*
|
|
*/
|
|
__pyx_v_n_indices = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_indptr_end - __pyx_v_indptr_start));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1952
|
|
* cdef SIZE_t indptr_end = self.X_indptr[feature + 1]
|
|
* cdef SIZE_t n_indices = <SIZE_t>(indptr_end - indptr_start)
|
|
* cdef SIZE_t n_samples = self.end - self.start # <<<<<<<<<<<<<<
|
|
*
|
|
* # Use binary search if n_samples * log(n_indices) <
|
|
*/
|
|
__pyx_v_n_samples = (__pyx_v_self->__pyx_base.end - __pyx_v_self->__pyx_base.start);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1960
|
|
* # approach.
|
|
* if ((1 - is_samples_sorted[0]) * n_samples * log(n_samples) +
|
|
* n_samples * log(n_indices) < EXTRACT_NNZ_SWITCH * n_indices): # <<<<<<<<<<<<<<
|
|
* extract_nnz_binary_search(self.X_indices, self.X_data,
|
|
* indptr_start, indptr_end,
|
|
*/
|
|
__pyx_t_1 = ((((((1 - (__pyx_v_is_samples_sorted[0])) * __pyx_v_n_samples) * __pyx_f_7sklearn_4tree_5_tree_log(__pyx_v_n_samples)) + (__pyx_v_n_samples * __pyx_f_7sklearn_4tree_5_tree_log(__pyx_v_n_indices))) < (__pyx_v_7sklearn_4tree_5_tree_EXTRACT_NNZ_SWITCH * __pyx_v_n_indices)) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1961
|
|
* if ((1 - is_samples_sorted[0]) * n_samples * log(n_samples) +
|
|
* n_samples * log(n_indices) < EXTRACT_NNZ_SWITCH * n_indices):
|
|
* extract_nnz_binary_search(self.X_indices, self.X_data, # <<<<<<<<<<<<<<
|
|
* indptr_start, indptr_end,
|
|
* self.samples, self.start, self.end,
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_extract_nnz_binary_search(__pyx_v_self->X_indices, __pyx_v_self->X_data, __pyx_v_indptr_start, __pyx_v_indptr_end, __pyx_v_self->__pyx_base.samples, __pyx_v_self->__pyx_base.start, __pyx_v_self->__pyx_base.end, __pyx_v_self->index_to_samples, __pyx_v_self->__pyx_base.feature_values, __pyx_v_end_negative, __pyx_v_start_positive, __pyx_v_self->sorted_samples, __pyx_v_is_samples_sorted);
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1972
|
|
* # index_to_samples is a mapping from X_indices to samples
|
|
* else:
|
|
* extract_nnz_index_to_samples(self.X_indices, self.X_data, # <<<<<<<<<<<<<<
|
|
* indptr_start, indptr_end,
|
|
* self.samples, self.start, self.end,
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_extract_nnz_index_to_samples(__pyx_v_self->X_indices, __pyx_v_self->X_data, __pyx_v_indptr_start, __pyx_v_indptr_end, __pyx_v_self->__pyx_base.samples, __pyx_v_self->__pyx_base.start, __pyx_v_self->__pyx_base.end, __pyx_v_self->index_to_samples, __pyx_v_self->__pyx_base.feature_values, __pyx_v_end_negative, __pyx_v_start_positive);
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1917
|
|
* return partition_end
|
|
*
|
|
* cdef inline void extract_nnz(self, SIZE_t feature, # <<<<<<<<<<<<<<
|
|
* SIZE_t* end_negative, SIZE_t* start_positive,
|
|
* bint* is_samples_sorted) nogil:
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1980
|
|
*
|
|
*
|
|
* cdef int compare_SIZE_t(const void* a, const void* b) nogil: # <<<<<<<<<<<<<<
|
|
* """Comparison function for sort"""
|
|
* return <int>((<SIZE_t*>a)[0] - (<SIZE_t*>b)[0])
|
|
*/
|
|
|
|
static int __pyx_f_7sklearn_4tree_5_tree_compare_SIZE_t(void const *__pyx_v_a, void const *__pyx_v_b) {
|
|
int __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1982
|
|
* cdef int compare_SIZE_t(const void* a, const void* b) nogil:
|
|
* """Comparison function for sort"""
|
|
* return <int>((<SIZE_t*>a)[0] - (<SIZE_t*>b)[0]) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = ((int)((((__pyx_t_7sklearn_4tree_5_tree_SIZE_t *)__pyx_v_a)[0]) - (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t *)__pyx_v_b)[0])));
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1980
|
|
*
|
|
*
|
|
* cdef int compare_SIZE_t(const void* a, const void* b) nogil: # <<<<<<<<<<<<<<
|
|
* """Comparison function for sort"""
|
|
* return <int>((<SIZE_t*>a)[0] - (<SIZE_t*>b)[0])
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":1985
|
|
*
|
|
*
|
|
* cdef inline void binary_search(INT32_t* sorted_array, # <<<<<<<<<<<<<<
|
|
* INT32_t start, INT32_t end,
|
|
* SIZE_t value, SIZE_t* index,
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_binary_search(__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_sorted_array, __pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_end, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_value, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index, __pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_new_start) {
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_pivot;
|
|
int __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1994
|
|
* """
|
|
* cdef INT32_t pivot
|
|
* index[0] = -1 # <<<<<<<<<<<<<<
|
|
* while start < end:
|
|
* pivot = start + (end - start) / 2
|
|
*/
|
|
(__pyx_v_index[0]) = -1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1995
|
|
* cdef INT32_t pivot
|
|
* index[0] = -1
|
|
* while start < end: # <<<<<<<<<<<<<<
|
|
* pivot = start + (end - start) / 2
|
|
*
|
|
*/
|
|
while (1) {
|
|
__pyx_t_1 = ((__pyx_v_start < __pyx_v_end) != 0);
|
|
if (!__pyx_t_1) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1996
|
|
* index[0] = -1
|
|
* while start < end:
|
|
* pivot = start + (end - start) / 2 # <<<<<<<<<<<<<<
|
|
*
|
|
* if sorted_array[pivot] == value:
|
|
*/
|
|
__pyx_v_pivot = (__pyx_v_start + ((__pyx_v_end - __pyx_v_start) / 2));
|
|
|
|
/* "sklearn/tree/_tree.pyx":1998
|
|
* pivot = start + (end - start) / 2
|
|
*
|
|
* if sorted_array[pivot] == value: # <<<<<<<<<<<<<<
|
|
* index[0] = pivot
|
|
* start = pivot + 1
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_sorted_array[__pyx_v_pivot]) == __pyx_v_value) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":1999
|
|
*
|
|
* if sorted_array[pivot] == value:
|
|
* index[0] = pivot # <<<<<<<<<<<<<<
|
|
* start = pivot + 1
|
|
* break
|
|
*/
|
|
(__pyx_v_index[0]) = __pyx_v_pivot;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2000
|
|
* if sorted_array[pivot] == value:
|
|
* index[0] = pivot
|
|
* start = pivot + 1 # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_v_start = (__pyx_v_pivot + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2001
|
|
* index[0] = pivot
|
|
* start = pivot + 1
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if sorted_array[pivot] < value:
|
|
*/
|
|
goto __pyx_L4_break;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2003
|
|
* break
|
|
*
|
|
* if sorted_array[pivot] < value: # <<<<<<<<<<<<<<
|
|
* start = pivot + 1
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_sorted_array[__pyx_v_pivot]) < __pyx_v_value) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2004
|
|
*
|
|
* if sorted_array[pivot] < value:
|
|
* start = pivot + 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* end = pivot
|
|
*/
|
|
__pyx_v_start = (__pyx_v_pivot + 1);
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2006
|
|
* start = pivot + 1
|
|
* else:
|
|
* end = pivot # <<<<<<<<<<<<<<
|
|
* new_start[0] = start
|
|
*
|
|
*/
|
|
__pyx_v_end = __pyx_v_pivot;
|
|
}
|
|
__pyx_L6:;
|
|
}
|
|
__pyx_L4_break:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2007
|
|
* else:
|
|
* end = pivot
|
|
* new_start[0] = start # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_new_start[0]) = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1985
|
|
*
|
|
*
|
|
* cdef inline void binary_search(INT32_t* sorted_array, # <<<<<<<<<<<<<<
|
|
* INT32_t start, INT32_t end,
|
|
* SIZE_t value, SIZE_t* index,
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2010
|
|
*
|
|
*
|
|
* cdef inline void extract_nnz_index_to_samples(INT32_t* X_indices, # <<<<<<<<<<<<<<
|
|
* DTYPE_t* X_data,
|
|
* INT32_t indptr_start,
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_extract_nnz_index_to_samples(__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indices, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_data, __pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_indptr_start, __pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_indptr_end, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_end_negative, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_start_positive) {
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_index;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end_negative_;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start_positive_;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_t_2;
|
|
int __pyx_t_3;
|
|
int __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2027
|
|
* cdef INT32_t k
|
|
* cdef SIZE_t index
|
|
* cdef SIZE_t end_negative_ = start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start_positive_ = end
|
|
*
|
|
*/
|
|
__pyx_v_end_negative_ = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2028
|
|
* cdef SIZE_t index
|
|
* cdef SIZE_t end_negative_ = start
|
|
* cdef SIZE_t start_positive_ = end # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(indptr_start, indptr_end):
|
|
*/
|
|
__pyx_v_start_positive_ = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2030
|
|
* cdef SIZE_t start_positive_ = end
|
|
*
|
|
* for k in range(indptr_start, indptr_end): # <<<<<<<<<<<<<<
|
|
* if start <= index_to_samples[X_indices[k]] < end:
|
|
* if X_data[k] > 0:
|
|
*/
|
|
__pyx_t_1 = __pyx_v_indptr_end;
|
|
for (__pyx_t_2 = __pyx_v_indptr_start; __pyx_t_2 < __pyx_t_1; __pyx_t_2+=1) {
|
|
__pyx_v_k = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2031
|
|
*
|
|
* for k in range(indptr_start, indptr_end):
|
|
* if start <= index_to_samples[X_indices[k]] < end: # <<<<<<<<<<<<<<
|
|
* if X_data[k] > 0:
|
|
* start_positive_ -= 1
|
|
*/
|
|
__pyx_t_3 = (__pyx_v_start <= (__pyx_v_index_to_samples[(__pyx_v_X_indices[__pyx_v_k])]));
|
|
if (__pyx_t_3) {
|
|
__pyx_t_3 = ((__pyx_v_index_to_samples[(__pyx_v_X_indices[__pyx_v_k])]) < __pyx_v_end);
|
|
}
|
|
__pyx_t_4 = (__pyx_t_3 != 0);
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2032
|
|
* for k in range(indptr_start, indptr_end):
|
|
* if start <= index_to_samples[X_indices[k]] < end:
|
|
* if X_data[k] > 0: # <<<<<<<<<<<<<<
|
|
* start_positive_ -= 1
|
|
* Xf[start_positive_] = X_data[k]
|
|
*/
|
|
__pyx_t_4 = (((__pyx_v_X_data[__pyx_v_k]) > 0.0) != 0);
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2033
|
|
* if start <= index_to_samples[X_indices[k]] < end:
|
|
* if X_data[k] > 0:
|
|
* start_positive_ -= 1 # <<<<<<<<<<<<<<
|
|
* Xf[start_positive_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
*/
|
|
__pyx_v_start_positive_ = (__pyx_v_start_positive_ - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2034
|
|
* if X_data[k] > 0:
|
|
* start_positive_ -= 1
|
|
* Xf[start_positive_] = X_data[k] # <<<<<<<<<<<<<<
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, start_positive_)
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_start_positive_]) = (__pyx_v_X_data[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2035
|
|
* start_positive_ -= 1
|
|
* Xf[start_positive_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]] # <<<<<<<<<<<<<<
|
|
* sparse_swap(index_to_samples, samples, index, start_positive_)
|
|
*
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index_to_samples[(__pyx_v_X_indices[__pyx_v_k])]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2036
|
|
* Xf[start_positive_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, start_positive_) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_v_index_to_samples, __pyx_v_samples, __pyx_v_index, __pyx_v_start_positive_);
|
|
goto __pyx_L6;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2039
|
|
*
|
|
*
|
|
* elif X_data[k] < 0: # <<<<<<<<<<<<<<
|
|
* Xf[end_negative_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
*/
|
|
__pyx_t_4 = (((__pyx_v_X_data[__pyx_v_k]) < 0.0) != 0);
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2040
|
|
*
|
|
* elif X_data[k] < 0:
|
|
* Xf[end_negative_] = X_data[k] # <<<<<<<<<<<<<<
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_end_negative_]) = (__pyx_v_X_data[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2041
|
|
* elif X_data[k] < 0:
|
|
* Xf[end_negative_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]] # <<<<<<<<<<<<<<
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
* end_negative_ += 1
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index_to_samples[(__pyx_v_X_indices[__pyx_v_k])]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2042
|
|
* Xf[end_negative_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_) # <<<<<<<<<<<<<<
|
|
* end_negative_ += 1
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_v_index_to_samples, __pyx_v_samples, __pyx_v_index, __pyx_v_end_negative_);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2043
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
* end_negative_ += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Returned values
|
|
*/
|
|
__pyx_v_end_negative_ = (__pyx_v_end_negative_ + 1);
|
|
goto __pyx_L6;
|
|
}
|
|
__pyx_L6:;
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2046
|
|
*
|
|
* # Returned values
|
|
* end_negative[0] = end_negative_ # <<<<<<<<<<<<<<
|
|
* start_positive[0] = start_positive_
|
|
*
|
|
*/
|
|
(__pyx_v_end_negative[0]) = __pyx_v_end_negative_;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2047
|
|
* # Returned values
|
|
* end_negative[0] = end_negative_
|
|
* start_positive[0] = start_positive_ # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_start_positive[0]) = __pyx_v_start_positive_;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2010
|
|
*
|
|
*
|
|
* cdef inline void extract_nnz_index_to_samples(INT32_t* X_indices, # <<<<<<<<<<<<<<
|
|
* DTYPE_t* X_data,
|
|
* INT32_t indptr_start,
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2050
|
|
*
|
|
*
|
|
* cdef inline void extract_nnz_binary_search(INT32_t* X_indices, # <<<<<<<<<<<<<<
|
|
* DTYPE_t* X_data,
|
|
* INT32_t indptr_start,
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_extract_nnz_binary_search(__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indices, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_data, __pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_indptr_start, __pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_indptr_end, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_end_negative, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_start_positive, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_sorted_samples, int *__pyx_v_is_samples_sorted) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_index;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end_negative_;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start_positive_;
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2073
|
|
* cdef SIZE_t n_samples
|
|
*
|
|
* if not is_samples_sorted[0]: # <<<<<<<<<<<<<<
|
|
* n_samples = end - start
|
|
* memcpy(sorted_samples + start, samples + start,
|
|
*/
|
|
__pyx_t_1 = ((!((__pyx_v_is_samples_sorted[0]) != 0)) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2074
|
|
*
|
|
* if not is_samples_sorted[0]:
|
|
* n_samples = end - start # <<<<<<<<<<<<<<
|
|
* memcpy(sorted_samples + start, samples + start,
|
|
* n_samples * sizeof(SIZE_t))
|
|
*/
|
|
__pyx_v_n_samples = (__pyx_v_end - __pyx_v_start);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2075
|
|
* if not is_samples_sorted[0]:
|
|
* n_samples = end - start
|
|
* memcpy(sorted_samples + start, samples + start, # <<<<<<<<<<<<<<
|
|
* n_samples * sizeof(SIZE_t))
|
|
* qsort(sorted_samples + start, n_samples, sizeof(SIZE_t),
|
|
*/
|
|
memcpy((__pyx_v_sorted_samples + __pyx_v_start), (__pyx_v_samples + __pyx_v_start), (__pyx_v_n_samples * (sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2077
|
|
* memcpy(sorted_samples + start, samples + start,
|
|
* n_samples * sizeof(SIZE_t))
|
|
* qsort(sorted_samples + start, n_samples, sizeof(SIZE_t), # <<<<<<<<<<<<<<
|
|
* compare_SIZE_t)
|
|
* is_samples_sorted[0] = 1
|
|
*/
|
|
qsort((__pyx_v_sorted_samples + __pyx_v_start), __pyx_v_n_samples, (sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)), __pyx_f_7sklearn_4tree_5_tree_compare_SIZE_t);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2079
|
|
* qsort(sorted_samples + start, n_samples, sizeof(SIZE_t),
|
|
* compare_SIZE_t)
|
|
* is_samples_sorted[0] = 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* while (indptr_start < indptr_end and
|
|
*/
|
|
(__pyx_v_is_samples_sorted[0]) = 1;
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2081
|
|
* is_samples_sorted[0] = 1
|
|
*
|
|
* while (indptr_start < indptr_end and # <<<<<<<<<<<<<<
|
|
* sorted_samples[start] > X_indices[indptr_start]):
|
|
* indptr_start += 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_1 = ((__pyx_v_indptr_start < __pyx_v_indptr_end) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2082
|
|
*
|
|
* while (indptr_start < indptr_end and
|
|
* sorted_samples[start] > X_indices[indptr_start]): # <<<<<<<<<<<<<<
|
|
* indptr_start += 1
|
|
*
|
|
*/
|
|
__pyx_t_2 = (((__pyx_v_sorted_samples[__pyx_v_start]) > (__pyx_v_X_indices[__pyx_v_indptr_start])) != 0);
|
|
__pyx_t_3 = __pyx_t_2;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_1;
|
|
}
|
|
if (!__pyx_t_3) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2083
|
|
* while (indptr_start < indptr_end and
|
|
* sorted_samples[start] > X_indices[indptr_start]):
|
|
* indptr_start += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* while (indptr_start < indptr_end and
|
|
*/
|
|
__pyx_v_indptr_start = (__pyx_v_indptr_start + 1);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2085
|
|
* indptr_start += 1
|
|
*
|
|
* while (indptr_start < indptr_end and # <<<<<<<<<<<<<<
|
|
* sorted_samples[end - 1] < X_indices[indptr_end - 1]):
|
|
* indptr_end -= 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_3 = ((__pyx_v_indptr_start < __pyx_v_indptr_end) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2086
|
|
*
|
|
* while (indptr_start < indptr_end and
|
|
* sorted_samples[end - 1] < X_indices[indptr_end - 1]): # <<<<<<<<<<<<<<
|
|
* indptr_end -= 1
|
|
*
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_sorted_samples[(__pyx_v_end - 1)]) < (__pyx_v_X_indices[(__pyx_v_indptr_end - 1)])) != 0);
|
|
__pyx_t_2 = __pyx_t_1;
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_3;
|
|
}
|
|
if (!__pyx_t_2) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2087
|
|
* while (indptr_start < indptr_end and
|
|
* sorted_samples[end - 1] < X_indices[indptr_end - 1]):
|
|
* indptr_end -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t p = start
|
|
*/
|
|
__pyx_v_indptr_end = (__pyx_v_indptr_end - 1);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2089
|
|
* indptr_end -= 1
|
|
*
|
|
* cdef SIZE_t p = start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t index
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2092
|
|
* cdef SIZE_t index
|
|
* cdef SIZE_t k
|
|
* cdef SIZE_t end_negative_ = start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start_positive_ = end
|
|
*
|
|
*/
|
|
__pyx_v_end_negative_ = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2093
|
|
* cdef SIZE_t k
|
|
* cdef SIZE_t end_negative_ = start
|
|
* cdef SIZE_t start_positive_ = end # <<<<<<<<<<<<<<
|
|
*
|
|
* while (p < end and indptr_start < indptr_end):
|
|
*/
|
|
__pyx_v_start_positive_ = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2095
|
|
* cdef SIZE_t start_positive_ = end
|
|
*
|
|
* while (p < end and indptr_start < indptr_end): # <<<<<<<<<<<<<<
|
|
* # Find index of sorted_samples[p] in X_indices
|
|
* binary_search(X_indices, indptr_start, indptr_end,
|
|
*/
|
|
while (1) {
|
|
__pyx_t_2 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (__pyx_t_2) {
|
|
__pyx_t_3 = ((__pyx_v_indptr_start < __pyx_v_indptr_end) != 0);
|
|
__pyx_t_1 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_1 = __pyx_t_2;
|
|
}
|
|
if (!__pyx_t_1) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2097
|
|
* while (p < end and indptr_start < indptr_end):
|
|
* # Find index of sorted_samples[p] in X_indices
|
|
* binary_search(X_indices, indptr_start, indptr_end, # <<<<<<<<<<<<<<
|
|
* sorted_samples[p], &k, &indptr_start)
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_binary_search(__pyx_v_X_indices, __pyx_v_indptr_start, __pyx_v_indptr_end, (__pyx_v_sorted_samples[__pyx_v_p]), (&__pyx_v_k), (&__pyx_v_indptr_start));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2100
|
|
* sorted_samples[p], &k, &indptr_start)
|
|
*
|
|
* if k != -1: # <<<<<<<<<<<<<<
|
|
* # If k != -1, we have found a non zero value
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_k != -1) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2103
|
|
* # If k != -1, we have found a non zero value
|
|
*
|
|
* if X_data[k] > 0: # <<<<<<<<<<<<<<
|
|
* start_positive_ -= 1
|
|
* Xf[start_positive_] = X_data[k]
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_X_data[__pyx_v_k]) > 0.0) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2104
|
|
*
|
|
* if X_data[k] > 0:
|
|
* start_positive_ -= 1 # <<<<<<<<<<<<<<
|
|
* Xf[start_positive_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
*/
|
|
__pyx_v_start_positive_ = (__pyx_v_start_positive_ - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2105
|
|
* if X_data[k] > 0:
|
|
* start_positive_ -= 1
|
|
* Xf[start_positive_] = X_data[k] # <<<<<<<<<<<<<<
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, start_positive_)
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_start_positive_]) = (__pyx_v_X_data[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2106
|
|
* start_positive_ -= 1
|
|
* Xf[start_positive_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]] # <<<<<<<<<<<<<<
|
|
* sparse_swap(index_to_samples, samples, index, start_positive_)
|
|
*
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index_to_samples[(__pyx_v_X_indices[__pyx_v_k])]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2107
|
|
* Xf[start_positive_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, start_positive_) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_v_index_to_samples, __pyx_v_samples, __pyx_v_index, __pyx_v_start_positive_);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2110
|
|
*
|
|
*
|
|
* elif X_data[k] < 0: # <<<<<<<<<<<<<<
|
|
* Xf[end_negative_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_X_data[__pyx_v_k]) < 0.0) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2111
|
|
*
|
|
* elif X_data[k] < 0:
|
|
* Xf[end_negative_] = X_data[k] # <<<<<<<<<<<<<<
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_end_negative_]) = (__pyx_v_X_data[__pyx_v_k]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2112
|
|
* elif X_data[k] < 0:
|
|
* Xf[end_negative_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]] # <<<<<<<<<<<<<<
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
* end_negative_ += 1
|
|
*/
|
|
__pyx_v_index = (__pyx_v_index_to_samples[(__pyx_v_X_indices[__pyx_v_k])]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2113
|
|
* Xf[end_negative_] = X_data[k]
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_) # <<<<<<<<<<<<<<
|
|
* end_negative_ += 1
|
|
* p += 1
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_v_index_to_samples, __pyx_v_samples, __pyx_v_index, __pyx_v_end_negative_);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2114
|
|
* index = index_to_samples[X_indices[k]]
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
* end_negative_ += 1 # <<<<<<<<<<<<<<
|
|
* p += 1
|
|
*
|
|
*/
|
|
__pyx_v_end_negative_ = (__pyx_v_end_negative_ + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
__pyx_L11:;
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2115
|
|
* sparse_swap(index_to_samples, samples, index, end_negative_)
|
|
* end_negative_ += 1
|
|
* p += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Returned values
|
|
*/
|
|
__pyx_v_p = (__pyx_v_p + 1);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2118
|
|
*
|
|
* # Returned values
|
|
* end_negative[0] = end_negative_ # <<<<<<<<<<<<<<
|
|
* start_positive[0] = start_positive_
|
|
*
|
|
*/
|
|
(__pyx_v_end_negative[0]) = __pyx_v_end_negative_;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2119
|
|
* # Returned values
|
|
* end_negative[0] = end_negative_
|
|
* start_positive[0] = start_positive_ # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_start_positive[0]) = __pyx_v_start_positive_;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2050
|
|
*
|
|
*
|
|
* cdef inline void extract_nnz_binary_search(INT32_t* X_indices, # <<<<<<<<<<<<<<
|
|
* DTYPE_t* X_data,
|
|
* INT32_t indptr_start,
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2122
|
|
*
|
|
*
|
|
* cdef inline void sparse_swap(SIZE_t* index_to_samples, SIZE_t* samples, # <<<<<<<<<<<<<<
|
|
* SIZE_t pos_1, SIZE_t pos_2) nogil :
|
|
* """Swap sample pos_1 and pos_2 preserving sparse invariant"""
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_7sklearn_4tree_5_tree_sparse_swap(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_pos_1, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_pos_2) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2125
|
|
* SIZE_t pos_1, SIZE_t pos_2) nogil :
|
|
* """Swap sample pos_1 and pos_2 preserving sparse invariant"""
|
|
* samples[pos_1], samples[pos_2] = samples[pos_2], samples[pos_1] # <<<<<<<<<<<<<<
|
|
* index_to_samples[samples[pos_1]] = pos_1
|
|
* index_to_samples[samples[pos_2]] = pos_2
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_samples[__pyx_v_pos_2]);
|
|
__pyx_t_2 = (__pyx_v_samples[__pyx_v_pos_1]);
|
|
(__pyx_v_samples[__pyx_v_pos_1]) = __pyx_t_1;
|
|
(__pyx_v_samples[__pyx_v_pos_2]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2126
|
|
* """Swap sample pos_1 and pos_2 preserving sparse invariant"""
|
|
* samples[pos_1], samples[pos_2] = samples[pos_2], samples[pos_1]
|
|
* index_to_samples[samples[pos_1]] = pos_1 # <<<<<<<<<<<<<<
|
|
* index_to_samples[samples[pos_2]] = pos_2
|
|
*
|
|
*/
|
|
(__pyx_v_index_to_samples[(__pyx_v_samples[__pyx_v_pos_1])]) = __pyx_v_pos_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2127
|
|
* samples[pos_1], samples[pos_2] = samples[pos_2], samples[pos_1]
|
|
* index_to_samples[samples[pos_1]] = pos_1
|
|
* index_to_samples[samples[pos_2]] = pos_2 # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_index_to_samples[(__pyx_v_samples[__pyx_v_pos_2])]) = __pyx_v_pos_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2122
|
|
*
|
|
*
|
|
* cdef inline void sparse_swap(SIZE_t* index_to_samples, SIZE_t* samples, # <<<<<<<<<<<<<<
|
|
* SIZE_t pos_1, SIZE_t pos_2) nogil :
|
|
* """Swap sample pos_1 and pos_2 preserving sparse invariant"""
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2133
|
|
* """Splitter for finding the best split, using the sparse data."""
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (BestSparseSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_18BestSparseSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_18BestSparseSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_18BestSparseSplitter___reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_18BestSparseSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2134
|
|
*
|
|
* def __reduce__(self):
|
|
* return (BestSparseSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2135
|
|
* def __reduce__(self):
|
|
* return (BestSparseSplitter, (self.criterion,
|
|
* self.max_features, # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.max_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2135; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2136
|
|
* return (BestSparseSplitter, (self.criterion,
|
|
* self.max_features,
|
|
* self.min_samples_leaf, # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__())
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2136; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2137
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf, # <<<<<<<<<<<<<<
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyFloat_FromDouble(__pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2137; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2134
|
|
*
|
|
* def __reduce__(self):
|
|
* return (BestSparseSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2134; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 3, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 4, __pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__Pyx_GIVEREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2138
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split,
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2138; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2138; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2134
|
|
*
|
|
* def __reduce__(self):
|
|
* return (BestSparseSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2134; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_BestSparseSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_BestSparseSplitter)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_BestSparseSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2133
|
|
* """Splitter for finding the best split, using the sparse data."""
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (BestSparseSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestSparseSplitter.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2140
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best split on node samples[start:end], using sparse
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_18BestSparseSplitter_node_split(struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter *__pyx_v_self, double __pyx_v_impurity, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_constant_features) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indices;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indptr;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_data;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_constant_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_sorted_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_best;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_current;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_j;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_visited_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_found_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_drawn_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_known_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p_next;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p_prev;
|
|
int __pyx_v_is_samples_sorted;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start_positive;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end_negative;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_4;
|
|
double __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2146
|
|
* """
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2147
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.start;
|
|
__pyx_v_start = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2148
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef INT32_t* X_indices = self.X_indices
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.end;
|
|
__pyx_v_end = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2150
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
* cdef INT32_t* X_indices = self.X_indices # <<<<<<<<<<<<<<
|
|
* cdef INT32_t* X_indptr = self.X_indptr
|
|
* cdef DTYPE_t* X_data = self.X_data
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X_indices;
|
|
__pyx_v_X_indices = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2151
|
|
*
|
|
* cdef INT32_t* X_indices = self.X_indices
|
|
* cdef INT32_t* X_indptr = self.X_indptr # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* X_data = self.X_data
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X_indptr;
|
|
__pyx_v_X_indptr = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2152
|
|
* cdef INT32_t* X_indices = self.X_indices
|
|
* cdef INT32_t* X_indptr = self.X_indptr
|
|
* cdef DTYPE_t* X_data = self.X_data # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.X_data;
|
|
__pyx_v_X_data = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2154
|
|
* cdef DTYPE_t* X_data = self.X_data
|
|
*
|
|
* cdef SIZE_t* features = self.features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.features;
|
|
__pyx_v_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2155
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.constant_features;
|
|
__pyx_v_constant_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2156
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_features;
|
|
__pyx_v_n_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2158
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.__pyx_base.feature_values;
|
|
__pyx_v_Xf = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2159
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t max_features = self.max_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.sorted_samples;
|
|
__pyx_v_sorted_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2160
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.index_to_samples;
|
|
__pyx_v_index_to_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2161
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t max_features = self.max_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.max_features;
|
|
__pyx_v_max_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2162
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2163
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2164
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SplitRecord best, current
|
|
*/
|
|
__pyx_v_random_state = (&__pyx_v_self->__pyx_base.__pyx_base.rand_r_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2167
|
|
*
|
|
* cdef SplitRecord best, current
|
|
* _init_split(&best, end) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t f_i = n_features
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree__init_split((&__pyx_v_best), __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2169
|
|
* _init_split(&best, end)
|
|
*
|
|
* cdef SIZE_t f_i = n_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* cdef SIZE_t n_visited_features = 0
|
|
*/
|
|
__pyx_v_f_i = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2171
|
|
* cdef SIZE_t f_i = n_features
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* cdef SIZE_t n_visited_features = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0
|
|
*/
|
|
__pyx_v_n_visited_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2173
|
|
* cdef SIZE_t n_visited_features = 0
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
*/
|
|
__pyx_v_n_found_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2175
|
|
* cdef SIZE_t n_found_constants = 0
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
*/
|
|
__pyx_v_n_drawn_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2176
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0] # <<<<<<<<<<<<<<
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
*/
|
|
__pyx_v_n_known_constants = (__pyx_v_n_constant_features[0]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2178
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t current_feature_value
|
|
*
|
|
*/
|
|
__pyx_v_n_total_constants = __pyx_v_n_known_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2183
|
|
* cdef SIZE_t p_next
|
|
* cdef SIZE_t p_prev
|
|
* cdef bint is_samples_sorted = 0 # indicate is sorted_samples is # <<<<<<<<<<<<<<
|
|
* # inititialized
|
|
*
|
|
*/
|
|
__pyx_v_is_samples_sorted = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2200
|
|
* # newly discovered constant features to spare computation on descendant
|
|
* # nodes.
|
|
* while (f_i > n_total_constants and # Stop early if remaining features # <<<<<<<<<<<<<<
|
|
* # are constant
|
|
* (n_visited_features < max_features or
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = ((__pyx_v_f_i > __pyx_v_n_total_constants) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2202
|
|
* while (f_i > n_total_constants and # Stop early if remaining features
|
|
* # are constant
|
|
* (n_visited_features < max_features or # <<<<<<<<<<<<<<
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_n_visited_features < __pyx_v_max_features) != 0);
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2204
|
|
* (n_visited_features < max_features or
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)): # <<<<<<<<<<<<<<
|
|
*
|
|
* n_visited_features += 1
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_n_visited_features <= (__pyx_v_n_found_constants + __pyx_v_n_drawn_constants)) != 0);
|
|
__pyx_t_9 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_7;
|
|
}
|
|
__pyx_t_7 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
if (!__pyx_t_7) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2206
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*
|
|
* n_visited_features += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Loop invariant: elements of features in
|
|
*/
|
|
__pyx_v_n_visited_features = (__pyx_v_n_visited_features + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2220
|
|
*
|
|
* # Draw a feature at random
|
|
* f_j = rand_int(n_drawn_constants, f_i - n_found_constants, # <<<<<<<<<<<<<<
|
|
* random_state)
|
|
*
|
|
*/
|
|
__pyx_v_f_j = __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_v_n_drawn_constants, (__pyx_v_f_i - __pyx_v_n_found_constants), __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2223
|
|
* random_state)
|
|
*
|
|
* if f_j < n_known_constants: # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_f_j < __pyx_v_n_known_constants) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2225
|
|
* if f_j < n_known_constants:
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j] # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2226
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants] # <<<<<<<<<<<<<<
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_drawn_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2227
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* n_drawn_constants += 1
|
|
*/
|
|
(__pyx_v_features[__pyx_v_n_drawn_constants]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2229
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
* n_drawn_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_drawn_constants = (__pyx_v_n_drawn_constants + 1);
|
|
goto __pyx_L5;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2233
|
|
* else:
|
|
* # f_j in the interval [n_known_constants, f_i - n_found_constants[
|
|
* f_j += n_found_constants # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
*/
|
|
__pyx_v_f_j = (__pyx_v_f_j + __pyx_v_n_found_constants);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2236
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
* current.feature = features[f_j] # <<<<<<<<<<<<<<
|
|
* self.extract_nnz(current.feature,
|
|
* &end_negative, &start_positive,
|
|
*/
|
|
__pyx_v_current.feature = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2237
|
|
*
|
|
* current.feature = features[f_j]
|
|
* self.extract_nnz(current.feature, # <<<<<<<<<<<<<<
|
|
* &end_negative, &start_positive,
|
|
* &is_samples_sorted)
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_current.feature, (&__pyx_v_end_negative), (&__pyx_v_start_positive), (&__pyx_v_is_samples_sorted));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2242
|
|
*
|
|
* # Sort the positive and negative parts of `Xf`
|
|
* sort(Xf + start, samples + start, end_negative - start) # <<<<<<<<<<<<<<
|
|
* sort(Xf + start_positive, samples + start_positive,
|
|
* end - start_positive)
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sort((__pyx_v_Xf + __pyx_v_start), (__pyx_v_samples + __pyx_v_start), (__pyx_v_end_negative - __pyx_v_start));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2243
|
|
* # Sort the positive and negative parts of `Xf`
|
|
* sort(Xf + start, samples + start, end_negative - start)
|
|
* sort(Xf + start_positive, samples + start_positive, # <<<<<<<<<<<<<<
|
|
* end - start_positive)
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_sort((__pyx_v_Xf + __pyx_v_start_positive), (__pyx_v_samples + __pyx_v_start_positive), (__pyx_v_end - __pyx_v_start_positive));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2247
|
|
*
|
|
* # Update index_to_samples to take into account the sort
|
|
* for p in range(start, end_negative): # <<<<<<<<<<<<<<
|
|
* index_to_samples[samples[p]] = p
|
|
* for p in range(start_positive, end):
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end_negative;
|
|
for (__pyx_t_10 = __pyx_v_start; __pyx_t_10 < __pyx_t_2; __pyx_t_10+=1) {
|
|
__pyx_v_p = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2248
|
|
* # Update index_to_samples to take into account the sort
|
|
* for p in range(start, end_negative):
|
|
* index_to_samples[samples[p]] = p # <<<<<<<<<<<<<<
|
|
* for p in range(start_positive, end):
|
|
* index_to_samples[samples[p]] = p
|
|
*/
|
|
(__pyx_v_index_to_samples[(__pyx_v_samples[__pyx_v_p])]) = __pyx_v_p;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2249
|
|
* for p in range(start, end_negative):
|
|
* index_to_samples[samples[p]] = p
|
|
* for p in range(start_positive, end): # <<<<<<<<<<<<<<
|
|
* index_to_samples[samples[p]] = p
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end;
|
|
for (__pyx_t_10 = __pyx_v_start_positive; __pyx_t_10 < __pyx_t_2; __pyx_t_10+=1) {
|
|
__pyx_v_p = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2250
|
|
* index_to_samples[samples[p]] = p
|
|
* for p in range(start_positive, end):
|
|
* index_to_samples[samples[p]] = p # <<<<<<<<<<<<<<
|
|
*
|
|
* # Add one or two zeros in Xf, if there is any
|
|
*/
|
|
(__pyx_v_index_to_samples[(__pyx_v_samples[__pyx_v_p])]) = __pyx_v_p;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2253
|
|
*
|
|
* # Add one or two zeros in Xf, if there is any
|
|
* if end_negative < start_positive: # <<<<<<<<<<<<<<
|
|
* start_positive -= 1
|
|
* Xf[start_positive] = 0.
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_end_negative < __pyx_v_start_positive) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2254
|
|
* # Add one or two zeros in Xf, if there is any
|
|
* if end_negative < start_positive:
|
|
* start_positive -= 1 # <<<<<<<<<<<<<<
|
|
* Xf[start_positive] = 0.
|
|
*
|
|
*/
|
|
__pyx_v_start_positive = (__pyx_v_start_positive - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2255
|
|
* if end_negative < start_positive:
|
|
* start_positive -= 1
|
|
* Xf[start_positive] = 0. # <<<<<<<<<<<<<<
|
|
*
|
|
* if end_negative != start_positive:
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_start_positive]) = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2257
|
|
* Xf[start_positive] = 0.
|
|
*
|
|
* if end_negative != start_positive: # <<<<<<<<<<<<<<
|
|
* Xf[end_negative] = 0.
|
|
* end_negative += 1
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_end_negative != __pyx_v_start_positive) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2258
|
|
*
|
|
* if end_negative != start_positive:
|
|
* Xf[end_negative] = 0. # <<<<<<<<<<<<<<
|
|
* end_negative += 1
|
|
*
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_end_negative]) = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2259
|
|
* if end_negative != start_positive:
|
|
* Xf[end_negative] = 0.
|
|
* end_negative += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
*/
|
|
__pyx_v_end_negative = (__pyx_v_end_negative + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
__pyx_L11:;
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2261
|
|
* end_negative += 1
|
|
*
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD: # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature
|
|
*/
|
|
__pyx_t_7 = (((__pyx_v_Xf[(__pyx_v_end - 1)]) <= ((__pyx_v_Xf[__pyx_v_start]) + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2262
|
|
*
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants] # <<<<<<<<<<<<<<
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_total_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2263
|
|
* if Xf[end - 1] <= Xf[start] + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature # <<<<<<<<<<<<<<
|
|
*
|
|
* n_found_constants += 1
|
|
*/
|
|
__pyx_t_2 = __pyx_v_current.feature;
|
|
(__pyx_v_features[__pyx_v_n_total_constants]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2265
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
* n_found_constants += 1 # <<<<<<<<<<<<<<
|
|
* n_total_constants += 1
|
|
*
|
|
*/
|
|
__pyx_v_n_found_constants = (__pyx_v_n_found_constants + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2266
|
|
*
|
|
* n_found_constants += 1
|
|
* n_total_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_total_constants = (__pyx_v_n_total_constants + 1);
|
|
goto __pyx_L12;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2269
|
|
*
|
|
* else:
|
|
* f_i -= 1 # <<<<<<<<<<<<<<
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i]
|
|
*
|
|
*/
|
|
__pyx_v_f_i = (__pyx_v_f_i - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2270
|
|
* else:
|
|
* f_i -= 1
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Evaluate all splits
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_features[__pyx_v_f_j]);
|
|
__pyx_t_10 = (__pyx_v_features[__pyx_v_f_i]);
|
|
(__pyx_v_features[__pyx_v_f_i]) = __pyx_t_2;
|
|
(__pyx_v_features[__pyx_v_f_j]) = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2273
|
|
*
|
|
* # Evaluate all splits
|
|
* self.criterion.reset() # <<<<<<<<<<<<<<
|
|
* p = start
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->reset(__pyx_v_self->__pyx_base.__pyx_base.criterion);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2274
|
|
* # Evaluate all splits
|
|
* self.criterion.reset()
|
|
* p = start # <<<<<<<<<<<<<<
|
|
*
|
|
* while p < end:
|
|
*/
|
|
__pyx_v_p = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2276
|
|
* p = start
|
|
*
|
|
* while p < end: # <<<<<<<<<<<<<<
|
|
* if p + 1 != end_negative:
|
|
* p_next = p + 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_7 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (!__pyx_t_7) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2277
|
|
*
|
|
* while p < end:
|
|
* if p + 1 != end_negative: # <<<<<<<<<<<<<<
|
|
* p_next = p + 1
|
|
* else:
|
|
*/
|
|
__pyx_t_7 = (((__pyx_v_p + 1) != __pyx_v_end_negative) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2278
|
|
* while p < end:
|
|
* if p + 1 != end_negative:
|
|
* p_next = p + 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* p_next = start_positive
|
|
*/
|
|
__pyx_v_p_next = (__pyx_v_p + 1);
|
|
goto __pyx_L15;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2280
|
|
* p_next = p + 1
|
|
* else:
|
|
* p_next = start_positive # <<<<<<<<<<<<<<
|
|
*
|
|
* while (p_next < end and
|
|
*/
|
|
__pyx_v_p_next = __pyx_v_start_positive;
|
|
}
|
|
__pyx_L15:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2282
|
|
* p_next = start_positive
|
|
*
|
|
* while (p_next < end and # <<<<<<<<<<<<<<
|
|
* Xf[p_next] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p = p_next
|
|
*/
|
|
while (1) {
|
|
__pyx_t_7 = ((__pyx_v_p_next < __pyx_v_end) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2283
|
|
*
|
|
* while (p_next < end and
|
|
* Xf[p_next] <= Xf[p] + FEATURE_THRESHOLD): # <<<<<<<<<<<<<<
|
|
* p = p_next
|
|
* if p + 1 != end_negative:
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_Xf[__pyx_v_p_next]) <= ((__pyx_v_Xf[__pyx_v_p]) + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
__pyx_t_9 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_7;
|
|
}
|
|
if (!__pyx_t_9) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2284
|
|
* while (p_next < end and
|
|
* Xf[p_next] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p = p_next # <<<<<<<<<<<<<<
|
|
* if p + 1 != end_negative:
|
|
* p_next = p + 1
|
|
*/
|
|
__pyx_v_p = __pyx_v_p_next;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2285
|
|
* Xf[p_next] <= Xf[p] + FEATURE_THRESHOLD):
|
|
* p = p_next
|
|
* if p + 1 != end_negative: # <<<<<<<<<<<<<<
|
|
* p_next = p + 1
|
|
* else:
|
|
*/
|
|
__pyx_t_9 = (((__pyx_v_p + 1) != __pyx_v_end_negative) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2286
|
|
* p = p_next
|
|
* if p + 1 != end_negative:
|
|
* p_next = p + 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* p_next = start_positive
|
|
*/
|
|
__pyx_v_p_next = (__pyx_v_p + 1);
|
|
goto __pyx_L18;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2288
|
|
* p_next = p + 1
|
|
* else:
|
|
* p_next = start_positive # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_v_p_next = __pyx_v_start_positive;
|
|
}
|
|
__pyx_L18:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2293
|
|
* # (p_next >= end) or (X[samples[p_next], current.feature] >
|
|
* # X[samples[p], current.feature])
|
|
* p_prev = p # <<<<<<<<<<<<<<
|
|
* p = p_next
|
|
* # (p >= end) or (X[samples[p], current.feature] >
|
|
*/
|
|
__pyx_v_p_prev = __pyx_v_p;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2294
|
|
* # X[samples[p], current.feature])
|
|
* p_prev = p
|
|
* p = p_next # <<<<<<<<<<<<<<
|
|
* # (p >= end) or (X[samples[p], current.feature] >
|
|
* # X[samples[p_prev], current.feature])
|
|
*/
|
|
__pyx_v_p = __pyx_v_p_next;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2299
|
|
*
|
|
*
|
|
* if p < end: # <<<<<<<<<<<<<<
|
|
* current.pos = p
|
|
*
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_p < __pyx_v_end) != 0);
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2300
|
|
*
|
|
* if p < end:
|
|
* current.pos = p # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
*/
|
|
__pyx_v_current.pos = __pyx_v_p;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2303
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_9 = (((__pyx_v_current.pos - __pyx_v_start) < __pyx_v_min_samples_leaf) != 0);
|
|
if (!__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2304
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_7 = (((__pyx_v_end - __pyx_v_current.pos) < __pyx_v_min_samples_leaf) != 0);
|
|
__pyx_t_6 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_9;
|
|
}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2305
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* self.criterion.update(current.pos)
|
|
*/
|
|
goto __pyx_L13_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2307
|
|
* continue
|
|
*
|
|
* self.criterion.update(current.pos) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->update(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_current.pos);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2310
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or # <<<<<<<<<<<<<<
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_left < __pyx_v_min_weight_leaf) != 0);
|
|
if (!__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2311
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_right < __pyx_v_min_weight_leaf) != 0);
|
|
__pyx_t_7 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2312
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*/
|
|
goto __pyx_L13_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2314
|
|
* continue
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity) # <<<<<<<<<<<<<<
|
|
* if current.improvement > best.improvement:
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
*/
|
|
__pyx_v_current.improvement = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->impurity_improvement(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_impurity);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2315
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
* if current.improvement > best.improvement: # <<<<<<<<<<<<<<
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current.improvement > __pyx_v_best.improvement) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2316
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
* if current.improvement > best.improvement:
|
|
* self.criterion.children_impurity(¤t.impurity_left, # <<<<<<<<<<<<<<
|
|
* ¤t.impurity_right)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->children_impurity(__pyx_v_self->__pyx_base.__pyx_base.criterion, (&__pyx_v_current.impurity_left), (&__pyx_v_current.impurity_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2319
|
|
* ¤t.impurity_right)
|
|
*
|
|
* current.threshold = (Xf[p_prev] + Xf[p]) / 2.0 # <<<<<<<<<<<<<<
|
|
* if current.threshold == Xf[p]:
|
|
* current.threshold = Xf[p_prev]
|
|
*/
|
|
__pyx_v_current.threshold = (((__pyx_v_Xf[__pyx_v_p_prev]) + (__pyx_v_Xf[__pyx_v_p])) / 2.0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2320
|
|
*
|
|
* current.threshold = (Xf[p_prev] + Xf[p]) / 2.0
|
|
* if current.threshold == Xf[p]: # <<<<<<<<<<<<<<
|
|
* current.threshold = Xf[p_prev]
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current.threshold == (__pyx_v_Xf[__pyx_v_p])) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2321
|
|
* current.threshold = (Xf[p_prev] + Xf[p]) / 2.0
|
|
* if current.threshold == Xf[p]:
|
|
* current.threshold = Xf[p_prev] # <<<<<<<<<<<<<<
|
|
*
|
|
* best = current
|
|
*/
|
|
__pyx_v_current.threshold = (__pyx_v_Xf[__pyx_v_p_prev]);
|
|
goto __pyx_L23;
|
|
}
|
|
__pyx_L23:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2323
|
|
* current.threshold = Xf[p_prev]
|
|
*
|
|
* best = current # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
*/
|
|
__pyx_v_best = __pyx_v_current;
|
|
goto __pyx_L22;
|
|
}
|
|
__pyx_L22:;
|
|
goto __pyx_L19;
|
|
}
|
|
__pyx_L19:;
|
|
__pyx_L13_continue:;
|
|
}
|
|
}
|
|
__pyx_L12:;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2326
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end: # <<<<<<<<<<<<<<
|
|
* self.extract_nnz(best.feature, &end_negative, &start_positive,
|
|
* &is_samples_sorted)
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_best.pos < __pyx_v_end) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2327
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end:
|
|
* self.extract_nnz(best.feature, &end_negative, &start_positive, # <<<<<<<<<<<<<<
|
|
* &is_samples_sorted)
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_best.feature, (&__pyx_v_end_negative), (&__pyx_v_start_positive), (&__pyx_v_is_samples_sorted));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2330
|
|
* &is_samples_sorted)
|
|
*
|
|
* self._partition(best.threshold, end_negative, start_positive, # <<<<<<<<<<<<<<
|
|
* best.pos)
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter__partition(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_best.threshold, __pyx_v_end_negative, __pyx_v_start_positive, __pyx_v_best.pos);
|
|
goto __pyx_L24;
|
|
}
|
|
__pyx_L24:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2336
|
|
* # element in features[:n_known_constants] must be preserved for sibling
|
|
* # and child nodes
|
|
* memcpy(features, constant_features, sizeof(SIZE_t) * n_known_constants) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Copy newly found constant features
|
|
*/
|
|
memcpy(__pyx_v_features, __pyx_v_constant_features, ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_known_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2339
|
|
*
|
|
* # Copy newly found constant features
|
|
* memcpy(constant_features + n_known_constants, # <<<<<<<<<<<<<<
|
|
* features + n_known_constants,
|
|
* sizeof(SIZE_t) * n_found_constants)
|
|
*/
|
|
memcpy((__pyx_v_constant_features + __pyx_v_n_known_constants), (__pyx_v_features + __pyx_v_n_known_constants), ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_found_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2344
|
|
*
|
|
* # Return values
|
|
* split[0] = best # <<<<<<<<<<<<<<
|
|
* n_constant_features[0] = n_total_constants
|
|
*
|
|
*/
|
|
(__pyx_v_split[0]) = __pyx_v_best;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2345
|
|
* # Return values
|
|
* split[0] = best
|
|
* n_constant_features[0] = n_total_constants # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_n_constant_features[0]) = __pyx_v_n_total_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2140
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find the best split on node samples[start:end], using sparse
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2351
|
|
* """Splitter for finding a random split, using the sparse data."""
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (RandomSparseSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_20RandomSparseSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_20RandomSparseSplitter_1__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_20RandomSparseSplitter___reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_20RandomSparseSplitter___reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2352
|
|
*
|
|
* def __reduce__(self):
|
|
* return (RandomSparseSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2353
|
|
* def __reduce__(self):
|
|
* return (RandomSparseSplitter, (self.criterion,
|
|
* self.max_features, # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.max_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2353; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2354
|
|
* return (RandomSparseSplitter, (self.criterion,
|
|
* self.max_features,
|
|
* self.min_samples_leaf, # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__())
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2354; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2355
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf, # <<<<<<<<<<<<<<
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyFloat_FromDouble(__pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2355; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2352
|
|
*
|
|
* def __reduce__(self):
|
|
* return (RandomSparseSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self->__pyx_base.__pyx_base.criterion));
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 3, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 4, __pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__Pyx_GIVEREF(__pyx_v_self->__pyx_base.__pyx_base.random_state);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2356
|
|
* self.min_samples_leaf,
|
|
* self.min_weight_leaf,
|
|
* self.random_state), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split,
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2356; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2356; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2352
|
|
*
|
|
* def __reduce__(self):
|
|
* return (RandomSparseSplitter, (self.criterion, # <<<<<<<<<<<<<<
|
|
* self.max_features,
|
|
* self.min_samples_leaf,
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RandomSparseSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RandomSparseSplitter)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_RandomSparseSplitter)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2351
|
|
* """Splitter for finding a random split, using the sparse data."""
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* return (RandomSparseSplitter, (self.criterion,
|
|
* self.max_features,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.RandomSparseSplitter.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2358
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find a random split on node samples[start:end], using sparse
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_20RandomSparseSplitter_node_split(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter *__pyx_v_self, double __pyx_v_impurity, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *__pyx_v_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_n_constant_features) {
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indices;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indptr;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_data;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_constant_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_Xf;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_sorted_samples;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_index_to_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_best;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_current;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_current_feature_value;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_f_j;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_p;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_visited_features;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_found_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_drawn_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_known_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_total_constants;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_min_feature_value;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_max_feature_value;
|
|
int __pyx_v_is_samples_sorted;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start_positive;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end_negative;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_4;
|
|
double __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2364
|
|
* """
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.samples;
|
|
__pyx_v_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2365
|
|
* # Find the best split
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.start;
|
|
__pyx_v_start = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2366
|
|
* cdef SIZE_t* samples = self.samples
|
|
* cdef SIZE_t start = self.start
|
|
* cdef SIZE_t end = self.end # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef INT32_t* X_indices = self.X_indices
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.end;
|
|
__pyx_v_end = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2368
|
|
* cdef SIZE_t end = self.end
|
|
*
|
|
* cdef INT32_t* X_indices = self.X_indices # <<<<<<<<<<<<<<
|
|
* cdef INT32_t* X_indptr = self.X_indptr
|
|
* cdef DTYPE_t* X_data = self.X_data
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X_indices;
|
|
__pyx_v_X_indices = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2369
|
|
*
|
|
* cdef INT32_t* X_indices = self.X_indices
|
|
* cdef INT32_t* X_indptr = self.X_indptr # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* X_data = self.X_data
|
|
*
|
|
*/
|
|
__pyx_t_3 = __pyx_v_self->__pyx_base.X_indptr;
|
|
__pyx_v_X_indptr = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2370
|
|
* cdef INT32_t* X_indices = self.X_indices
|
|
* cdef INT32_t* X_indptr = self.X_indptr
|
|
* cdef DTYPE_t* X_data = self.X_data # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.X_data;
|
|
__pyx_v_X_data = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2372
|
|
* cdef DTYPE_t* X_data = self.X_data
|
|
*
|
|
* cdef SIZE_t* features = self.features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.features;
|
|
__pyx_v_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2373
|
|
*
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.__pyx_base.constant_features;
|
|
__pyx_v_constant_features = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2374
|
|
* cdef SIZE_t* features = self.features
|
|
* cdef SIZE_t* constant_features = self.constant_features
|
|
* cdef SIZE_t n_features = self.n_features # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.n_features;
|
|
__pyx_v_n_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2376
|
|
* cdef SIZE_t n_features = self.n_features
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->__pyx_base.__pyx_base.feature_values;
|
|
__pyx_v_Xf = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2377
|
|
*
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t max_features = self.max_features
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.sorted_samples;
|
|
__pyx_v_sorted_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2378
|
|
* cdef DTYPE_t* Xf = self.feature_values
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->__pyx_base.index_to_samples;
|
|
__pyx_v_index_to_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2379
|
|
* cdef SIZE_t* sorted_samples = self.sorted_samples
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t max_features = self.max_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.max_features;
|
|
__pyx_v_max_features = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2380
|
|
* cdef SIZE_t* index_to_samples = self.index_to_samples
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*/
|
|
__pyx_t_2 = __pyx_v_self->__pyx_base.__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2381
|
|
* cdef SIZE_t max_features = self.max_features
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef UINT32_t* random_state = &self.rand_r_state
|
|
*
|
|
*/
|
|
__pyx_t_5 = __pyx_v_self->__pyx_base.__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2382
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef UINT32_t* random_state = &self.rand_r_state # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SplitRecord best, current
|
|
*/
|
|
__pyx_v_random_state = (&__pyx_v_self->__pyx_base.__pyx_base.rand_r_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2385
|
|
*
|
|
* cdef SplitRecord best, current
|
|
* _init_split(&best, end) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t current_feature_value
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree__init_split((&__pyx_v_best), __pyx_v_end);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2389
|
|
* cdef DTYPE_t current_feature_value
|
|
*
|
|
* cdef SIZE_t f_i = n_features # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* cdef SIZE_t n_visited_features = 0
|
|
*/
|
|
__pyx_v_f_i = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2391
|
|
* cdef SIZE_t f_i = n_features
|
|
* cdef SIZE_t f_j, p, tmp
|
|
* cdef SIZE_t n_visited_features = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0
|
|
*/
|
|
__pyx_v_n_visited_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2393
|
|
* cdef SIZE_t n_visited_features = 0
|
|
* # Number of features discovered to be constant during the split search
|
|
* cdef SIZE_t n_found_constants = 0 # <<<<<<<<<<<<<<
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
*/
|
|
__pyx_v_n_found_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2395
|
|
* cdef SIZE_t n_found_constants = 0
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
*/
|
|
__pyx_v_n_drawn_constants = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2396
|
|
* # Number of features known to be constant and drawn without replacement
|
|
* cdef SIZE_t n_drawn_constants = 0
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0] # <<<<<<<<<<<<<<
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants
|
|
*/
|
|
__pyx_v_n_known_constants = (__pyx_v_n_constant_features[0]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2398
|
|
* cdef SIZE_t n_known_constants = n_constant_features[0]
|
|
* # n_total_constants = n_known_constants + n_found_constants
|
|
* cdef SIZE_t n_total_constants = n_known_constants # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t partition_end
|
|
*
|
|
*/
|
|
__pyx_v_n_total_constants = __pyx_v_n_known_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2404
|
|
* cdef DTYPE_t max_feature_value
|
|
*
|
|
* cdef bint is_samples_sorted = 0 # indicate that sorted_samples is # <<<<<<<<<<<<<<
|
|
* # inititialized
|
|
*
|
|
*/
|
|
__pyx_v_is_samples_sorted = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2421
|
|
* # newly discovered constant features to spare computation on descendant
|
|
* # nodes.
|
|
* while (f_i > n_total_constants and # Stop early if remaining features # <<<<<<<<<<<<<<
|
|
* # are constant
|
|
* (n_visited_features < max_features or
|
|
*/
|
|
while (1) {
|
|
__pyx_t_6 = ((__pyx_v_f_i > __pyx_v_n_total_constants) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2423
|
|
* while (f_i > n_total_constants and # Stop early if remaining features
|
|
* # are constant
|
|
* (n_visited_features < max_features or # <<<<<<<<<<<<<<
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_n_visited_features < __pyx_v_max_features) != 0);
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2425
|
|
* (n_visited_features < max_features or
|
|
* # At least one drawn features must be non constant
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)): # <<<<<<<<<<<<<<
|
|
*
|
|
* n_visited_features += 1
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_n_visited_features <= (__pyx_v_n_found_constants + __pyx_v_n_drawn_constants)) != 0);
|
|
__pyx_t_9 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_7;
|
|
}
|
|
__pyx_t_7 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
if (!__pyx_t_7) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2427
|
|
* n_visited_features <= n_found_constants + n_drawn_constants)):
|
|
*
|
|
* n_visited_features += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Loop invariant: elements of features in
|
|
*/
|
|
__pyx_v_n_visited_features = (__pyx_v_n_visited_features + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2441
|
|
*
|
|
* # Draw a feature at random
|
|
* f_j = rand_int(n_drawn_constants, f_i - n_found_constants, # <<<<<<<<<<<<<<
|
|
* random_state)
|
|
*
|
|
*/
|
|
__pyx_v_f_j = __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_v_n_drawn_constants, (__pyx_v_f_i - __pyx_v_n_found_constants), __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2444
|
|
* random_state)
|
|
*
|
|
* if f_j < n_known_constants: # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_f_j < __pyx_v_n_known_constants) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2446
|
|
* if f_j < n_known_constants:
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j] # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp
|
|
*/
|
|
__pyx_v_tmp = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2447
|
|
* # f_j in the interval [n_drawn_constants, n_known_constants[
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants] # <<<<<<<<<<<<<<
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_drawn_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2448
|
|
* tmp = features[f_j]
|
|
* features[f_j] = features[n_drawn_constants]
|
|
* features[n_drawn_constants] = tmp # <<<<<<<<<<<<<<
|
|
*
|
|
* n_drawn_constants += 1
|
|
*/
|
|
(__pyx_v_features[__pyx_v_n_drawn_constants]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2450
|
|
* features[n_drawn_constants] = tmp
|
|
*
|
|
* n_drawn_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_drawn_constants = (__pyx_v_n_drawn_constants + 1);
|
|
goto __pyx_L5;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2454
|
|
* else:
|
|
* # f_j in the interval [n_known_constants, f_i - n_found_constants[
|
|
* f_j += n_found_constants # <<<<<<<<<<<<<<
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
*/
|
|
__pyx_v_f_j = (__pyx_v_f_j + __pyx_v_n_found_constants);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2457
|
|
* # f_j in the interval [n_total_constants, f_i[
|
|
*
|
|
* current.feature = features[f_j] # <<<<<<<<<<<<<<
|
|
*
|
|
* self.extract_nnz(current.feature,
|
|
*/
|
|
__pyx_v_current.feature = (__pyx_v_features[__pyx_v_f_j]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2459
|
|
* current.feature = features[f_j]
|
|
*
|
|
* self.extract_nnz(current.feature, # <<<<<<<<<<<<<<
|
|
* &end_negative, &start_positive,
|
|
* &is_samples_sorted)
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_current.feature, (&__pyx_v_end_negative), (&__pyx_v_start_positive), (&__pyx_v_is_samples_sorted));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2464
|
|
*
|
|
* # Add one or two zeros in Xf, if there is any
|
|
* if end_negative < start_positive: # <<<<<<<<<<<<<<
|
|
* start_positive -= 1
|
|
* Xf[start_positive] = 0.
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_end_negative < __pyx_v_start_positive) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2465
|
|
* # Add one or two zeros in Xf, if there is any
|
|
* if end_negative < start_positive:
|
|
* start_positive -= 1 # <<<<<<<<<<<<<<
|
|
* Xf[start_positive] = 0.
|
|
*
|
|
*/
|
|
__pyx_v_start_positive = (__pyx_v_start_positive - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2466
|
|
* if end_negative < start_positive:
|
|
* start_positive -= 1
|
|
* Xf[start_positive] = 0. # <<<<<<<<<<<<<<
|
|
*
|
|
* if end_negative != start_positive:
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_start_positive]) = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2468
|
|
* Xf[start_positive] = 0.
|
|
*
|
|
* if end_negative != start_positive: # <<<<<<<<<<<<<<
|
|
* Xf[end_negative] = 0.
|
|
* end_negative += 1
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_end_negative != __pyx_v_start_positive) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2469
|
|
*
|
|
* if end_negative != start_positive:
|
|
* Xf[end_negative] = 0. # <<<<<<<<<<<<<<
|
|
* end_negative += 1
|
|
*
|
|
*/
|
|
(__pyx_v_Xf[__pyx_v_end_negative]) = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2470
|
|
* if end_negative != start_positive:
|
|
* Xf[end_negative] = 0.
|
|
* end_negative += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Find min, max in Xf[start:end_negative]
|
|
*/
|
|
__pyx_v_end_negative = (__pyx_v_end_negative + 1);
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
goto __pyx_L6;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2473
|
|
*
|
|
* # Find min, max in Xf[start:end_negative]
|
|
* min_feature_value = Xf[start] # <<<<<<<<<<<<<<
|
|
* max_feature_value = min_feature_value
|
|
*
|
|
*/
|
|
__pyx_v_min_feature_value = (__pyx_v_Xf[__pyx_v_start]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2474
|
|
* # Find min, max in Xf[start:end_negative]
|
|
* min_feature_value = Xf[start]
|
|
* max_feature_value = min_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* for p in range(start, end_negative):
|
|
*/
|
|
__pyx_v_max_feature_value = __pyx_v_min_feature_value;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2476
|
|
* max_feature_value = min_feature_value
|
|
*
|
|
* for p in range(start, end_negative): # <<<<<<<<<<<<<<
|
|
* current_feature_value = Xf[p]
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end_negative;
|
|
for (__pyx_t_10 = __pyx_v_start; __pyx_t_10 < __pyx_t_2; __pyx_t_10+=1) {
|
|
__pyx_v_p = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2477
|
|
*
|
|
* for p in range(start, end_negative):
|
|
* current_feature_value = Xf[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if current_feature_value < min_feature_value:
|
|
*/
|
|
__pyx_v_current_feature_value = (__pyx_v_Xf[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2479
|
|
* current_feature_value = Xf[p]
|
|
*
|
|
* if current_feature_value < min_feature_value: # <<<<<<<<<<<<<<
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value:
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current_feature_value < __pyx_v_min_feature_value) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2480
|
|
*
|
|
* if current_feature_value < min_feature_value:
|
|
* min_feature_value = current_feature_value # <<<<<<<<<<<<<<
|
|
* elif current_feature_value > max_feature_value:
|
|
* max_feature_value = current_feature_value
|
|
*/
|
|
__pyx_v_min_feature_value = __pyx_v_current_feature_value;
|
|
goto __pyx_L10;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2481
|
|
* if current_feature_value < min_feature_value:
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value: # <<<<<<<<<<<<<<
|
|
* max_feature_value = current_feature_value
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current_feature_value > __pyx_v_max_feature_value) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2482
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value:
|
|
* max_feature_value = current_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* # Update min, max given Xf[start_positive:end]
|
|
*/
|
|
__pyx_v_max_feature_value = __pyx_v_current_feature_value;
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2485
|
|
*
|
|
* # Update min, max given Xf[start_positive:end]
|
|
* for p in range(start_positive, end): # <<<<<<<<<<<<<<
|
|
* current_feature_value = Xf[p]
|
|
*
|
|
*/
|
|
__pyx_t_2 = __pyx_v_end;
|
|
for (__pyx_t_10 = __pyx_v_start_positive; __pyx_t_10 < __pyx_t_2; __pyx_t_10+=1) {
|
|
__pyx_v_p = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2486
|
|
* # Update min, max given Xf[start_positive:end]
|
|
* for p in range(start_positive, end):
|
|
* current_feature_value = Xf[p] # <<<<<<<<<<<<<<
|
|
*
|
|
* if current_feature_value < min_feature_value:
|
|
*/
|
|
__pyx_v_current_feature_value = (__pyx_v_Xf[__pyx_v_p]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2488
|
|
* current_feature_value = Xf[p]
|
|
*
|
|
* if current_feature_value < min_feature_value: # <<<<<<<<<<<<<<
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value:
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current_feature_value < __pyx_v_min_feature_value) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2489
|
|
*
|
|
* if current_feature_value < min_feature_value:
|
|
* min_feature_value = current_feature_value # <<<<<<<<<<<<<<
|
|
* elif current_feature_value > max_feature_value:
|
|
* max_feature_value = current_feature_value
|
|
*/
|
|
__pyx_v_min_feature_value = __pyx_v_current_feature_value;
|
|
goto __pyx_L13;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2490
|
|
* if current_feature_value < min_feature_value:
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value: # <<<<<<<<<<<<<<
|
|
* max_feature_value = current_feature_value
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current_feature_value > __pyx_v_max_feature_value) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2491
|
|
* min_feature_value = current_feature_value
|
|
* elif current_feature_value > max_feature_value:
|
|
* max_feature_value = current_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD:
|
|
*/
|
|
__pyx_v_max_feature_value = __pyx_v_current_feature_value;
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2493
|
|
* max_feature_value = current_feature_value
|
|
*
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD: # <<<<<<<<<<<<<<
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_max_feature_value <= (__pyx_v_min_feature_value + __pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD)) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2494
|
|
*
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants] # <<<<<<<<<<<<<<
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
*/
|
|
(__pyx_v_features[__pyx_v_f_j]) = (__pyx_v_features[__pyx_v_n_total_constants]);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2495
|
|
* if max_feature_value <= min_feature_value + FEATURE_THRESHOLD:
|
|
* features[f_j] = features[n_total_constants]
|
|
* features[n_total_constants] = current.feature # <<<<<<<<<<<<<<
|
|
*
|
|
* n_found_constants += 1
|
|
*/
|
|
__pyx_t_2 = __pyx_v_current.feature;
|
|
(__pyx_v_features[__pyx_v_n_total_constants]) = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2497
|
|
* features[n_total_constants] = current.feature
|
|
*
|
|
* n_found_constants += 1 # <<<<<<<<<<<<<<
|
|
* n_total_constants += 1
|
|
*
|
|
*/
|
|
__pyx_v_n_found_constants = (__pyx_v_n_found_constants + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2498
|
|
*
|
|
* n_found_constants += 1
|
|
* n_total_constants += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_n_total_constants = (__pyx_v_n_total_constants + 1);
|
|
goto __pyx_L14;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2501
|
|
*
|
|
* else:
|
|
* f_i -= 1 # <<<<<<<<<<<<<<
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i]
|
|
*
|
|
*/
|
|
__pyx_v_f_i = (__pyx_v_f_i - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2502
|
|
* else:
|
|
* f_i -= 1
|
|
* features[f_i], features[f_j] = features[f_j], features[f_i] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Draw a random threshold
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_features[__pyx_v_f_j]);
|
|
__pyx_t_10 = (__pyx_v_features[__pyx_v_f_i]);
|
|
(__pyx_v_features[__pyx_v_f_i]) = __pyx_t_2;
|
|
(__pyx_v_features[__pyx_v_f_j]) = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2505
|
|
*
|
|
* # Draw a random threshold
|
|
* current.threshold = rand_uniform(min_feature_value, # <<<<<<<<<<<<<<
|
|
* max_feature_value,
|
|
* random_state)
|
|
*/
|
|
__pyx_v_current.threshold = __pyx_f_7sklearn_4tree_5_tree_rand_uniform(__pyx_v_min_feature_value, __pyx_v_max_feature_value, __pyx_v_random_state);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2509
|
|
* random_state)
|
|
*
|
|
* if current.threshold == max_feature_value: # <<<<<<<<<<<<<<
|
|
* current.threshold = min_feature_value
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_current.threshold == __pyx_v_max_feature_value) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2510
|
|
*
|
|
* if current.threshold == max_feature_value:
|
|
* current.threshold = min_feature_value # <<<<<<<<<<<<<<
|
|
*
|
|
* # Partition
|
|
*/
|
|
__pyx_v_current.threshold = __pyx_v_min_feature_value;
|
|
goto __pyx_L15;
|
|
}
|
|
__pyx_L15:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2513
|
|
*
|
|
* # Partition
|
|
* current.pos = self._partition(current.threshold, # <<<<<<<<<<<<<<
|
|
* end_negative,
|
|
* start_positive,
|
|
*/
|
|
__pyx_v_current.pos = __pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter__partition(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_current.threshold, __pyx_v_end_negative, __pyx_v_start_positive, (__pyx_v_start_positive + ((__pyx_v_Xf[__pyx_v_start_positive]) == 0.)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2520
|
|
*
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_7 = (((__pyx_v_current.pos - __pyx_v_start) < __pyx_v_min_samples_leaf) != 0);
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2521
|
|
* # Reject if min_samples_leaf is not guaranteed
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_end - __pyx_v_current.pos) < __pyx_v_min_samples_leaf) != 0);
|
|
__pyx_t_9 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_7;
|
|
}
|
|
if (__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2522
|
|
* if (((current.pos - start) < min_samples_leaf) or
|
|
* ((end - current.pos) < min_samples_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* # Evaluate split
|
|
*/
|
|
goto __pyx_L3_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2525
|
|
*
|
|
* # Evaluate split
|
|
* self.criterion.reset() # <<<<<<<<<<<<<<
|
|
* self.criterion.update(current.pos)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->reset(__pyx_v_self->__pyx_base.__pyx_base.criterion);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2526
|
|
* # Evaluate split
|
|
* self.criterion.reset()
|
|
* self.criterion.update(current.pos) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->update(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_current.pos);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2529
|
|
*
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or # <<<<<<<<<<<<<<
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue
|
|
*/
|
|
__pyx_t_9 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_left < __pyx_v_min_weight_leaf) != 0);
|
|
if (!__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2530
|
|
* # Reject if min_weight_leaf is not satisfied
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)): # <<<<<<<<<<<<<<
|
|
* continue
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_self->__pyx_base.__pyx_base.criterion->weighted_n_right < __pyx_v_min_weight_leaf) != 0);
|
|
__pyx_t_6 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_9;
|
|
}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2531
|
|
* if ((self.criterion.weighted_n_left < min_weight_leaf) or
|
|
* (self.criterion.weighted_n_right < min_weight_leaf)):
|
|
* continue # <<<<<<<<<<<<<<
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*/
|
|
goto __pyx_L3_continue;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2533
|
|
* continue
|
|
*
|
|
* current.improvement = self.criterion.impurity_improvement(impurity) # <<<<<<<<<<<<<<
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
*/
|
|
__pyx_v_current.improvement = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->impurity_improvement(__pyx_v_self->__pyx_base.__pyx_base.criterion, __pyx_v_impurity);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2535
|
|
* current.improvement = self.criterion.impurity_improvement(impurity)
|
|
*
|
|
* if current.improvement > best.improvement: # <<<<<<<<<<<<<<
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_current.improvement > __pyx_v_best.improvement) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2536
|
|
*
|
|
* if current.improvement > best.improvement:
|
|
* self.criterion.children_impurity(¤t.impurity_left, # <<<<<<<<<<<<<<
|
|
* ¤t.impurity_right)
|
|
* best = current
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self->__pyx_base.__pyx_base.criterion->__pyx_vtab)->children_impurity(__pyx_v_self->__pyx_base.__pyx_base.criterion, (&__pyx_v_current.impurity_left), (&__pyx_v_current.impurity_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2538
|
|
* self.criterion.children_impurity(¤t.impurity_left,
|
|
* ¤t.impurity_right)
|
|
* best = current # <<<<<<<<<<<<<<
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
*/
|
|
__pyx_v_best = __pyx_v_current;
|
|
goto __pyx_L18;
|
|
}
|
|
__pyx_L18:;
|
|
}
|
|
__pyx_L14:;
|
|
}
|
|
__pyx_L5:;
|
|
__pyx_L3_continue:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2541
|
|
*
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end and current.feature != best.feature: # <<<<<<<<<<<<<<
|
|
* self.extract_nnz(best.feature, &end_negative, &start_positive,
|
|
* &is_samples_sorted)
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_best.pos < __pyx_v_end) != 0);
|
|
if (__pyx_t_6) {
|
|
__pyx_t_9 = ((__pyx_v_current.feature != __pyx_v_best.feature) != 0);
|
|
__pyx_t_7 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2542
|
|
* # Reorganize into samples[start:best.pos] + samples[best.pos:end]
|
|
* if best.pos < end and current.feature != best.feature:
|
|
* self.extract_nnz(best.feature, &end_negative, &start_positive, # <<<<<<<<<<<<<<
|
|
* &is_samples_sorted)
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_best.feature, (&__pyx_v_end_negative), (&__pyx_v_start_positive), (&__pyx_v_is_samples_sorted));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2545
|
|
* &is_samples_sorted)
|
|
*
|
|
* self._partition(best.threshold, end_negative, start_positive, # <<<<<<<<<<<<<<
|
|
* best.pos)
|
|
*
|
|
*/
|
|
__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter__partition(((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)__pyx_v_self), __pyx_v_best.threshold, __pyx_v_end_negative, __pyx_v_start_positive, __pyx_v_best.pos);
|
|
goto __pyx_L19;
|
|
}
|
|
__pyx_L19:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2551
|
|
* # element in features[:n_known_constants] must be preserved for sibling
|
|
* # and child nodes
|
|
* memcpy(features, constant_features, sizeof(SIZE_t) * n_known_constants) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Copy newly found constant features
|
|
*/
|
|
memcpy(__pyx_v_features, __pyx_v_constant_features, ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_known_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2554
|
|
*
|
|
* # Copy newly found constant features
|
|
* memcpy(constant_features + n_known_constants, # <<<<<<<<<<<<<<
|
|
* features + n_known_constants,
|
|
* sizeof(SIZE_t) * n_found_constants)
|
|
*/
|
|
memcpy((__pyx_v_constant_features + __pyx_v_n_known_constants), (__pyx_v_features + __pyx_v_n_known_constants), ((sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)) * __pyx_v_n_found_constants));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2559
|
|
*
|
|
* # Return values
|
|
* split[0] = best # <<<<<<<<<<<<<<
|
|
* n_constant_features[0] = n_total_constants
|
|
*
|
|
*/
|
|
(__pyx_v_split[0]) = __pyx_v_best;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2560
|
|
* # Return values
|
|
* split[0] = best
|
|
* n_constant_features[0] = n_total_constants # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
(__pyx_v_n_constant_features[0]) = __pyx_v_n_total_constants;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2358
|
|
* self.random_state), self.__getstate__())
|
|
*
|
|
* cdef void node_split(self, double impurity, SplitRecord* split, # <<<<<<<<<<<<<<
|
|
* SIZE_t* n_constant_features) nogil:
|
|
* """Find a random split on node samples[start:end], using sparse
|
|
*/
|
|
|
|
/* function exit code */
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2569
|
|
* """Interface for different tree building strategies. """
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_11TreeBuilder_1build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static PyObject *__pyx_f_7sklearn_4tree_5_tree_11TreeBuilder_build(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *__pyx_v_self, CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, CYTHON_UNUSED PyObject *__pyx_v_X, CYTHON_UNUSED PyArrayObject *__pyx_v_y, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build *__pyx_optional_args) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2570
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
* pass
|
|
*/
|
|
PyArrayObject *__pyx_v_sample_weight = ((PyArrayObject *)Py_None);
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("build", 0);
|
|
if (__pyx_optional_args) {
|
|
if (__pyx_optional_args->__pyx_n > 0) {
|
|
__pyx_v_sample_weight = __pyx_optional_args->sample_weight;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2569
|
|
* """Interface for different tree building strategies. """
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
/* Check if called by wrapper */
|
|
if (unlikely(__pyx_skip_dispatch)) ;
|
|
/* Check if overridden in Python */
|
|
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_build); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_11TreeBuilder_1build)) {
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = PyTuple_New(4); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_tree));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_v_tree));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_tree));
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_y));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 2, ((PyObject *)__pyx_v_y));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_y));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_sample_weight));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 3, ((PyObject *)__pyx_v_sample_weight));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_sample_weight));
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2571
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y).""" # <<<<<<<<<<<<<<
|
|
* pass
|
|
*
|
|
*/
|
|
|
|
/* 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_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.TreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2569
|
|
* """Interface for different tree building strategies. """
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_11TreeBuilder_1build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_11TreeBuilder_build[] = "Build a decision tree from the training set (X, y).";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_11TreeBuilder_1build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree = 0;
|
|
PyObject *__pyx_v_X = 0;
|
|
PyArrayObject *__pyx_v_y = 0;
|
|
PyArrayObject *__pyx_v_sample_weight = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("build (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_tree,&__pyx_n_s_X,&__pyx_n_s_y,&__pyx_n_s_sample_weight,0};
|
|
PyObject* values[4] = {0,0,0,0};
|
|
|
|
/* "sklearn/tree/_tree.pyx":2570
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
* pass
|
|
*/
|
|
values[3] = (PyObject *)((PyArrayObject *)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 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_tree)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_X)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_y)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s_sample_weight);
|
|
if (value) { values[3] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "build") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 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_tree = ((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)values[0]);
|
|
__pyx_v_X = values[1];
|
|
__pyx_v_y = ((PyArrayObject *)values[2]);
|
|
__pyx_v_sample_weight = ((PyArrayObject *)values[3]);
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.TreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_tree), __pyx_ptype_7sklearn_4tree_5_tree_Tree, 1, "tree", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_y), __pyx_ptype_5numpy_ndarray, 1, "y", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_sample_weight), __pyx_ptype_5numpy_ndarray, 1, "sample_weight", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2570; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_11TreeBuilder_build(((struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)__pyx_v_self), __pyx_v_tree, __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2569
|
|
* """Interface for different tree building strategies. """
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_11TreeBuilder_build(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build __pyx_t_2;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("build", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2.__pyx_n = 1;
|
|
__pyx_t_2.sample_weight = __pyx_v_sample_weight;
|
|
__pyx_t_1 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder *)__pyx_v_self->__pyx_vtab)->build(__pyx_v_self, __pyx_v_tree, __pyx_v_X, __pyx_v_y, 1, &__pyx_t_2); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2569; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.TreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2574
|
|
* pass
|
|
*
|
|
* cdef inline _check_input(self, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight):
|
|
* """Check input dtype, layout and format"""
|
|
*/
|
|
|
|
static PyObject *__pyx_f_7sklearn_4tree_5_tree_11TreeBuilder__check_input(CYTHON_UNUSED struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *__pyx_v_self, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
int __pyx_t_4;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_check_input", 0);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_y);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2577
|
|
* np.ndarray sample_weight):
|
|
* """Check input dtype, layout and format"""
|
|
* if issparse(X): # <<<<<<<<<<<<<<
|
|
* X = X.tocsc()
|
|
* X.sort_indices()
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_issparse); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2577; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2577; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2577; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2577; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2578
|
|
* """Check input dtype, layout and format"""
|
|
* if issparse(X):
|
|
* X = X.tocsc() # <<<<<<<<<<<<<<
|
|
* X.sort_indices()
|
|
*
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_tocsc); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2578; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2578; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_X, __pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2579
|
|
* if issparse(X):
|
|
* X = X.tocsc()
|
|
* X.sort_indices() # <<<<<<<<<<<<<<
|
|
*
|
|
* if X.data.dtype != DTYPE:
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_sort_indices); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2579; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2579; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2581
|
|
* X.sort_indices()
|
|
*
|
|
* if X.data.dtype != DTYPE: # <<<<<<<<<<<<<<
|
|
* X.data = np.ascontiguousarray(X.data, dtype=DTYPE)
|
|
*
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_data); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2581; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_dtype); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2581; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_GetModuleGlobalName(__pyx_n_s_DTYPE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2581; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_1 = PyObject_RichCompare(__pyx_t_2, __pyx_t_3, Py_NE); __Pyx_XGOTREF(__pyx_t_1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2581; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2581; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2582
|
|
*
|
|
* if X.data.dtype != DTYPE:
|
|
* X.data = np.ascontiguousarray(X.data, dtype=DTYPE) # <<<<<<<<<<<<<<
|
|
*
|
|
* if X.indices.dtype != np.int32 or X.indptr.dtype != np.int32:
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_ascontiguousarray); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_data); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_5 = __Pyx_GetModuleGlobalName(__pyx_n_s_DTYPE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_5) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_2, __pyx_t_1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (__Pyx_PyObject_SetAttrStr(__pyx_v_X, __pyx_n_s_data, __pyx_t_5) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2582; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
goto __pyx_L4;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2584
|
|
* X.data = np.ascontiguousarray(X.data, dtype=DTYPE)
|
|
*
|
|
* if X.indices.dtype != np.int32 or X.indptr.dtype != np.int32: # <<<<<<<<<<<<<<
|
|
* raise ValueError("No support for np.int64 index based "
|
|
* "sparse matrices")
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_indices); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_dtype); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_int32); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_t_1, __pyx_t_2, Py_NE); __Pyx_XGOTREF(__pyx_t_5); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (!__pyx_t_4) {
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_indptr); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_dtype); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_int32); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_t_2, __pyx_t_1, Py_NE); __Pyx_XGOTREF(__pyx_t_5); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2584; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_7 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_4;
|
|
}
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2585
|
|
*
|
|
* if X.indices.dtype != np.int32 or X.indptr.dtype != np.int32:
|
|
* raise ValueError("No support for np.int64 index based " # <<<<<<<<<<<<<<
|
|
* "sparse matrices")
|
|
*
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__2, NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2585; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_Raise(__pyx_t_5, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 2585; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
goto __pyx_L3;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2588
|
|
* "sparse matrices")
|
|
*
|
|
* elif X.dtype != DTYPE: # <<<<<<<<<<<<<<
|
|
* # since we have to copy we will make it fortran for efficiency
|
|
* X = np.asfortranarray(X, dtype=DTYPE)
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_dtype); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2588; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_DTYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2588; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_RichCompare(__pyx_t_5, __pyx_t_1, Py_NE); __Pyx_XGOTREF(__pyx_t_2); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2588; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_7 = __Pyx_PyObject_IsTrue(__pyx_t_2); if (unlikely(__pyx_t_7 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2588; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2590
|
|
* elif X.dtype != DTYPE:
|
|
* # since we have to copy we will make it fortran for efficiency
|
|
* X = np.asfortranarray(X, dtype=DTYPE) # <<<<<<<<<<<<<<
|
|
*
|
|
* if y.dtype != DOUBLE or not y.flags.contiguous:
|
|
*/
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_asfortranarray); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__pyx_t_5 = PyDict_New(); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = __Pyx_GetModuleGlobalName(__pyx_n_s_DTYPE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
if (PyDict_SetItem(__pyx_t_5, __pyx_n_s_dtype, __pyx_t_3) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, __pyx_t_5); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2590; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_X, __pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2592
|
|
* X = np.asfortranarray(X, dtype=DTYPE)
|
|
*
|
|
* if y.dtype != DOUBLE or not y.flags.contiguous: # <<<<<<<<<<<<<<
|
|
* y = np.ascontiguousarray(y, dtype=DOUBLE)
|
|
*
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_y), __pyx_n_s_dtype); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = __Pyx_GetModuleGlobalName(__pyx_n_s_DOUBLE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_2 = PyObject_RichCompare(__pyx_t_3, __pyx_t_5, Py_NE); __Pyx_XGOTREF(__pyx_t_2); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_7 = __Pyx_PyObject_IsTrue(__pyx_t_2); if (unlikely(__pyx_t_7 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (!__pyx_t_7) {
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_y), __pyx_n_s_flags); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_contiguous); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2592; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = (!__pyx_t_4);
|
|
__pyx_t_4 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_4 = __pyx_t_7;
|
|
}
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2593
|
|
*
|
|
* if y.dtype != DOUBLE or not y.flags.contiguous:
|
|
* y = np.ascontiguousarray(y, dtype=DOUBLE) # <<<<<<<<<<<<<<
|
|
*
|
|
* if (sample_weight is not None and
|
|
*/
|
|
__pyx_t_5 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_ascontiguousarray); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_y));
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, ((PyObject *)__pyx_v_y));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_y));
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_DOUBLE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_t_3, __pyx_n_s_dtype, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_t_5, __pyx_t_3); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2593; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF_SET(__pyx_v_y, ((PyArrayObject *)__pyx_t_1));
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L6;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2595
|
|
* y = np.ascontiguousarray(y, dtype=DOUBLE)
|
|
*
|
|
* if (sample_weight is not None and # <<<<<<<<<<<<<<
|
|
* (sample_weight.dtype != DOUBLE or
|
|
* not sample_weight.flags.contiguous)):
|
|
*/
|
|
__pyx_t_4 = (((PyObject *)__pyx_v_sample_weight) != Py_None);
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2596
|
|
*
|
|
* if (sample_weight is not None and
|
|
* (sample_weight.dtype != DOUBLE or # <<<<<<<<<<<<<<
|
|
* not sample_weight.flags.contiguous)):
|
|
* sample_weight = np.asarray(sample_weight, dtype=DOUBLE,
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_sample_weight), __pyx_n_s_dtype); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2596; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_GetModuleGlobalName(__pyx_n_s_DOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2596; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_t_1, __pyx_t_3, Py_NE); __Pyx_XGOTREF(__pyx_t_5); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2596; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_7 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_7 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2596; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2597
|
|
* if (sample_weight is not None and
|
|
* (sample_weight.dtype != DOUBLE or
|
|
* not sample_weight.flags.contiguous)): # <<<<<<<<<<<<<<
|
|
* sample_weight = np.asarray(sample_weight, dtype=DOUBLE,
|
|
* order="C")
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_sample_weight), __pyx_n_s_flags); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2597; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_contiguous); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2597; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2597; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_8 = (!__pyx_t_6);
|
|
__pyx_t_6 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_7;
|
|
}
|
|
__pyx_t_7 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_4;
|
|
}
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2598
|
|
* (sample_weight.dtype != DOUBLE or
|
|
* not sample_weight.flags.contiguous)):
|
|
* sample_weight = np.asarray(sample_weight, dtype=DOUBLE, # <<<<<<<<<<<<<<
|
|
* order="C")
|
|
*
|
|
*/
|
|
__pyx_t_3 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_asarray); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_sample_weight));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)__pyx_v_sample_weight));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_sample_weight));
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_DOUBLE); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_order, __pyx_n_s_C) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_5, __pyx_t_3, __pyx_t_1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2598; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF_SET(__pyx_v_sample_weight, ((PyArrayObject *)__pyx_t_2));
|
|
__pyx_t_2 = 0;
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2601
|
|
* order="C")
|
|
*
|
|
* return X, y, sample_weight # <<<<<<<<<<<<<<
|
|
*
|
|
* # Depth first builder ---------------------------------------------------------
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = PyTuple_New(3); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2601; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_y));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, ((PyObject *)__pyx_v_y));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_y));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_sample_weight));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 2, ((PyObject *)__pyx_v_sample_weight));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_sample_weight));
|
|
__pyx_r = __pyx_t_2;
|
|
__pyx_t_2 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2574
|
|
* pass
|
|
*
|
|
* cdef inline _check_input(self, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight):
|
|
* """Check input dtype, layout and format"""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.TreeBuilder._check_input", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_X);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_y);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_sample_weight);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2608
|
|
* """Build a decision tree in depth-first fashion."""
|
|
*
|
|
* def __cinit__(self, Splitter splitter, SIZE_t min_samples_split, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* SIZE_t max_depth):
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_splitter,&__pyx_n_s_min_samples_split,&__pyx_n_s_min_samples_leaf,&__pyx_n_s_min_weight_leaf,&__pyx_n_s_max_depth,0};
|
|
PyObject* values[5] = {0,0,0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_splitter)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_split)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (likely((values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_weight_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 4:
|
|
if (likely((values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_depth)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 5) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
}
|
|
__pyx_v_splitter = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)values[0]);
|
|
__pyx_v_min_samples_split = __Pyx_PyInt_As_Py_intptr_t(values[1]); if (unlikely((__pyx_v_min_samples_split == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_samples_leaf = __Pyx_PyInt_As_Py_intptr_t(values[2]); if (unlikely((__pyx_v_min_samples_leaf == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2609; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_weight_leaf = __pyx_PyFloat_AsDouble(values[3]); if (unlikely((__pyx_v_min_weight_leaf == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2609; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_max_depth = __Pyx_PyInt_As_Py_intptr_t(values[4]); if (unlikely((__pyx_v_max_depth == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2610; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 5, 5, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.DepthFirstTreeBuilder.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_splitter), __pyx_ptype_7sklearn_4tree_5_tree_Splitter, 1, "splitter", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2608; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *)__pyx_v_self), __pyx_v_splitter, __pyx_v_min_samples_split, __pyx_v_min_samples_leaf, __pyx_v_min_weight_leaf, __pyx_v_max_depth);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, double __pyx_v_min_weight_leaf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2611
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* SIZE_t max_depth):
|
|
* self.splitter = splitter # <<<<<<<<<<<<<<
|
|
* self.min_samples_split = min_samples_split
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
*/
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_splitter));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_splitter));
|
|
__Pyx_GOTREF(__pyx_v_self->__pyx_base.splitter);
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_self->__pyx_base.splitter));
|
|
__pyx_v_self->__pyx_base.splitter = __pyx_v_splitter;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2612
|
|
* SIZE_t max_depth):
|
|
* self.splitter = splitter
|
|
* self.min_samples_split = min_samples_split # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
*/
|
|
__pyx_v_self->__pyx_base.min_samples_split = __pyx_v_min_samples_split;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2613
|
|
* self.splitter = splitter
|
|
* self.min_samples_split = min_samples_split
|
|
* self.min_samples_leaf = min_samples_leaf # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.max_depth = max_depth
|
|
*/
|
|
__pyx_v_self->__pyx_base.min_samples_leaf = __pyx_v_min_samples_leaf;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2614
|
|
* self.min_samples_split = min_samples_split
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf # <<<<<<<<<<<<<<
|
|
* self.max_depth = max_depth
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.min_weight_leaf = __pyx_v_min_weight_leaf;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2615
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.max_depth = max_depth # <<<<<<<<<<<<<<
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
*/
|
|
__pyx_v_self->__pyx_base.max_depth = __pyx_v_max_depth;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2608
|
|
* """Build a decision tree in depth-first fashion."""
|
|
*
|
|
* def __cinit__(self, Splitter splitter, SIZE_t min_samples_split, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, double min_weight_leaf,
|
|
* SIZE_t max_depth):
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2617
|
|
* self.max_depth = max_depth
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_3build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static PyObject *__pyx_f_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_build(struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_build *__pyx_optional_args) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2618
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*
|
|
*/
|
|
PyArrayObject *__pyx_v_sample_weight = ((PyArrayObject *)Py_None);
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight_ptr;
|
|
int __pyx_v_init_capacity;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
double __pyx_v_min_weight_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_depth;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_parent;
|
|
int __pyx_v_is_left;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_node_samples;
|
|
CYTHON_UNUSED double __pyx_v_weighted_n_samples;
|
|
double __pyx_v_weighted_n_node_samples;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_split;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_node_id;
|
|
double __pyx_v_impurity;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_constant_features;
|
|
int __pyx_v_is_leaf;
|
|
int __pyx_v_first;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth_seen;
|
|
int __pyx_v_rc;
|
|
struct __pyx_obj_7sklearn_4tree_6_utils_Stack *__pyx_v_stack = 0;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_StackRecord __pyx_v_stack_record;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyObject *(*__pyx_t_6)(PyObject *);
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
double __pyx_t_10;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t __pyx_t_11;
|
|
int __pyx_t_12;
|
|
int __pyx_t_13;
|
|
int __pyx_t_14;
|
|
int __pyx_t_15;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c __pyx_t_16;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("build", 0);
|
|
if (__pyx_optional_args) {
|
|
if (__pyx_optional_args->__pyx_n > 0) {
|
|
__pyx_v_sample_weight = __pyx_optional_args->sample_weight;
|
|
}
|
|
}
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_y);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2617
|
|
* self.max_depth = max_depth
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
/* Check if called by wrapper */
|
|
if (unlikely(__pyx_skip_dispatch)) ;
|
|
/* Check if overridden in Python */
|
|
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_build); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_3build)) {
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = PyTuple_New(4); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_tree));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_v_tree));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_tree));
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_y));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 2, ((PyObject *)__pyx_v_y));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_y));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_sample_weight));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 3, ((PyObject *)__pyx_v_sample_weight));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_sample_weight));
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2622
|
|
*
|
|
* # check input
|
|
* X, y, sample_weight = self._check_input(X, y, sample_weight) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL
|
|
*/
|
|
__pyx_t_1 = __pyx_f_7sklearn_4tree_5_tree_11TreeBuilder__check_input(((struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)__pyx_v_self), __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if ((likely(PyTuple_CheckExact(__pyx_t_1))) || (PyList_CheckExact(__pyx_t_1))) {
|
|
PyObject* sequence = __pyx_t_1;
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
Py_ssize_t size = Py_SIZE(sequence);
|
|
#else
|
|
Py_ssize_t size = PySequence_Size(sequence);
|
|
#endif
|
|
if (unlikely(size != 3)) {
|
|
if (size > 3) __Pyx_RaiseTooManyValuesError(3);
|
|
else if (size >= 0) __Pyx_RaiseNeedMoreValuesError(size);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
if (likely(PyTuple_CheckExact(sequence))) {
|
|
__pyx_t_3 = PyTuple_GET_ITEM(sequence, 0);
|
|
__pyx_t_2 = PyTuple_GET_ITEM(sequence, 1);
|
|
__pyx_t_4 = PyTuple_GET_ITEM(sequence, 2);
|
|
} else {
|
|
__pyx_t_3 = PyList_GET_ITEM(sequence, 0);
|
|
__pyx_t_2 = PyList_GET_ITEM(sequence, 1);
|
|
__pyx_t_4 = PyList_GET_ITEM(sequence, 2);
|
|
}
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
#else
|
|
__pyx_t_3 = PySequence_ITEM(sequence, 0); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = PySequence_ITEM(sequence, 1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_4 = PySequence_ITEM(sequence, 2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
#endif
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
} else {
|
|
Py_ssize_t index = -1;
|
|
__pyx_t_5 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_6 = Py_TYPE(__pyx_t_5)->tp_iternext;
|
|
index = 0; __pyx_t_3 = __pyx_t_6(__pyx_t_5); if (unlikely(!__pyx_t_3)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
index = 1; __pyx_t_2 = __pyx_t_6(__pyx_t_5); if (unlikely(!__pyx_t_2)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
index = 2; __pyx_t_4 = __pyx_t_6(__pyx_t_5); if (unlikely(!__pyx_t_4)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (__Pyx_IternextUnpackEndCheck(__pyx_t_6(__pyx_t_5), 3) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_6 = NULL;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
goto __pyx_L4_unpacking_done;
|
|
__pyx_L3_unpacking_failed:;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = NULL;
|
|
if (__Pyx_IterFinish() == 0) __Pyx_RaiseNeedMoreValuesError(index);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_L4_unpacking_done:;
|
|
}
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2622; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF_SET(__pyx_v_X, __pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_y, ((PyArrayObject *)__pyx_t_2));
|
|
__pyx_t_2 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_sample_weight, ((PyArrayObject *)__pyx_t_4));
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2624
|
|
* X, y, sample_weight = self._check_input(X, y, sample_weight)
|
|
*
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL # <<<<<<<<<<<<<<
|
|
* if sample_weight is not None:
|
|
* sample_weight_ptr = <DOUBLE_t*> sample_weight.data
|
|
*/
|
|
__pyx_v_sample_weight_ptr = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2625
|
|
*
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL
|
|
* if sample_weight is not None: # <<<<<<<<<<<<<<
|
|
* sample_weight_ptr = <DOUBLE_t*> sample_weight.data
|
|
*
|
|
*/
|
|
__pyx_t_7 = (((PyObject *)__pyx_v_sample_weight) != Py_None);
|
|
__pyx_t_8 = (__pyx_t_7 != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2626
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL
|
|
* if sample_weight is not None:
|
|
* sample_weight_ptr = <DOUBLE_t*> sample_weight.data # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initial capacity
|
|
*/
|
|
__pyx_v_sample_weight_ptr = ((__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *)__pyx_v_sample_weight->data);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2631
|
|
* cdef int init_capacity
|
|
*
|
|
* if tree.max_depth <= 10: # <<<<<<<<<<<<<<
|
|
* init_capacity = (2 ** (tree.max_depth + 1)) - 1
|
|
* else:
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_tree->max_depth <= 10) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2632
|
|
*
|
|
* if tree.max_depth <= 10:
|
|
* init_capacity = (2 ** (tree.max_depth + 1)) - 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* init_capacity = 2047
|
|
*/
|
|
__pyx_v_init_capacity = (__Pyx_pow_long(2, (__pyx_v_tree->max_depth + 1)) - 1);
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2634
|
|
* init_capacity = (2 ** (tree.max_depth + 1)) - 1
|
|
* else:
|
|
* init_capacity = 2047 # <<<<<<<<<<<<<<
|
|
*
|
|
* tree._resize(init_capacity)
|
|
*/
|
|
__pyx_v_init_capacity = 2047;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2636
|
|
* init_capacity = 2047
|
|
*
|
|
* tree._resize(init_capacity) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Parameters
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_tree->__pyx_vtab)->_resize(__pyx_v_tree, __pyx_v_init_capacity); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2636; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2639
|
|
*
|
|
* # Parameters
|
|
* cdef Splitter splitter = self.splitter # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_depth = self.max_depth
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
*/
|
|
__pyx_t_1 = ((PyObject *)__pyx_v_self->__pyx_base.splitter);
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__pyx_v_splitter = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2640
|
|
* # Parameters
|
|
* cdef Splitter splitter = self.splitter
|
|
* cdef SIZE_t max_depth = self.max_depth # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_9 = __pyx_v_self->__pyx_base.max_depth;
|
|
__pyx_v_max_depth = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2641
|
|
* cdef Splitter splitter = self.splitter
|
|
* cdef SIZE_t max_depth = self.max_depth
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef SIZE_t min_samples_split = self.min_samples_split
|
|
*/
|
|
__pyx_t_9 = __pyx_v_self->__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2642
|
|
* cdef SIZE_t max_depth = self.max_depth
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_split = self.min_samples_split
|
|
*
|
|
*/
|
|
__pyx_t_10 = __pyx_v_self->__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2643
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef SIZE_t min_samples_split = self.min_samples_split # <<<<<<<<<<<<<<
|
|
*
|
|
* # Recursive partition (without actual recursion)
|
|
*/
|
|
__pyx_t_9 = __pyx_v_self->__pyx_base.min_samples_split;
|
|
__pyx_v_min_samples_split = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2646
|
|
*
|
|
* # Recursive partition (without actual recursion)
|
|
* splitter.init(X, y, sample_weight_ptr) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t start
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->init(__pyx_v_splitter, __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight_ptr);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2653
|
|
* cdef SIZE_t parent
|
|
* cdef bint is_left
|
|
* cdef SIZE_t n_node_samples = splitter.n_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_samples = splitter.weighted_n_samples
|
|
* cdef double weighted_n_node_samples
|
|
*/
|
|
__pyx_t_9 = __pyx_v_splitter->n_samples;
|
|
__pyx_v_n_node_samples = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2654
|
|
* cdef bint is_left
|
|
* cdef SIZE_t n_node_samples = splitter.n_samples
|
|
* cdef double weighted_n_samples = splitter.weighted_n_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_node_samples
|
|
* cdef SplitRecord split
|
|
*/
|
|
__pyx_t_10 = __pyx_v_splitter->weighted_n_samples;
|
|
__pyx_v_weighted_n_samples = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2660
|
|
*
|
|
* cdef double threshold
|
|
* cdef double impurity = INFINITY # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_constant_features
|
|
* cdef bint is_leaf
|
|
*/
|
|
__pyx_v_impurity = __pyx_v_7sklearn_4tree_5_tree_INFINITY;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2663
|
|
* cdef SIZE_t n_constant_features
|
|
* cdef bint is_leaf
|
|
* cdef bint first = 1 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_depth_seen = -1
|
|
* cdef int rc = 0
|
|
*/
|
|
__pyx_v_first = 1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2664
|
|
* cdef bint is_leaf
|
|
* cdef bint first = 1
|
|
* cdef SIZE_t max_depth_seen = -1 # <<<<<<<<<<<<<<
|
|
* cdef int rc = 0
|
|
*
|
|
*/
|
|
__pyx_v_max_depth_seen = -1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2665
|
|
* cdef bint first = 1
|
|
* cdef SIZE_t max_depth_seen = -1
|
|
* cdef int rc = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef Stack stack = Stack(INITIAL_STACK_SIZE)
|
|
*/
|
|
__pyx_v_rc = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2667
|
|
* cdef int rc = 0
|
|
*
|
|
* cdef Stack stack = Stack(INITIAL_STACK_SIZE) # <<<<<<<<<<<<<<
|
|
* cdef StackRecord stack_record
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_7sklearn_4tree_5_tree_INITIAL_STACK_SIZE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2667; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2667; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_6_utils_Stack)), __pyx_t_4, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2667; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_v_stack = ((struct __pyx_obj_7sklearn_4tree_6_utils_Stack *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2671
|
|
*
|
|
* # push root node onto stack
|
|
* rc = stack.push(0, n_node_samples, 0, _TREE_UNDEFINED, 0, INFINITY, 0) # <<<<<<<<<<<<<<
|
|
* if rc == -1:
|
|
* # got return code -1 - out-of-memory
|
|
*/
|
|
__pyx_v_rc = ((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *)__pyx_v_stack->__pyx_vtab)->push(__pyx_v_stack, 0, __pyx_v_n_node_samples, 0, __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED, 0, __pyx_v_7sklearn_4tree_5_tree_INFINITY, 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2672
|
|
* # push root node onto stack
|
|
* rc = stack.push(0, n_node_samples, 0, _TREE_UNDEFINED, 0, INFINITY, 0)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* # got return code -1 - out-of-memory
|
|
* raise MemoryError()
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2674
|
|
* if rc == -1:
|
|
* # got return code -1 - out-of-memory
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
* with nogil:
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2674; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2676
|
|
* raise MemoryError()
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* while not stack.is_empty():
|
|
* stack.pop(&stack_record)
|
|
*/
|
|
{
|
|
#ifdef WITH_THREAD
|
|
PyThreadState *_save;
|
|
Py_UNBLOCK_THREADS
|
|
#endif
|
|
/*try:*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2677
|
|
*
|
|
* with nogil:
|
|
* while not stack.is_empty(): # <<<<<<<<<<<<<<
|
|
* stack.pop(&stack_record)
|
|
*
|
|
*/
|
|
while (1) {
|
|
__pyx_t_8 = ((!(((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *)__pyx_v_stack->__pyx_vtab)->is_empty(__pyx_v_stack) != 0)) != 0);
|
|
if (!__pyx_t_8) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2678
|
|
* with nogil:
|
|
* while not stack.is_empty():
|
|
* stack.pop(&stack_record) # <<<<<<<<<<<<<<
|
|
*
|
|
* start = stack_record.start
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *)__pyx_v_stack->__pyx_vtab)->pop(__pyx_v_stack, (&__pyx_v_stack_record));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2680
|
|
* stack.pop(&stack_record)
|
|
*
|
|
* start = stack_record.start # <<<<<<<<<<<<<<
|
|
* end = stack_record.end
|
|
* depth = stack_record.depth
|
|
*/
|
|
__pyx_t_11 = __pyx_v_stack_record.start;
|
|
__pyx_v_start = __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2681
|
|
*
|
|
* start = stack_record.start
|
|
* end = stack_record.end # <<<<<<<<<<<<<<
|
|
* depth = stack_record.depth
|
|
* parent = stack_record.parent
|
|
*/
|
|
__pyx_t_11 = __pyx_v_stack_record.end;
|
|
__pyx_v_end = __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2682
|
|
* start = stack_record.start
|
|
* end = stack_record.end
|
|
* depth = stack_record.depth # <<<<<<<<<<<<<<
|
|
* parent = stack_record.parent
|
|
* is_left = stack_record.is_left
|
|
*/
|
|
__pyx_t_11 = __pyx_v_stack_record.depth;
|
|
__pyx_v_depth = __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2683
|
|
* end = stack_record.end
|
|
* depth = stack_record.depth
|
|
* parent = stack_record.parent # <<<<<<<<<<<<<<
|
|
* is_left = stack_record.is_left
|
|
* impurity = stack_record.impurity
|
|
*/
|
|
__pyx_t_11 = __pyx_v_stack_record.parent;
|
|
__pyx_v_parent = __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2684
|
|
* depth = stack_record.depth
|
|
* parent = stack_record.parent
|
|
* is_left = stack_record.is_left # <<<<<<<<<<<<<<
|
|
* impurity = stack_record.impurity
|
|
* n_constant_features = stack_record.n_constant_features
|
|
*/
|
|
__pyx_t_8 = __pyx_v_stack_record.is_left;
|
|
__pyx_v_is_left = __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2685
|
|
* parent = stack_record.parent
|
|
* is_left = stack_record.is_left
|
|
* impurity = stack_record.impurity # <<<<<<<<<<<<<<
|
|
* n_constant_features = stack_record.n_constant_features
|
|
*
|
|
*/
|
|
__pyx_t_10 = __pyx_v_stack_record.impurity;
|
|
__pyx_v_impurity = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2686
|
|
* is_left = stack_record.is_left
|
|
* impurity = stack_record.impurity
|
|
* n_constant_features = stack_record.n_constant_features # <<<<<<<<<<<<<<
|
|
*
|
|
* n_node_samples = end - start
|
|
*/
|
|
__pyx_t_11 = __pyx_v_stack_record.n_constant_features;
|
|
__pyx_v_n_constant_features = __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2688
|
|
* n_constant_features = stack_record.n_constant_features
|
|
*
|
|
* n_node_samples = end - start # <<<<<<<<<<<<<<
|
|
* splitter.node_reset(start, end, &weighted_n_node_samples)
|
|
*
|
|
*/
|
|
__pyx_v_n_node_samples = (__pyx_v_end - __pyx_v_start);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2689
|
|
*
|
|
* n_node_samples = end - start
|
|
* splitter.node_reset(start, end, &weighted_n_node_samples) # <<<<<<<<<<<<<<
|
|
*
|
|
* is_leaf = ((depth >= max_depth) or
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_reset(__pyx_v_splitter, __pyx_v_start, __pyx_v_end, (&__pyx_v_weighted_n_node_samples));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2691
|
|
* splitter.node_reset(start, end, &weighted_n_node_samples)
|
|
*
|
|
* is_leaf = ((depth >= max_depth) or # <<<<<<<<<<<<<<
|
|
* (n_node_samples < min_samples_split) or
|
|
* (n_node_samples < 2 * min_samples_leaf) or
|
|
*/
|
|
__pyx_t_8 = (__pyx_v_depth >= __pyx_v_max_depth);
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2692
|
|
*
|
|
* is_leaf = ((depth >= max_depth) or
|
|
* (n_node_samples < min_samples_split) or # <<<<<<<<<<<<<<
|
|
* (n_node_samples < 2 * min_samples_leaf) or
|
|
* (weighted_n_node_samples < min_weight_leaf))
|
|
*/
|
|
__pyx_t_7 = (__pyx_v_n_node_samples < __pyx_v_min_samples_split);
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2693
|
|
* is_leaf = ((depth >= max_depth) or
|
|
* (n_node_samples < min_samples_split) or
|
|
* (n_node_samples < 2 * min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* (weighted_n_node_samples < min_weight_leaf))
|
|
*
|
|
*/
|
|
__pyx_t_12 = (__pyx_v_n_node_samples < (2 * __pyx_v_min_samples_leaf));
|
|
if (!__pyx_t_12) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2694
|
|
* (n_node_samples < min_samples_split) or
|
|
* (n_node_samples < 2 * min_samples_leaf) or
|
|
* (weighted_n_node_samples < min_weight_leaf)) # <<<<<<<<<<<<<<
|
|
*
|
|
* if first:
|
|
*/
|
|
__pyx_t_13 = (__pyx_v_weighted_n_node_samples < __pyx_v_min_weight_leaf);
|
|
__pyx_t_14 = __pyx_t_13;
|
|
} else {
|
|
__pyx_t_14 = __pyx_t_12;
|
|
}
|
|
__pyx_t_12 = __pyx_t_14;
|
|
} else {
|
|
__pyx_t_12 = __pyx_t_7;
|
|
}
|
|
__pyx_t_7 = __pyx_t_12;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_8;
|
|
}
|
|
__pyx_v_is_leaf = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2696
|
|
* (weighted_n_node_samples < min_weight_leaf))
|
|
*
|
|
* if first: # <<<<<<<<<<<<<<
|
|
* impurity = splitter.node_impurity()
|
|
* first = 0
|
|
*/
|
|
__pyx_t_7 = (__pyx_v_first != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2697
|
|
*
|
|
* if first:
|
|
* impurity = splitter.node_impurity() # <<<<<<<<<<<<<<
|
|
* first = 0
|
|
*
|
|
*/
|
|
__pyx_v_impurity = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_impurity(__pyx_v_splitter);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2698
|
|
* if first:
|
|
* impurity = splitter.node_impurity()
|
|
* first = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* is_leaf = is_leaf or (impurity <= MIN_IMPURITY_SPLIT)
|
|
*/
|
|
__pyx_v_first = 0;
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2700
|
|
* first = 0
|
|
*
|
|
* is_leaf = is_leaf or (impurity <= MIN_IMPURITY_SPLIT) # <<<<<<<<<<<<<<
|
|
*
|
|
* if not is_leaf:
|
|
*/
|
|
if (!__pyx_v_is_leaf) {
|
|
__pyx_t_7 = (__pyx_v_impurity <= __pyx_v_7sklearn_4tree_5_tree_MIN_IMPURITY_SPLIT);
|
|
__pyx_t_8 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_8 = __pyx_v_is_leaf;
|
|
}
|
|
__pyx_v_is_leaf = __pyx_t_8;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2702
|
|
* is_leaf = is_leaf or (impurity <= MIN_IMPURITY_SPLIT)
|
|
*
|
|
* if not is_leaf: # <<<<<<<<<<<<<<
|
|
* splitter.node_split(impurity, &split, &n_constant_features)
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*/
|
|
__pyx_t_8 = ((!(__pyx_v_is_leaf != 0)) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2703
|
|
*
|
|
* if not is_leaf:
|
|
* splitter.node_split(impurity, &split, &n_constant_features) # <<<<<<<<<<<<<<
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_split(__pyx_v_splitter, __pyx_v_impurity, (&__pyx_v_split), (&__pyx_v_n_constant_features));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2704
|
|
* if not is_leaf:
|
|
* splitter.node_split(impurity, &split, &n_constant_features)
|
|
* is_leaf = is_leaf or (split.pos >= end) # <<<<<<<<<<<<<<
|
|
*
|
|
* node_id = tree._add_node(parent, is_left, is_leaf, split.feature,
|
|
*/
|
|
if (!__pyx_v_is_leaf) {
|
|
__pyx_t_8 = (__pyx_v_split.pos >= __pyx_v_end);
|
|
__pyx_t_7 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_7 = __pyx_v_is_leaf;
|
|
}
|
|
__pyx_v_is_leaf = __pyx_t_7;
|
|
goto __pyx_L14;
|
|
}
|
|
__pyx_L14:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2706
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*
|
|
* node_id = tree._add_node(parent, is_left, is_leaf, split.feature, # <<<<<<<<<<<<<<
|
|
* split.threshold, impurity, n_node_samples,
|
|
* weighted_n_node_samples)
|
|
*/
|
|
__pyx_v_node_id = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_tree->__pyx_vtab)->_add_node(__pyx_v_tree, __pyx_v_parent, __pyx_v_is_left, __pyx_v_is_leaf, __pyx_v_split.feature, __pyx_v_split.threshold, __pyx_v_impurity, __pyx_v_n_node_samples, __pyx_v_weighted_n_node_samples);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2710
|
|
* weighted_n_node_samples)
|
|
*
|
|
* if is_leaf: # <<<<<<<<<<<<<<
|
|
* # Don't store value for internal nodes
|
|
* splitter.node_value(tree.value +
|
|
*/
|
|
__pyx_t_7 = (__pyx_v_is_leaf != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2712
|
|
* if is_leaf:
|
|
* # Don't store value for internal nodes
|
|
* splitter.node_value(tree.value + # <<<<<<<<<<<<<<
|
|
* node_id * tree.value_stride)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_value(__pyx_v_splitter, (__pyx_v_tree->value + (__pyx_v_node_id * __pyx_v_tree->value_stride)));
|
|
goto __pyx_L15;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2717
|
|
* else:
|
|
* # Push right child on stack
|
|
* rc = stack.push(split.pos, end, depth + 1, node_id, 0, # <<<<<<<<<<<<<<
|
|
* split.impurity_right, n_constant_features)
|
|
* if rc == -1:
|
|
*/
|
|
__pyx_v_rc = ((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *)__pyx_v_stack->__pyx_vtab)->push(__pyx_v_stack, __pyx_v_split.pos, __pyx_v_end, (__pyx_v_depth + 1), __pyx_v_node_id, 0, __pyx_v_split.impurity_right, __pyx_v_n_constant_features);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2719
|
|
* rc = stack.push(split.pos, end, depth + 1, node_id, 0,
|
|
* split.impurity_right, n_constant_features)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2720
|
|
* split.impurity_right, n_constant_features)
|
|
* if rc == -1:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* # Push left child on stack
|
|
*/
|
|
goto __pyx_L12_break;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2723
|
|
*
|
|
* # Push left child on stack
|
|
* rc = stack.push(start, split.pos, depth + 1, node_id, 1, # <<<<<<<<<<<<<<
|
|
* split.impurity_left, n_constant_features)
|
|
* if rc == -1:
|
|
*/
|
|
__pyx_v_rc = ((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack *)__pyx_v_stack->__pyx_vtab)->push(__pyx_v_stack, __pyx_v_start, __pyx_v_split.pos, (__pyx_v_depth + 1), __pyx_v_node_id, 1, __pyx_v_split.impurity_left, __pyx_v_n_constant_features);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2725
|
|
* rc = stack.push(start, split.pos, depth + 1, node_id, 1,
|
|
* split.impurity_left, n_constant_features)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2726
|
|
* split.impurity_left, n_constant_features)
|
|
* if rc == -1:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if depth > max_depth_seen:
|
|
*/
|
|
goto __pyx_L12_break;
|
|
}
|
|
}
|
|
__pyx_L15:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2728
|
|
* break
|
|
*
|
|
* if depth > max_depth_seen: # <<<<<<<<<<<<<<
|
|
* max_depth_seen = depth
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_depth > __pyx_v_max_depth_seen) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2729
|
|
*
|
|
* if depth > max_depth_seen:
|
|
* max_depth_seen = depth # <<<<<<<<<<<<<<
|
|
*
|
|
* if rc >= 0:
|
|
*/
|
|
__pyx_v_max_depth_seen = __pyx_v_depth;
|
|
goto __pyx_L18;
|
|
}
|
|
__pyx_L18:;
|
|
}
|
|
__pyx_L12_break:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2731
|
|
* max_depth_seen = depth
|
|
*
|
|
* if rc >= 0: # <<<<<<<<<<<<<<
|
|
* rc = tree._resize_c(tree.node_count)
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc >= 0) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2732
|
|
*
|
|
* if rc >= 0:
|
|
* rc = tree._resize_c(tree.node_count) # <<<<<<<<<<<<<<
|
|
*
|
|
* if rc >= 0:
|
|
*/
|
|
__pyx_t_16.__pyx_n = 1;
|
|
__pyx_t_16.capacity = __pyx_v_tree->node_count;
|
|
__pyx_t_15 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_tree->__pyx_vtab)->_resize_c(__pyx_v_tree, &__pyx_t_16);
|
|
__pyx_v_rc = __pyx_t_15;
|
|
goto __pyx_L19;
|
|
}
|
|
__pyx_L19:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2734
|
|
* rc = tree._resize_c(tree.node_count)
|
|
*
|
|
* if rc >= 0: # <<<<<<<<<<<<<<
|
|
* tree.max_depth = max_depth_seen
|
|
* if rc == -1:
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc >= 0) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2735
|
|
*
|
|
* if rc >= 0:
|
|
* tree.max_depth = max_depth_seen # <<<<<<<<<<<<<<
|
|
* if rc == -1:
|
|
* raise MemoryError()
|
|
*/
|
|
__pyx_v_tree->max_depth = __pyx_v_max_depth_seen;
|
|
goto __pyx_L20;
|
|
}
|
|
__pyx_L20:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2676
|
|
* raise MemoryError()
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* while not stack.is_empty():
|
|
* stack.pop(&stack_record)
|
|
*/
|
|
/*finally:*/ {
|
|
/*normal exit:*/{
|
|
#ifdef WITH_THREAD
|
|
Py_BLOCK_THREADS
|
|
#endif
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2736
|
|
* if rc >= 0:
|
|
* tree.max_depth = max_depth_seen
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* raise MemoryError()
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2737
|
|
* tree.max_depth = max_depth_seen
|
|
* if rc == -1:
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2737; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2617
|
|
* self.max_depth = max_depth
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
/* 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_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.DepthFirstTreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_splitter);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_stack);
|
|
__Pyx_XDECREF(__pyx_v_X);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_y);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_sample_weight);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_3build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_2build[] = "Build a decision tree from the training set (X, y).";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_3build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree = 0;
|
|
PyObject *__pyx_v_X = 0;
|
|
PyArrayObject *__pyx_v_y = 0;
|
|
PyArrayObject *__pyx_v_sample_weight = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("build (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_tree,&__pyx_n_s_X,&__pyx_n_s_y,&__pyx_n_s_sample_weight,0};
|
|
PyObject* values[4] = {0,0,0,0};
|
|
|
|
/* "sklearn/tree/_tree.pyx":2618
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*
|
|
*/
|
|
values[3] = (PyObject *)((PyArrayObject *)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 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_tree)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_X)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_y)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s_sample_weight);
|
|
if (value) { values[3] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "build") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 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_tree = ((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)values[0]);
|
|
__pyx_v_X = values[1];
|
|
__pyx_v_y = ((PyArrayObject *)values[2]);
|
|
__pyx_v_sample_weight = ((PyArrayObject *)values[3]);
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.DepthFirstTreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_tree), __pyx_ptype_7sklearn_4tree_5_tree_Tree, 1, "tree", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_y), __pyx_ptype_5numpy_ndarray, 1, "y", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_sample_weight), __pyx_ptype_5numpy_ndarray, 1, "sample_weight", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2618; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_2build(((struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *)__pyx_v_self), __pyx_v_tree, __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2617
|
|
* self.max_depth = max_depth
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_2build(struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build __pyx_t_2;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("build", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2.__pyx_n = 1;
|
|
__pyx_t_2.sample_weight = __pyx_v_sample_weight;
|
|
__pyx_t_1 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.build(((struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)__pyx_v_self), __pyx_v_tree, __pyx_v_X, __pyx_v_y, 1, &__pyx_t_2); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2617; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.DepthFirstTreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2742
|
|
* # Best first builder ----------------------------------------------------------
|
|
*
|
|
* cdef inline int _add_to_frontier(PriorityHeapRecord* rec, # <<<<<<<<<<<<<<
|
|
* PriorityHeap frontier) nogil:
|
|
* """Adds record ``rec`` to the priority queue ``frontier``; returns -1
|
|
*/
|
|
|
|
static CYTHON_INLINE int __pyx_f_7sklearn_4tree_5_tree__add_to_frontier(struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *__pyx_v_rec, struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *__pyx_v_frontier) {
|
|
int __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2746
|
|
* """Adds record ``rec`` to the priority queue ``frontier``; returns -1
|
|
* on memory-error. """
|
|
* return frontier.push(rec.node_id, rec.start, rec.end, rec.pos, rec.depth, # <<<<<<<<<<<<<<
|
|
* rec.is_leaf, rec.improvement, rec.impurity,
|
|
* rec.impurity_left, rec.impurity_right)
|
|
*/
|
|
__pyx_r = ((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap *)__pyx_v_frontier->__pyx_vtab)->push(__pyx_v_frontier, __pyx_v_rec->node_id, __pyx_v_rec->start, __pyx_v_rec->end, __pyx_v_rec->pos, __pyx_v_rec->depth, __pyx_v_rec->is_leaf, __pyx_v_rec->improvement, __pyx_v_rec->impurity, __pyx_v_rec->impurity_left, __pyx_v_rec->impurity_right);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2742
|
|
* # Best first builder ----------------------------------------------------------
|
|
*
|
|
* cdef inline int _add_to_frontier(PriorityHeapRecord* rec, # <<<<<<<<<<<<<<
|
|
* PriorityHeap frontier) nogil:
|
|
* """Adds record ``rec`` to the priority queue ``frontier``; returns -1
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2761
|
|
* cdef SIZE_t max_leaf_nodes
|
|
*
|
|
* def __cinit__(self, Splitter splitter, SIZE_t min_samples_split, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, min_weight_leaf,
|
|
* SIZE_t max_depth, SIZE_t max_leaf_nodes):
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
PyObject *__pyx_v_min_weight_leaf = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_leaf_nodes;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_splitter,&__pyx_n_s_min_samples_split,&__pyx_n_s_min_samples_leaf,&__pyx_n_s_min_weight_leaf,&__pyx_n_s_max_depth,&__pyx_n_s_max_leaf_nodes,0};
|
|
PyObject* values[6] = {0,0,0,0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 6: values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
|
|
case 5: values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_splitter)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_split)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 6, 6, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_samples_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 6, 6, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (likely((values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_min_weight_leaf)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 6, 6, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 4:
|
|
if (likely((values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_depth)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 6, 6, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 5:
|
|
if (likely((values[5] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_max_leaf_nodes)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 6, 6, 5); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 6) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
values[4] = PyTuple_GET_ITEM(__pyx_args, 4);
|
|
values[5] = PyTuple_GET_ITEM(__pyx_args, 5);
|
|
}
|
|
__pyx_v_splitter = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)values[0]);
|
|
__pyx_v_min_samples_split = __Pyx_PyInt_As_Py_intptr_t(values[1]); if (unlikely((__pyx_v_min_samples_split == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_samples_leaf = __Pyx_PyInt_As_Py_intptr_t(values[2]); if (unlikely((__pyx_v_min_samples_leaf == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2762; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_min_weight_leaf = values[3];
|
|
__pyx_v_max_depth = __Pyx_PyInt_As_Py_intptr_t(values[4]); if (unlikely((__pyx_v_max_depth == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2763; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_max_leaf_nodes = __Pyx_PyInt_As_Py_intptr_t(values[5]); if (unlikely((__pyx_v_max_leaf_nodes == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2763; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 6, 6, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestFirstTreeBuilder.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_splitter), __pyx_ptype_7sklearn_4tree_5_tree_Splitter, 1, "splitter", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2761; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_20BestFirstTreeBuilder___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *)__pyx_v_self), __pyx_v_splitter, __pyx_v_min_samples_split, __pyx_v_min_samples_leaf, __pyx_v_min_weight_leaf, __pyx_v_max_depth, __pyx_v_max_leaf_nodes);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_20BestFirstTreeBuilder___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf, PyObject *__pyx_v_min_weight_leaf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_leaf_nodes) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
double __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2764
|
|
* SIZE_t min_samples_leaf, min_weight_leaf,
|
|
* SIZE_t max_depth, SIZE_t max_leaf_nodes):
|
|
* self.splitter = splitter # <<<<<<<<<<<<<<
|
|
* self.min_samples_split = min_samples_split
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
*/
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_splitter));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_splitter));
|
|
__Pyx_GOTREF(__pyx_v_self->__pyx_base.splitter);
|
|
__Pyx_DECREF(((PyObject *)__pyx_v_self->__pyx_base.splitter));
|
|
__pyx_v_self->__pyx_base.splitter = __pyx_v_splitter;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2765
|
|
* SIZE_t max_depth, SIZE_t max_leaf_nodes):
|
|
* self.splitter = splitter
|
|
* self.min_samples_split = min_samples_split # <<<<<<<<<<<<<<
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
*/
|
|
__pyx_v_self->__pyx_base.min_samples_split = __pyx_v_min_samples_split;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2766
|
|
* self.splitter = splitter
|
|
* self.min_samples_split = min_samples_split
|
|
* self.min_samples_leaf = min_samples_leaf # <<<<<<<<<<<<<<
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.max_depth = max_depth
|
|
*/
|
|
__pyx_v_self->__pyx_base.min_samples_leaf = __pyx_v_min_samples_leaf;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2767
|
|
* self.min_samples_split = min_samples_split
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf # <<<<<<<<<<<<<<
|
|
* self.max_depth = max_depth
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*/
|
|
__pyx_t_1 = __pyx_PyFloat_AsDouble(__pyx_v_min_weight_leaf); if (unlikely((__pyx_t_1 == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2767; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->__pyx_base.min_weight_leaf = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2768
|
|
* self.min_samples_leaf = min_samples_leaf
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.max_depth = max_depth # <<<<<<<<<<<<<<
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*
|
|
*/
|
|
__pyx_v_self->__pyx_base.max_depth = __pyx_v_max_depth;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2769
|
|
* self.min_weight_leaf = min_weight_leaf
|
|
* self.max_depth = max_depth
|
|
* self.max_leaf_nodes = max_leaf_nodes # <<<<<<<<<<<<<<
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
*/
|
|
__pyx_v_self->max_leaf_nodes = __pyx_v_max_leaf_nodes;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2761
|
|
* cdef SIZE_t max_leaf_nodes
|
|
*
|
|
* def __cinit__(self, Splitter splitter, SIZE_t min_samples_split, # <<<<<<<<<<<<<<
|
|
* SIZE_t min_samples_leaf, min_weight_leaf,
|
|
* SIZE_t max_depth, SIZE_t max_leaf_nodes):
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestFirstTreeBuilder.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2771
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_3build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static PyObject *__pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_build(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_build *__pyx_optional_args) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2772
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*
|
|
*/
|
|
PyArrayObject *__pyx_v_sample_weight = ((PyArrayObject *)Py_None);
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_sample_weight_ptr;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_leaf_nodes;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_leaf;
|
|
CYTHON_UNUSED double __pyx_v_min_weight_leaf;
|
|
CYTHON_UNUSED __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_min_samples_split;
|
|
struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *__pyx_v_frontier = 0;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord __pyx_v_record;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord __pyx_v_split_node_left;
|
|
struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord __pyx_v_split_node_right;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_split_nodes;
|
|
int __pyx_v_is_leaf;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_max_depth_seen;
|
|
int __pyx_v_rc;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_node;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_init_capacity;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyObject *(*__pyx_t_6)(PyObject *);
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
double __pyx_t_10;
|
|
__pyx_t_7sklearn_4tree_6_utils_SIZE_t __pyx_t_11;
|
|
int __pyx_t_12;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c __pyx_t_13;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("build", 0);
|
|
if (__pyx_optional_args) {
|
|
if (__pyx_optional_args->__pyx_n > 0) {
|
|
__pyx_v_sample_weight = __pyx_optional_args->sample_weight;
|
|
}
|
|
}
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_y);
|
|
__Pyx_INCREF((PyObject *)__pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2771
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
/* Check if called by wrapper */
|
|
if (unlikely(__pyx_skip_dispatch)) ;
|
|
/* Check if overridden in Python */
|
|
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_build); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_3build)) {
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = PyTuple_New(4); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_tree));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_v_tree));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_tree));
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_y));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 2, ((PyObject *)__pyx_v_y));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_y));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_sample_weight));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 3, ((PyObject *)__pyx_v_sample_weight));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_sample_weight));
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2776
|
|
*
|
|
* # check input
|
|
* X, y, sample_weight = self._check_input(X, y, sample_weight) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL
|
|
*/
|
|
__pyx_t_1 = __pyx_f_7sklearn_4tree_5_tree_11TreeBuilder__check_input(((struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)__pyx_v_self), __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if ((likely(PyTuple_CheckExact(__pyx_t_1))) || (PyList_CheckExact(__pyx_t_1))) {
|
|
PyObject* sequence = __pyx_t_1;
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
Py_ssize_t size = Py_SIZE(sequence);
|
|
#else
|
|
Py_ssize_t size = PySequence_Size(sequence);
|
|
#endif
|
|
if (unlikely(size != 3)) {
|
|
if (size > 3) __Pyx_RaiseTooManyValuesError(3);
|
|
else if (size >= 0) __Pyx_RaiseNeedMoreValuesError(size);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
if (likely(PyTuple_CheckExact(sequence))) {
|
|
__pyx_t_3 = PyTuple_GET_ITEM(sequence, 0);
|
|
__pyx_t_2 = PyTuple_GET_ITEM(sequence, 1);
|
|
__pyx_t_4 = PyTuple_GET_ITEM(sequence, 2);
|
|
} else {
|
|
__pyx_t_3 = PyList_GET_ITEM(sequence, 0);
|
|
__pyx_t_2 = PyList_GET_ITEM(sequence, 1);
|
|
__pyx_t_4 = PyList_GET_ITEM(sequence, 2);
|
|
}
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
#else
|
|
__pyx_t_3 = PySequence_ITEM(sequence, 0); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = PySequence_ITEM(sequence, 1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_4 = PySequence_ITEM(sequence, 2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
#endif
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
} else {
|
|
Py_ssize_t index = -1;
|
|
__pyx_t_5 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_6 = Py_TYPE(__pyx_t_5)->tp_iternext;
|
|
index = 0; __pyx_t_3 = __pyx_t_6(__pyx_t_5); if (unlikely(!__pyx_t_3)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
index = 1; __pyx_t_2 = __pyx_t_6(__pyx_t_5); if (unlikely(!__pyx_t_2)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
index = 2; __pyx_t_4 = __pyx_t_6(__pyx_t_5); if (unlikely(!__pyx_t_4)) goto __pyx_L3_unpacking_failed;
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (__Pyx_IternextUnpackEndCheck(__pyx_t_6(__pyx_t_5), 3) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_6 = NULL;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
goto __pyx_L4_unpacking_done;
|
|
__pyx_L3_unpacking_failed:;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_6 = NULL;
|
|
if (__Pyx_IterFinish() == 0) __Pyx_RaiseNeedMoreValuesError(index);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_L4_unpacking_done:;
|
|
}
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2776; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF_SET(__pyx_v_X, __pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_y, ((PyArrayObject *)__pyx_t_2));
|
|
__pyx_t_2 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_sample_weight, ((PyArrayObject *)__pyx_t_4));
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2778
|
|
* X, y, sample_weight = self._check_input(X, y, sample_weight)
|
|
*
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL # <<<<<<<<<<<<<<
|
|
* if sample_weight is not None:
|
|
* sample_weight_ptr = <DOUBLE_t*> sample_weight.data
|
|
*/
|
|
__pyx_v_sample_weight_ptr = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2779
|
|
*
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL
|
|
* if sample_weight is not None: # <<<<<<<<<<<<<<
|
|
* sample_weight_ptr = <DOUBLE_t*> sample_weight.data
|
|
*
|
|
*/
|
|
__pyx_t_7 = (((PyObject *)__pyx_v_sample_weight) != Py_None);
|
|
__pyx_t_8 = (__pyx_t_7 != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2780
|
|
* cdef DOUBLE_t* sample_weight_ptr = NULL
|
|
* if sample_weight is not None:
|
|
* sample_weight_ptr = <DOUBLE_t*> sample_weight.data # <<<<<<<<<<<<<<
|
|
*
|
|
* # Parameters
|
|
*/
|
|
__pyx_v_sample_weight_ptr = ((__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *)__pyx_v_sample_weight->data);
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2783
|
|
*
|
|
* # Parameters
|
|
* cdef Splitter splitter = self.splitter # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_leaf_nodes = self.max_leaf_nodes
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
*/
|
|
__pyx_t_1 = ((PyObject *)__pyx_v_self->__pyx_base.splitter);
|
|
__Pyx_INCREF(__pyx_t_1);
|
|
__pyx_v_splitter = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2784
|
|
* # Parameters
|
|
* cdef Splitter splitter = self.splitter
|
|
* cdef SIZE_t max_leaf_nodes = self.max_leaf_nodes # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
*/
|
|
__pyx_t_9 = __pyx_v_self->max_leaf_nodes;
|
|
__pyx_v_max_leaf_nodes = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2785
|
|
* cdef Splitter splitter = self.splitter
|
|
* cdef SIZE_t max_leaf_nodes = self.max_leaf_nodes
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf # <<<<<<<<<<<<<<
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef SIZE_t min_samples_split = self.min_samples_split
|
|
*/
|
|
__pyx_t_9 = __pyx_v_self->__pyx_base.min_samples_leaf;
|
|
__pyx_v_min_samples_leaf = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2786
|
|
* cdef SIZE_t max_leaf_nodes = self.max_leaf_nodes
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t min_samples_split = self.min_samples_split
|
|
*
|
|
*/
|
|
__pyx_t_10 = __pyx_v_self->__pyx_base.min_weight_leaf;
|
|
__pyx_v_min_weight_leaf = __pyx_t_10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2787
|
|
* cdef SIZE_t min_samples_leaf = self.min_samples_leaf
|
|
* cdef double min_weight_leaf = self.min_weight_leaf
|
|
* cdef SIZE_t min_samples_split = self.min_samples_split # <<<<<<<<<<<<<<
|
|
*
|
|
* # Recursive partition (without actual recursion)
|
|
*/
|
|
__pyx_t_9 = __pyx_v_self->__pyx_base.min_samples_split;
|
|
__pyx_v_min_samples_split = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2790
|
|
*
|
|
* # Recursive partition (without actual recursion)
|
|
* splitter.init(X, y, sample_weight_ptr) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef PriorityHeap frontier = PriorityHeap(INITIAL_STACK_SIZE)
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->init(__pyx_v_splitter, __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight_ptr);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2792
|
|
* splitter.init(X, y, sample_weight_ptr)
|
|
*
|
|
* cdef PriorityHeap frontier = PriorityHeap(INITIAL_STACK_SIZE) # <<<<<<<<<<<<<<
|
|
* cdef PriorityHeapRecord record
|
|
* cdef PriorityHeapRecord split_node_left
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_7sklearn_4tree_5_tree_INITIAL_STACK_SIZE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2792; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2792; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_6_utils_PriorityHeap)), __pyx_t_4, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2792; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_v_frontier = ((struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2797
|
|
* cdef PriorityHeapRecord split_node_right
|
|
*
|
|
* cdef SIZE_t n_node_samples = splitter.n_samples # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t max_split_nodes = max_leaf_nodes - 1
|
|
* cdef bint is_leaf
|
|
*/
|
|
__pyx_t_9 = __pyx_v_splitter->n_samples;
|
|
__pyx_v_n_node_samples = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2798
|
|
*
|
|
* cdef SIZE_t n_node_samples = splitter.n_samples
|
|
* cdef SIZE_t max_split_nodes = max_leaf_nodes - 1 # <<<<<<<<<<<<<<
|
|
* cdef bint is_leaf
|
|
* cdef SIZE_t max_depth_seen = -1
|
|
*/
|
|
__pyx_v_max_split_nodes = (__pyx_v_max_leaf_nodes - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2800
|
|
* cdef SIZE_t max_split_nodes = max_leaf_nodes - 1
|
|
* cdef bint is_leaf
|
|
* cdef SIZE_t max_depth_seen = -1 # <<<<<<<<<<<<<<
|
|
* cdef int rc = 0
|
|
* cdef Node* node
|
|
*/
|
|
__pyx_v_max_depth_seen = -1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2801
|
|
* cdef bint is_leaf
|
|
* cdef SIZE_t max_depth_seen = -1
|
|
* cdef int rc = 0 # <<<<<<<<<<<<<<
|
|
* cdef Node* node
|
|
*
|
|
*/
|
|
__pyx_v_rc = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2805
|
|
*
|
|
* # Initial capacity
|
|
* cdef SIZE_t init_capacity = max_split_nodes + max_leaf_nodes # <<<<<<<<<<<<<<
|
|
* tree._resize(init_capacity)
|
|
*
|
|
*/
|
|
__pyx_v_init_capacity = (__pyx_v_max_split_nodes + __pyx_v_max_leaf_nodes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2806
|
|
* # Initial capacity
|
|
* cdef SIZE_t init_capacity = max_split_nodes + max_leaf_nodes
|
|
* tree._resize(init_capacity) # <<<<<<<<<<<<<<
|
|
*
|
|
* with nogil:
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_tree->__pyx_vtab)->_resize(__pyx_v_tree, __pyx_v_init_capacity); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2806; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2808
|
|
* tree._resize(init_capacity)
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* # add root to frontier
|
|
* rc = self._add_split_node(splitter, tree, 0, n_node_samples,
|
|
*/
|
|
{
|
|
#ifdef WITH_THREAD
|
|
PyThreadState *_save;
|
|
Py_UNBLOCK_THREADS
|
|
#endif
|
|
/*try:*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2810
|
|
* with nogil:
|
|
* # add root to frontier
|
|
* rc = self._add_split_node(splitter, tree, 0, n_node_samples, # <<<<<<<<<<<<<<
|
|
* INFINITY, IS_FIRST, IS_LEFT, NULL, 0,
|
|
* &split_node_left)
|
|
*/
|
|
__pyx_v_rc = __pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder__add_split_node(__pyx_v_self, __pyx_v_splitter, __pyx_v_tree, 0, __pyx_v_n_node_samples, __pyx_v_7sklearn_4tree_5_tree_INFINITY, __pyx_v_7sklearn_4tree_5_tree_IS_FIRST, __pyx_v_7sklearn_4tree_5_tree_IS_LEFT, NULL, 0, (&__pyx_v_split_node_left));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2813
|
|
* INFINITY, IS_FIRST, IS_LEFT, NULL, 0,
|
|
* &split_node_left)
|
|
* if rc >= 0: # <<<<<<<<<<<<<<
|
|
* rc = _add_to_frontier(&split_node_left, frontier)
|
|
* if rc == -1:
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_rc >= 0) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2814
|
|
* &split_node_left)
|
|
* if rc >= 0:
|
|
* rc = _add_to_frontier(&split_node_left, frontier) # <<<<<<<<<<<<<<
|
|
* if rc == -1:
|
|
* raise MemoryError()
|
|
*/
|
|
__pyx_v_rc = __pyx_f_7sklearn_4tree_5_tree__add_to_frontier((&__pyx_v_split_node_left), __pyx_v_frontier);
|
|
goto __pyx_L9;
|
|
}
|
|
__pyx_L9:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2808
|
|
* tree._resize(init_capacity)
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* # add root to frontier
|
|
* rc = self._add_split_node(splitter, tree, 0, n_node_samples,
|
|
*/
|
|
/*finally:*/ {
|
|
/*normal exit:*/{
|
|
#ifdef WITH_THREAD
|
|
Py_BLOCK_THREADS
|
|
#endif
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2815
|
|
* if rc >= 0:
|
|
* rc = _add_to_frontier(&split_node_left, frontier)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* raise MemoryError()
|
|
*
|
|
*/
|
|
__pyx_t_8 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2816
|
|
* rc = _add_to_frontier(&split_node_left, frontier)
|
|
* if rc == -1:
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
* with nogil:
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2816; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2818
|
|
* raise MemoryError()
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* while not frontier.is_empty():
|
|
* frontier.pop(&record)
|
|
*/
|
|
{
|
|
#ifdef WITH_THREAD
|
|
PyThreadState *_save;
|
|
Py_UNBLOCK_THREADS
|
|
#endif
|
|
/*try:*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2819
|
|
*
|
|
* with nogil:
|
|
* while not frontier.is_empty(): # <<<<<<<<<<<<<<
|
|
* frontier.pop(&record)
|
|
*
|
|
*/
|
|
while (1) {
|
|
__pyx_t_8 = ((!(((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap *)__pyx_v_frontier->__pyx_vtab)->is_empty(__pyx_v_frontier) != 0)) != 0);
|
|
if (!__pyx_t_8) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2820
|
|
* with nogil:
|
|
* while not frontier.is_empty():
|
|
* frontier.pop(&record) # <<<<<<<<<<<<<<
|
|
*
|
|
* node = &tree.nodes[record.node_id]
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap *)__pyx_v_frontier->__pyx_vtab)->pop(__pyx_v_frontier, (&__pyx_v_record));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2822
|
|
* frontier.pop(&record)
|
|
*
|
|
* node = &tree.nodes[record.node_id] # <<<<<<<<<<<<<<
|
|
* is_leaf = (record.is_leaf or max_split_nodes <= 0)
|
|
*
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_tree->nodes[__pyx_v_record.node_id]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2823
|
|
*
|
|
* node = &tree.nodes[record.node_id]
|
|
* is_leaf = (record.is_leaf or max_split_nodes <= 0) # <<<<<<<<<<<<<<
|
|
*
|
|
* if is_leaf:
|
|
*/
|
|
if (!__pyx_v_record.is_leaf) {
|
|
__pyx_t_8 = (__pyx_v_max_split_nodes <= 0);
|
|
__pyx_t_7 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_7 = __pyx_v_record.is_leaf;
|
|
}
|
|
__pyx_v_is_leaf = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2825
|
|
* is_leaf = (record.is_leaf or max_split_nodes <= 0)
|
|
*
|
|
* if is_leaf: # <<<<<<<<<<<<<<
|
|
* # Node is not expandable; set node as leaf
|
|
* node.left_child = _TREE_LEAF
|
|
*/
|
|
__pyx_t_7 = (__pyx_v_is_leaf != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2827
|
|
* if is_leaf:
|
|
* # Node is not expandable; set node as leaf
|
|
* node.left_child = _TREE_LEAF # <<<<<<<<<<<<<<
|
|
* node.right_child = _TREE_LEAF
|
|
* node.feature = _TREE_UNDEFINED
|
|
*/
|
|
__pyx_v_node->left_child = __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2828
|
|
* # Node is not expandable; set node as leaf
|
|
* node.left_child = _TREE_LEAF
|
|
* node.right_child = _TREE_LEAF # <<<<<<<<<<<<<<
|
|
* node.feature = _TREE_UNDEFINED
|
|
* node.threshold = _TREE_UNDEFINED
|
|
*/
|
|
__pyx_v_node->right_child = __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2829
|
|
* node.left_child = _TREE_LEAF
|
|
* node.right_child = _TREE_LEAF
|
|
* node.feature = _TREE_UNDEFINED # <<<<<<<<<<<<<<
|
|
* node.threshold = _TREE_UNDEFINED
|
|
*
|
|
*/
|
|
__pyx_v_node->feature = __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2830
|
|
* node.right_child = _TREE_LEAF
|
|
* node.feature = _TREE_UNDEFINED
|
|
* node.threshold = _TREE_UNDEFINED # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_node->threshold = __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED;
|
|
goto __pyx_L16;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2836
|
|
*
|
|
* # Decrement number of split nodes available
|
|
* max_split_nodes -= 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Compute left split node
|
|
*/
|
|
__pyx_v_max_split_nodes = (__pyx_v_max_split_nodes - 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2839
|
|
*
|
|
* # Compute left split node
|
|
* rc = self._add_split_node(splitter, tree, # <<<<<<<<<<<<<<
|
|
* record.start, record.pos,
|
|
* record.impurity_left,
|
|
*/
|
|
__pyx_v_rc = __pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder__add_split_node(__pyx_v_self, __pyx_v_splitter, __pyx_v_tree, __pyx_v_record.start, __pyx_v_record.pos, __pyx_v_record.impurity_left, __pyx_v_7sklearn_4tree_5_tree_IS_NOT_FIRST, __pyx_v_7sklearn_4tree_5_tree_IS_LEFT, __pyx_v_node, (__pyx_v_record.depth + 1), (&__pyx_v_split_node_left));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2845
|
|
* record.depth + 1,
|
|
* &split_node_left)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2846
|
|
* &split_node_left)
|
|
* if rc == -1:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* # tree.nodes may have changed
|
|
*/
|
|
goto __pyx_L15_break;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2849
|
|
*
|
|
* # tree.nodes may have changed
|
|
* node = &tree.nodes[record.node_id] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Compute right split node
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_tree->nodes[__pyx_v_record.node_id]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2852
|
|
*
|
|
* # Compute right split node
|
|
* rc = self._add_split_node(splitter, tree, record.pos, # <<<<<<<<<<<<<<
|
|
* record.end,
|
|
* record.impurity_right,
|
|
*/
|
|
__pyx_v_rc = __pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder__add_split_node(__pyx_v_self, __pyx_v_splitter, __pyx_v_tree, __pyx_v_record.pos, __pyx_v_record.end, __pyx_v_record.impurity_right, __pyx_v_7sklearn_4tree_5_tree_IS_NOT_FIRST, __pyx_v_7sklearn_4tree_5_tree_IS_NOT_LEFT, __pyx_v_node, (__pyx_v_record.depth + 1), (&__pyx_v_split_node_right));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2858
|
|
* record.depth + 1,
|
|
* &split_node_right)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2859
|
|
* &split_node_right)
|
|
* if rc == -1:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* # Add nodes to queue
|
|
*/
|
|
goto __pyx_L15_break;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2862
|
|
*
|
|
* # Add nodes to queue
|
|
* rc = _add_to_frontier(&split_node_left, frontier) # <<<<<<<<<<<<<<
|
|
* if rc == -1:
|
|
* break
|
|
*/
|
|
__pyx_v_rc = __pyx_f_7sklearn_4tree_5_tree__add_to_frontier((&__pyx_v_split_node_left), __pyx_v_frontier);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2863
|
|
* # Add nodes to queue
|
|
* rc = _add_to_frontier(&split_node_left, frontier)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2864
|
|
* rc = _add_to_frontier(&split_node_left, frontier)
|
|
* if rc == -1:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* rc = _add_to_frontier(&split_node_right, frontier)
|
|
*/
|
|
goto __pyx_L15_break;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2866
|
|
* break
|
|
*
|
|
* rc = _add_to_frontier(&split_node_right, frontier) # <<<<<<<<<<<<<<
|
|
* if rc == -1:
|
|
* break
|
|
*/
|
|
__pyx_v_rc = __pyx_f_7sklearn_4tree_5_tree__add_to_frontier((&__pyx_v_split_node_right), __pyx_v_frontier);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2867
|
|
*
|
|
* rc = _add_to_frontier(&split_node_right, frontier)
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* break
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2868
|
|
* rc = _add_to_frontier(&split_node_right, frontier)
|
|
* if rc == -1:
|
|
* break # <<<<<<<<<<<<<<
|
|
*
|
|
* if record.depth > max_depth_seen:
|
|
*/
|
|
goto __pyx_L15_break;
|
|
}
|
|
}
|
|
__pyx_L16:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2870
|
|
* break
|
|
*
|
|
* if record.depth > max_depth_seen: # <<<<<<<<<<<<<<
|
|
* max_depth_seen = record.depth
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_record.depth > __pyx_v_max_depth_seen) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2871
|
|
*
|
|
* if record.depth > max_depth_seen:
|
|
* max_depth_seen = record.depth # <<<<<<<<<<<<<<
|
|
*
|
|
* if rc >= 0:
|
|
*/
|
|
__pyx_t_11 = __pyx_v_record.depth;
|
|
__pyx_v_max_depth_seen = __pyx_t_11;
|
|
goto __pyx_L21;
|
|
}
|
|
__pyx_L21:;
|
|
}
|
|
__pyx_L15_break:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2873
|
|
* max_depth_seen = record.depth
|
|
*
|
|
* if rc >= 0: # <<<<<<<<<<<<<<
|
|
* rc = tree._resize_c(tree.node_count)
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc >= 0) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2874
|
|
*
|
|
* if rc >= 0:
|
|
* rc = tree._resize_c(tree.node_count) # <<<<<<<<<<<<<<
|
|
*
|
|
* if rc >= 0:
|
|
*/
|
|
__pyx_t_13.__pyx_n = 1;
|
|
__pyx_t_13.capacity = __pyx_v_tree->node_count;
|
|
__pyx_t_12 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_tree->__pyx_vtab)->_resize_c(__pyx_v_tree, &__pyx_t_13);
|
|
__pyx_v_rc = __pyx_t_12;
|
|
goto __pyx_L22;
|
|
}
|
|
__pyx_L22:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2876
|
|
* rc = tree._resize_c(tree.node_count)
|
|
*
|
|
* if rc >= 0: # <<<<<<<<<<<<<<
|
|
* tree.max_depth = max_depth_seen
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc >= 0) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2877
|
|
*
|
|
* if rc >= 0:
|
|
* tree.max_depth = max_depth_seen # <<<<<<<<<<<<<<
|
|
*
|
|
* if rc == -1:
|
|
*/
|
|
__pyx_v_tree->max_depth = __pyx_v_max_depth_seen;
|
|
goto __pyx_L23;
|
|
}
|
|
__pyx_L23:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2818
|
|
* raise MemoryError()
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* while not frontier.is_empty():
|
|
* frontier.pop(&record)
|
|
*/
|
|
/*finally:*/ {
|
|
/*normal exit:*/{
|
|
#ifdef WITH_THREAD
|
|
Py_BLOCK_THREADS
|
|
#endif
|
|
goto __pyx_L13;
|
|
}
|
|
__pyx_L13:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2879
|
|
* tree.max_depth = max_depth_seen
|
|
*
|
|
* if rc == -1: # <<<<<<<<<<<<<<
|
|
* raise MemoryError()
|
|
*
|
|
*/
|
|
__pyx_t_7 = ((__pyx_v_rc == -1) != 0);
|
|
if (__pyx_t_7) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2880
|
|
*
|
|
* if rc == -1:
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline int _add_split_node(self, Splitter splitter, Tree tree,
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2880; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2771
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
/* 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_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestFirstTreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_splitter);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_frontier);
|
|
__Pyx_XDECREF(__pyx_v_X);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_y);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_sample_weight);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_3build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_2build[] = "Build a decision tree from the training set (X, y).";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_3build(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree = 0;
|
|
PyObject *__pyx_v_X = 0;
|
|
PyArrayObject *__pyx_v_y = 0;
|
|
PyArrayObject *__pyx_v_sample_weight = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("build (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_tree,&__pyx_n_s_X,&__pyx_n_s_y,&__pyx_n_s_sample_weight,0};
|
|
PyObject* values[4] = {0,0,0,0};
|
|
|
|
/* "sklearn/tree/_tree.pyx":2772
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y,
|
|
* np.ndarray sample_weight=None): # <<<<<<<<<<<<<<
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*
|
|
*/
|
|
values[3] = (PyObject *)((PyArrayObject *)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 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_tree)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_X)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_y)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 3:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s_sample_weight);
|
|
if (value) { values[3] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "build") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 4: values[3] = PyTuple_GET_ITEM(__pyx_args, 3);
|
|
case 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_tree = ((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)values[0]);
|
|
__pyx_v_X = values[1];
|
|
__pyx_v_y = ((PyArrayObject *)values[2]);
|
|
__pyx_v_sample_weight = ((PyArrayObject *)values[3]);
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("build", 0, 3, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestFirstTreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_tree), __pyx_ptype_7sklearn_4tree_5_tree_Tree, 1, "tree", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_y), __pyx_ptype_5numpy_ndarray, 1, "y", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_sample_weight), __pyx_ptype_5numpy_ndarray, 1, "sample_weight", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2772; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_2build(((struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *)__pyx_v_self), __pyx_v_tree, __pyx_v_X, __pyx_v_y, __pyx_v_sample_weight);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2771
|
|
* self.max_leaf_nodes = max_leaf_nodes
|
|
*
|
|
* cpdef build(self, Tree tree, object X, np.ndarray y, # <<<<<<<<<<<<<<
|
|
* np.ndarray sample_weight=None):
|
|
* """Build a decision tree from the training set (X, y)."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_2build(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, PyObject *__pyx_v_X, PyArrayObject *__pyx_v_y, PyArrayObject *__pyx_v_sample_weight) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build __pyx_t_2;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("build", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2.__pyx_n = 1;
|
|
__pyx_t_2.sample_weight = __pyx_v_sample_weight;
|
|
__pyx_t_1 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestFirstTreeBuilder *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.build(((struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)__pyx_v_self), __pyx_v_tree, __pyx_v_X, __pyx_v_y, 1, &__pyx_t_2); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.BestFirstTreeBuilder.build", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2882
|
|
* raise MemoryError()
|
|
*
|
|
* cdef inline int _add_split_node(self, Splitter splitter, Tree tree, # <<<<<<<<<<<<<<
|
|
* SIZE_t start, SIZE_t end, double impurity,
|
|
* bint is_first, bint is_left, Node* parent,
|
|
*/
|
|
|
|
static CYTHON_INLINE int __pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder__add_split_node(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *__pyx_v_self, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *__pyx_v_splitter, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_tree, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_start, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_end, double __pyx_v_impurity, int __pyx_v_is_first, int __pyx_v_is_left, struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_parent, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_depth, struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *__pyx_v_res) {
|
|
struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord __pyx_v_split;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_node_id;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_node_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_constant_features;
|
|
CYTHON_UNUSED double __pyx_v_weighted_n_samples;
|
|
double __pyx_v_weighted_n_node_samples;
|
|
int __pyx_v_is_leaf;
|
|
int __pyx_r;
|
|
double __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
int __pyx_t_4;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
ptrdiff_t __pyx_t_8;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2891
|
|
* cdef SIZE_t node_id
|
|
* cdef SIZE_t n_node_samples
|
|
* cdef SIZE_t n_constant_features = 0 # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_samples = splitter.weighted_n_samples
|
|
* cdef double weighted_n_node_samples
|
|
*/
|
|
__pyx_v_n_constant_features = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2892
|
|
* cdef SIZE_t n_node_samples
|
|
* cdef SIZE_t n_constant_features = 0
|
|
* cdef double weighted_n_samples = splitter.weighted_n_samples # <<<<<<<<<<<<<<
|
|
* cdef double weighted_n_node_samples
|
|
* cdef bint is_leaf
|
|
*/
|
|
__pyx_t_1 = __pyx_v_splitter->weighted_n_samples;
|
|
__pyx_v_weighted_n_samples = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2898
|
|
* cdef double imp_diff
|
|
*
|
|
* splitter.node_reset(start, end, &weighted_n_node_samples) # <<<<<<<<<<<<<<
|
|
*
|
|
* if is_first:
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_reset(__pyx_v_splitter, __pyx_v_start, __pyx_v_end, (&__pyx_v_weighted_n_node_samples));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2900
|
|
* splitter.node_reset(start, end, &weighted_n_node_samples)
|
|
*
|
|
* if is_first: # <<<<<<<<<<<<<<
|
|
* impurity = splitter.node_impurity()
|
|
*
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_is_first != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2901
|
|
*
|
|
* if is_first:
|
|
* impurity = splitter.node_impurity() # <<<<<<<<<<<<<<
|
|
*
|
|
* n_node_samples = end - start
|
|
*/
|
|
__pyx_v_impurity = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_impurity(__pyx_v_splitter);
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2903
|
|
* impurity = splitter.node_impurity()
|
|
*
|
|
* n_node_samples = end - start # <<<<<<<<<<<<<<
|
|
* is_leaf = ((depth > self.max_depth) or
|
|
* (n_node_samples < self.min_samples_split) or
|
|
*/
|
|
__pyx_v_n_node_samples = (__pyx_v_end - __pyx_v_start);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2904
|
|
*
|
|
* n_node_samples = end - start
|
|
* is_leaf = ((depth > self.max_depth) or # <<<<<<<<<<<<<<
|
|
* (n_node_samples < self.min_samples_split) or
|
|
* (n_node_samples < 2 * self.min_samples_leaf) or
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_depth > __pyx_v_self->__pyx_base.max_depth);
|
|
if (!__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2905
|
|
* n_node_samples = end - start
|
|
* is_leaf = ((depth > self.max_depth) or
|
|
* (n_node_samples < self.min_samples_split) or # <<<<<<<<<<<<<<
|
|
* (n_node_samples < 2 * self.min_samples_leaf) or
|
|
* (weighted_n_node_samples < self.min_weight_leaf) or
|
|
*/
|
|
__pyx_t_3 = (__pyx_v_n_node_samples < __pyx_v_self->__pyx_base.min_samples_split);
|
|
if (!__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2906
|
|
* is_leaf = ((depth > self.max_depth) or
|
|
* (n_node_samples < self.min_samples_split) or
|
|
* (n_node_samples < 2 * self.min_samples_leaf) or # <<<<<<<<<<<<<<
|
|
* (weighted_n_node_samples < self.min_weight_leaf) or
|
|
* (impurity <= MIN_IMPURITY_SPLIT))
|
|
*/
|
|
__pyx_t_4 = (__pyx_v_n_node_samples < (2 * __pyx_v_self->__pyx_base.min_samples_leaf));
|
|
if (!__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2907
|
|
* (n_node_samples < self.min_samples_split) or
|
|
* (n_node_samples < 2 * self.min_samples_leaf) or
|
|
* (weighted_n_node_samples < self.min_weight_leaf) or # <<<<<<<<<<<<<<
|
|
* (impurity <= MIN_IMPURITY_SPLIT))
|
|
*
|
|
*/
|
|
__pyx_t_5 = (__pyx_v_weighted_n_node_samples < __pyx_v_self->__pyx_base.min_weight_leaf);
|
|
if (!__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2908
|
|
* (n_node_samples < 2 * self.min_samples_leaf) or
|
|
* (weighted_n_node_samples < self.min_weight_leaf) or
|
|
* (impurity <= MIN_IMPURITY_SPLIT)) # <<<<<<<<<<<<<<
|
|
*
|
|
* if not is_leaf:
|
|
*/
|
|
__pyx_t_6 = (__pyx_v_impurity <= __pyx_v_7sklearn_4tree_5_tree_MIN_IMPURITY_SPLIT);
|
|
__pyx_t_7 = __pyx_t_6;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_5;
|
|
}
|
|
__pyx_t_5 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_5 = __pyx_t_4;
|
|
}
|
|
__pyx_t_4 = __pyx_t_5;
|
|
} else {
|
|
__pyx_t_4 = __pyx_t_3;
|
|
}
|
|
__pyx_t_3 = __pyx_t_4;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_2;
|
|
}
|
|
__pyx_v_is_leaf = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2910
|
|
* (impurity <= MIN_IMPURITY_SPLIT))
|
|
*
|
|
* if not is_leaf: # <<<<<<<<<<<<<<
|
|
* splitter.node_split(impurity, &split, &n_constant_features)
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*/
|
|
__pyx_t_3 = ((!(__pyx_v_is_leaf != 0)) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2911
|
|
*
|
|
* if not is_leaf:
|
|
* splitter.node_split(impurity, &split, &n_constant_features) # <<<<<<<<<<<<<<
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_split(__pyx_v_splitter, __pyx_v_impurity, (&__pyx_v_split), (&__pyx_v_n_constant_features));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2912
|
|
* if not is_leaf:
|
|
* splitter.node_split(impurity, &split, &n_constant_features)
|
|
* is_leaf = is_leaf or (split.pos >= end) # <<<<<<<<<<<<<<
|
|
*
|
|
* node_id = tree._add_node(parent - tree.nodes
|
|
*/
|
|
if (!__pyx_v_is_leaf) {
|
|
__pyx_t_3 = (__pyx_v_split.pos >= __pyx_v_end);
|
|
__pyx_t_2 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_2 = __pyx_v_is_leaf;
|
|
}
|
|
__pyx_v_is_leaf = __pyx_t_2;
|
|
goto __pyx_L4;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2915
|
|
*
|
|
* node_id = tree._add_node(parent - tree.nodes
|
|
* if parent != NULL # <<<<<<<<<<<<<<
|
|
* else _TREE_UNDEFINED,
|
|
* is_left, is_leaf,
|
|
*/
|
|
if (((__pyx_v_parent != NULL) != 0)) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2914
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*
|
|
* node_id = tree._add_node(parent - tree.nodes # <<<<<<<<<<<<<<
|
|
* if parent != NULL
|
|
* else _TREE_UNDEFINED,
|
|
*/
|
|
__pyx_t_8 = (__pyx_v_parent - __pyx_v_tree->nodes);
|
|
} else {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2916
|
|
* node_id = tree._add_node(parent - tree.nodes
|
|
* if parent != NULL
|
|
* else _TREE_UNDEFINED, # <<<<<<<<<<<<<<
|
|
* is_left, is_leaf,
|
|
* split.feature, split.threshold, impurity, n_node_samples,
|
|
*/
|
|
__pyx_t_8 = __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2914
|
|
* is_leaf = is_leaf or (split.pos >= end)
|
|
*
|
|
* node_id = tree._add_node(parent - tree.nodes # <<<<<<<<<<<<<<
|
|
* if parent != NULL
|
|
* else _TREE_UNDEFINED,
|
|
*/
|
|
__pyx_v_node_id = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_tree->__pyx_vtab)->_add_node(__pyx_v_tree, __pyx_t_8, __pyx_v_is_left, __pyx_v_is_leaf, __pyx_v_split.feature, __pyx_v_split.threshold, __pyx_v_impurity, __pyx_v_n_node_samples, __pyx_v_weighted_n_node_samples);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2920
|
|
* split.feature, split.threshold, impurity, n_node_samples,
|
|
* weighted_n_node_samples)
|
|
* if node_id == <SIZE_t>(-1): # <<<<<<<<<<<<<<
|
|
* return -1
|
|
*
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_node_id == ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)-1)) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2921
|
|
* weighted_n_node_samples)
|
|
* if node_id == <SIZE_t>(-1):
|
|
* return -1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # compute values also for split nodes (might become leafs later).
|
|
*/
|
|
__pyx_r = -1;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":2924
|
|
*
|
|
* # compute values also for split nodes (might become leafs later).
|
|
* splitter.node_value(tree.value + node_id * tree.value_stride) # <<<<<<<<<<<<<<
|
|
*
|
|
* res.node_id = node_id
|
|
*/
|
|
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter *)__pyx_v_splitter->__pyx_vtab)->node_value(__pyx_v_splitter, (__pyx_v_tree->value + (__pyx_v_node_id * __pyx_v_tree->value_stride)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":2926
|
|
* splitter.node_value(tree.value + node_id * tree.value_stride)
|
|
*
|
|
* res.node_id = node_id # <<<<<<<<<<<<<<
|
|
* res.start = start
|
|
* res.end = end
|
|
*/
|
|
__pyx_v_res->node_id = __pyx_v_node_id;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2927
|
|
*
|
|
* res.node_id = node_id
|
|
* res.start = start # <<<<<<<<<<<<<<
|
|
* res.end = end
|
|
* res.depth = depth
|
|
*/
|
|
__pyx_v_res->start = __pyx_v_start;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2928
|
|
* res.node_id = node_id
|
|
* res.start = start
|
|
* res.end = end # <<<<<<<<<<<<<<
|
|
* res.depth = depth
|
|
* res.impurity = impurity
|
|
*/
|
|
__pyx_v_res->end = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2929
|
|
* res.start = start
|
|
* res.end = end
|
|
* res.depth = depth # <<<<<<<<<<<<<<
|
|
* res.impurity = impurity
|
|
*
|
|
*/
|
|
__pyx_v_res->depth = __pyx_v_depth;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2930
|
|
* res.end = end
|
|
* res.depth = depth
|
|
* res.impurity = impurity # <<<<<<<<<<<<<<
|
|
*
|
|
* if not is_leaf:
|
|
*/
|
|
__pyx_v_res->impurity = __pyx_v_impurity;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2932
|
|
* res.impurity = impurity
|
|
*
|
|
* if not is_leaf: # <<<<<<<<<<<<<<
|
|
* # is split node
|
|
* res.pos = split.pos
|
|
*/
|
|
__pyx_t_2 = ((!(__pyx_v_is_leaf != 0)) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2934
|
|
* if not is_leaf:
|
|
* # is split node
|
|
* res.pos = split.pos # <<<<<<<<<<<<<<
|
|
* res.is_leaf = 0
|
|
* res.improvement = split.improvement
|
|
*/
|
|
__pyx_t_9 = __pyx_v_split.pos;
|
|
__pyx_v_res->pos = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2935
|
|
* # is split node
|
|
* res.pos = split.pos
|
|
* res.is_leaf = 0 # <<<<<<<<<<<<<<
|
|
* res.improvement = split.improvement
|
|
* res.impurity_left = split.impurity_left
|
|
*/
|
|
__pyx_v_res->is_leaf = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2936
|
|
* res.pos = split.pos
|
|
* res.is_leaf = 0
|
|
* res.improvement = split.improvement # <<<<<<<<<<<<<<
|
|
* res.impurity_left = split.impurity_left
|
|
* res.impurity_right = split.impurity_right
|
|
*/
|
|
__pyx_t_1 = __pyx_v_split.improvement;
|
|
__pyx_v_res->improvement = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2937
|
|
* res.is_leaf = 0
|
|
* res.improvement = split.improvement
|
|
* res.impurity_left = split.impurity_left # <<<<<<<<<<<<<<
|
|
* res.impurity_right = split.impurity_right
|
|
*
|
|
*/
|
|
__pyx_t_1 = __pyx_v_split.impurity_left;
|
|
__pyx_v_res->impurity_left = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2938
|
|
* res.improvement = split.improvement
|
|
* res.impurity_left = split.impurity_left
|
|
* res.impurity_right = split.impurity_right # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = __pyx_v_split.impurity_right;
|
|
__pyx_v_res->impurity_right = __pyx_t_1;
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":2942
|
|
* else:
|
|
* # is leaf => 0 improvement
|
|
* res.pos = end # <<<<<<<<<<<<<<
|
|
* res.is_leaf = 1
|
|
* res.improvement = 0.0
|
|
*/
|
|
__pyx_v_res->pos = __pyx_v_end;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2943
|
|
* # is leaf => 0 improvement
|
|
* res.pos = end
|
|
* res.is_leaf = 1 # <<<<<<<<<<<<<<
|
|
* res.improvement = 0.0
|
|
* res.impurity_left = impurity
|
|
*/
|
|
__pyx_v_res->is_leaf = 1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2944
|
|
* res.pos = end
|
|
* res.is_leaf = 1
|
|
* res.improvement = 0.0 # <<<<<<<<<<<<<<
|
|
* res.impurity_left = impurity
|
|
* res.impurity_right = impurity
|
|
*/
|
|
__pyx_v_res->improvement = 0.0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2945
|
|
* res.is_leaf = 1
|
|
* res.improvement = 0.0
|
|
* res.impurity_left = impurity # <<<<<<<<<<<<<<
|
|
* res.impurity_right = impurity
|
|
*
|
|
*/
|
|
__pyx_v_res->impurity_left = __pyx_v_impurity;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2946
|
|
* res.improvement = 0.0
|
|
* res.impurity_left = impurity
|
|
* res.impurity_right = impurity # <<<<<<<<<<<<<<
|
|
*
|
|
* return 0
|
|
*/
|
|
__pyx_v_res->impurity_right = __pyx_v_impurity;
|
|
}
|
|
__pyx_L6:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2948
|
|
* res.impurity_right = impurity
|
|
*
|
|
* return 0 # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":2882
|
|
* raise MemoryError()
|
|
*
|
|
* cdef inline int _add_split_node(self, Splitter splitter, Tree tree, # <<<<<<<<<<<<<<
|
|
* SIZE_t start, SIZE_t end, double impurity,
|
|
* bint is_first, bint is_left, Node* parent,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3014
|
|
* # (i.e. through `_resize` or `__setstate__`)
|
|
* property n_classes:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* # it's small; copy for memory safety
|
|
* return sizet_ptr_to_ndarray(self.n_classes, self.n_outputs).copy()
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_classes_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_classes_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_classes___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_classes___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3016
|
|
* def __get__(self):
|
|
* # it's small; copy for memory safety
|
|
* return sizet_ptr_to_ndarray(self.n_classes, self.n_outputs).copy() # <<<<<<<<<<<<<<
|
|
*
|
|
* property children_left:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)__pyx_f_7sklearn_4tree_5_tree_sizet_ptr_to_ndarray(__pyx_v_self->n_classes, __pyx_v_self->n_outputs)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3016; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_copy); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3016; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3016; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3014
|
|
* # (i.e. through `_resize` or `__setstate__`)
|
|
* property n_classes:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* # it's small; copy for memory safety
|
|
* return sizet_ptr_to_ndarray(self.n_classes, self.n_outputs).copy()
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.n_classes.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3019
|
|
*
|
|
* property children_left:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['left_child'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13children_left_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13children_left_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_13children_left___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_13children_left___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3020
|
|
* property children_left:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['left_child'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property children_right:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3020; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_left_child); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3020; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3020; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3019
|
|
*
|
|
* property children_left:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['left_child'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.children_left.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3023
|
|
*
|
|
* property children_right:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['right_child'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_14children_right_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_14children_right_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_14children_right___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_14children_right___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3024
|
|
* property children_right:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['right_child'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property feature:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3024; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_right_child); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3024; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3024; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3023
|
|
*
|
|
* property children_right:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['right_child'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.children_right.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3027
|
|
*
|
|
* property feature:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['feature'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_7feature_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_7feature_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_7feature___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_7feature___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3028
|
|
* property feature:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['feature'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property threshold:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3028; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_feature); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3028; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3028; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3027
|
|
*
|
|
* property feature:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['feature'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.feature.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3031
|
|
*
|
|
* property threshold:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['threshold'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9threshold_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9threshold_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_9threshold___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9threshold___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3032
|
|
* property threshold:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['threshold'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property impurity:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3032; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_threshold); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3032; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3032; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3031
|
|
*
|
|
* property threshold:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['threshold'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.threshold.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3035
|
|
*
|
|
* property impurity:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['impurity'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_8impurity_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_8impurity_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_8impurity___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_8impurity___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3036
|
|
* property impurity:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['impurity'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property n_node_samples:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3036; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_impurity); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3036; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3036; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3035
|
|
*
|
|
* property impurity:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['impurity'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.impurity.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3039
|
|
*
|
|
* property n_node_samples:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['n_node_samples'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_14n_node_samples_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_14n_node_samples_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_14n_node_samples___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_14n_node_samples___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3040
|
|
* property n_node_samples:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['n_node_samples'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property weighted_n_node_samples:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3040; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_n_node_samples); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3040; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3040; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3039
|
|
*
|
|
* property n_node_samples:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['n_node_samples'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.n_node_samples.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3043
|
|
*
|
|
* property weighted_n_node_samples:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['weighted_n_node_samples'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_23weighted_n_node_samples_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_23weighted_n_node_samples_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_23weighted_n_node_samples___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_23weighted_n_node_samples___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3044
|
|
* property weighted_n_node_samples:
|
|
* def __get__(self):
|
|
* return self._get_node_ndarray()['weighted_n_node_samples'][:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* property value:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3044; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_GetItem(__pyx_t_1, __pyx_n_s_weighted_n_node_samples); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3044; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_GetSlice(__pyx_t_2, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3044; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3043
|
|
*
|
|
* property weighted_n_node_samples:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_node_ndarray()['weighted_n_node_samples'][:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.weighted_n_node_samples.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3047
|
|
*
|
|
* property value:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_value_ndarray()[:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_5value_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_5value_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_5value___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_5value___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__get__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3048
|
|
* property value:
|
|
* def __get__(self):
|
|
* return self._get_value_ndarray()[:self.node_count] # <<<<<<<<<<<<<<
|
|
*
|
|
* def __cinit__(self, int n_features, np.ndarray[SIZE_t, ndim=1] n_classes,
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_value_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3048; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyObject_GetSlice(__pyx_t_1, 0, __pyx_v_self->node_count, NULL, NULL, NULL, 0, 1, 0); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3048; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_r = __pyx_t_2;
|
|
__pyx_t_2 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3047
|
|
*
|
|
* property value:
|
|
* def __get__(self): # <<<<<<<<<<<<<<
|
|
* return self._get_value_ndarray()[:self.node_count]
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.value.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3050
|
|
* return self._get_value_ndarray()[:self.node_count]
|
|
*
|
|
* def __cinit__(self, int n_features, np.ndarray[SIZE_t, ndim=1] n_classes, # <<<<<<<<<<<<<<
|
|
* int n_outputs):
|
|
* """Constructor."""
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_1__cinit__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
int __pyx_v_n_features;
|
|
PyArrayObject *__pyx_v_n_classes = 0;
|
|
int __pyx_v_n_outputs;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__cinit__ (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_n_features,&__pyx_n_s_n_classes,&__pyx_n_s_n_outputs,0};
|
|
PyObject* values[3] = {0,0,0};
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 3: values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
case 2: values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (likely((values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_n_features)) != 0)) kw_args--;
|
|
else goto __pyx_L5_argtuple_error;
|
|
case 1:
|
|
if (likely((values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_n_classes)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 3, 3, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
case 2:
|
|
if (likely((values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s_n_outputs)) != 0)) kw_args--;
|
|
else {
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 3, 3, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "__cinit__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else if (PyTuple_GET_SIZE(__pyx_args) != 3) {
|
|
goto __pyx_L5_argtuple_error;
|
|
} else {
|
|
values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
values[1] = PyTuple_GET_ITEM(__pyx_args, 1);
|
|
values[2] = PyTuple_GET_ITEM(__pyx_args, 2);
|
|
}
|
|
__pyx_v_n_features = __Pyx_PyInt_As_int(values[0]); if (unlikely((__pyx_v_n_features == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_v_n_classes = ((PyArrayObject *)values[1]);
|
|
__pyx_v_n_outputs = __Pyx_PyInt_As_int(values[2]); if (unlikely((__pyx_v_n_outputs == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3051; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("__cinit__", 1, 3, 3, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_n_classes), __pyx_ptype_5numpy_ndarray, 1, "n_classes", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree___cinit__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), __pyx_v_n_features, __pyx_v_n_classes, __pyx_v_n_outputs);
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree___cinit__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, int __pyx_v_n_features, PyArrayObject *__pyx_v_n_classes, int __pyx_v_n_outputs) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_k;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_n_classes;
|
|
__Pyx_Buffer __pyx_pybuffer_n_classes;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_5;
|
|
int __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__cinit__", 0);
|
|
__pyx_pybuffer_n_classes.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_n_classes.refcount = 0;
|
|
__pyx_pybuffernd_n_classes.data = NULL;
|
|
__pyx_pybuffernd_n_classes.rcbuffer = &__pyx_pybuffer_n_classes;
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_n_classes.rcbuffer->pybuffer, (PyObject*)__pyx_v_n_classes, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_SIZE_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3050; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_pybuffernd_n_classes.diminfo[0].strides = __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_n_classes.diminfo[0].shape = __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.shape[0];
|
|
|
|
/* "sklearn/tree/_tree.pyx":3054
|
|
* """Constructor."""
|
|
* # Input/Output layout
|
|
* self.n_features = n_features # <<<<<<<<<<<<<<
|
|
* self.n_outputs = n_outputs
|
|
* self.n_classes = NULL
|
|
*/
|
|
__pyx_v_self->n_features = __pyx_v_n_features;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3055
|
|
* # Input/Output layout
|
|
* self.n_features = n_features
|
|
* self.n_outputs = n_outputs # <<<<<<<<<<<<<<
|
|
* self.n_classes = NULL
|
|
* safe_realloc(&self.n_classes, n_outputs)
|
|
*/
|
|
__pyx_v_self->n_outputs = __pyx_v_n_outputs;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3056
|
|
* self.n_features = n_features
|
|
* self.n_outputs = n_outputs
|
|
* self.n_classes = NULL # <<<<<<<<<<<<<<
|
|
* safe_realloc(&self.n_classes, n_outputs)
|
|
*
|
|
*/
|
|
__pyx_v_self->n_classes = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3057
|
|
* self.n_outputs = n_outputs
|
|
* self.n_classes = NULL
|
|
* safe_realloc(&self.n_classes, n_outputs) # <<<<<<<<<<<<<<
|
|
*
|
|
* self.max_n_classes = np.max(n_classes)
|
|
*/
|
|
__pyx_t_1 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_self->n_classes), __pyx_v_n_outputs); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3057; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3059
|
|
* safe_realloc(&self.n_classes, n_outputs)
|
|
*
|
|
* self.max_n_classes = np.max(n_classes) # <<<<<<<<<<<<<<
|
|
* self.value_stride = n_outputs * self.max_n_classes
|
|
*
|
|
*/
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_max); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_n_classes));
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_v_n_classes));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_n_classes));
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_2, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_4); if (unlikely((__pyx_t_5 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3059; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_v_self->max_n_classes = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3060
|
|
*
|
|
* self.max_n_classes = np.max(n_classes)
|
|
* self.value_stride = n_outputs * self.max_n_classes # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t k
|
|
*/
|
|
__pyx_v_self->value_stride = (__pyx_v_n_outputs * __pyx_v_self->max_n_classes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3063
|
|
*
|
|
* cdef SIZE_t k
|
|
* for k in range(n_outputs): # <<<<<<<<<<<<<<
|
|
* self.n_classes[k] = n_classes[k]
|
|
*
|
|
*/
|
|
__pyx_t_6 = __pyx_v_n_outputs;
|
|
for (__pyx_t_5 = 0; __pyx_t_5 < __pyx_t_6; __pyx_t_5+=1) {
|
|
__pyx_v_k = __pyx_t_5;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3064
|
|
* cdef SIZE_t k
|
|
* for k in range(n_outputs):
|
|
* self.n_classes[k] = n_classes[k] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Inner structures
|
|
*/
|
|
__pyx_t_7 = __pyx_v_k;
|
|
(__pyx_v_self->n_classes[__pyx_v_k]) = (*__Pyx_BufPtrStrided1d(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_pybuffernd_n_classes.rcbuffer->pybuffer.buf, __pyx_t_7, __pyx_pybuffernd_n_classes.diminfo[0].strides));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3067
|
|
*
|
|
* # Inner structures
|
|
* self.max_depth = 0 # <<<<<<<<<<<<<<
|
|
* self.node_count = 0
|
|
* self.capacity = 0
|
|
*/
|
|
__pyx_v_self->max_depth = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3068
|
|
* # Inner structures
|
|
* self.max_depth = 0
|
|
* self.node_count = 0 # <<<<<<<<<<<<<<
|
|
* self.capacity = 0
|
|
* self.value = NULL
|
|
*/
|
|
__pyx_v_self->node_count = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3069
|
|
* self.max_depth = 0
|
|
* self.node_count = 0
|
|
* self.capacity = 0 # <<<<<<<<<<<<<<
|
|
* self.value = NULL
|
|
* self.nodes = NULL
|
|
*/
|
|
__pyx_v_self->capacity = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3070
|
|
* self.node_count = 0
|
|
* self.capacity = 0
|
|
* self.value = NULL # <<<<<<<<<<<<<<
|
|
* self.nodes = NULL
|
|
*
|
|
*/
|
|
__pyx_v_self->value = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3071
|
|
* self.capacity = 0
|
|
* self.value = NULL
|
|
* self.nodes = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* def __dealloc__(self):
|
|
*/
|
|
__pyx_v_self->nodes = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3050
|
|
* return self._get_value_ndarray()[:self.node_count]
|
|
*
|
|
* def __cinit__(self, int n_features, np.ndarray[SIZE_t, ndim=1] n_classes, # <<<<<<<<<<<<<<
|
|
* int n_outputs):
|
|
* """Constructor."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_n_classes.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.__cinit__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_n_classes.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3073
|
|
* self.nodes = NULL
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* # Free all inner structures
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_4Tree_3__dealloc__(PyObject *__pyx_v_self); /*proto*/
|
|
static void __pyx_pw_7sklearn_4tree_5_tree_4Tree_3__dealloc__(PyObject *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__ (wrapper)", 0);
|
|
__pyx_pf_7sklearn_4tree_5_tree_4Tree_2__dealloc__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_7sklearn_4tree_5_tree_4Tree_2__dealloc__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__dealloc__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3076
|
|
* """Destructor."""
|
|
* # Free all inner structures
|
|
* free(self.n_classes) # <<<<<<<<<<<<<<
|
|
* free(self.value)
|
|
* free(self.nodes)
|
|
*/
|
|
free(__pyx_v_self->n_classes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3077
|
|
* # Free all inner structures
|
|
* free(self.n_classes)
|
|
* free(self.value) # <<<<<<<<<<<<<<
|
|
* free(self.nodes)
|
|
*
|
|
*/
|
|
free(__pyx_v_self->value);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3078
|
|
* free(self.n_classes)
|
|
* free(self.value)
|
|
* free(self.nodes) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __reduce__(self):
|
|
*/
|
|
free(__pyx_v_self->nodes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3073
|
|
* self.nodes = NULL
|
|
*
|
|
* def __dealloc__(self): # <<<<<<<<<<<<<<
|
|
* """Destructor."""
|
|
* # Free all inner structures
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3080
|
|
* free(self.nodes)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* """Reduce re-implementation, for pickling."""
|
|
* return (Tree, (self.n_features,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_4Tree_4__reduce__[] = "Reduce re-implementation, for pickling.";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_5__reduce__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__reduce__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_4__reduce__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_4__reduce__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__reduce__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3082
|
|
* def __reduce__(self):
|
|
* """Reduce re-implementation, for pickling."""
|
|
* return (Tree, (self.n_features, # <<<<<<<<<<<<<<
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs),
|
|
* self.n_outputs), self.__getstate__())
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->n_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3082; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3083
|
|
* """Reduce re-implementation, for pickling."""
|
|
* return (Tree, (self.n_features,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs), # <<<<<<<<<<<<<<
|
|
* self.n_outputs), self.__getstate__())
|
|
*
|
|
*/
|
|
__pyx_t_2 = ((PyObject *)__pyx_f_7sklearn_4tree_5_tree_sizet_ptr_to_ndarray(__pyx_v_self->n_classes, __pyx_v_self->n_outputs)); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3083; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3084
|
|
* return (Tree, (self.n_features,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs),
|
|
* self.n_outputs), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __getstate__(self):
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->n_outputs); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3084; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3082
|
|
* def __reduce__(self):
|
|
* """Reduce re-implementation, for pickling."""
|
|
* return (Tree, (self.n_features, # <<<<<<<<<<<<<<
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs),
|
|
* self.n_outputs), self.__getstate__())
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(3); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3082; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 2, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3084
|
|
* return (Tree, (self.n_features,
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs),
|
|
* self.n_outputs), self.__getstate__()) # <<<<<<<<<<<<<<
|
|
*
|
|
* def __getstate__(self):
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_getstate); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3084; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_empty_tuple, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3084; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3082
|
|
* def __reduce__(self):
|
|
* """Reduce re-implementation, for pickling."""
|
|
* return (Tree, (self.n_features, # <<<<<<<<<<<<<<
|
|
* sizet_ptr_to_ndarray(self.n_classes, self.n_outputs),
|
|
* self.n_outputs), self.__getstate__())
|
|
*/
|
|
__pyx_t_3 = PyTuple_New(3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3082; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_Tree)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_Tree)));
|
|
__Pyx_GIVEREF(((PyObject *)((PyObject*)__pyx_ptype_7sklearn_4tree_5_tree_Tree)));
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 2, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3080
|
|
* free(self.nodes)
|
|
*
|
|
* def __reduce__(self): # <<<<<<<<<<<<<<
|
|
* """Reduce re-implementation, for pickling."""
|
|
* return (Tree, (self.n_features,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.__reduce__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3086
|
|
* self.n_outputs), self.__getstate__())
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* """Getstate re-implementation, for pickling."""
|
|
* d = {}
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_7__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_4Tree_6__getstate__[] = "Getstate re-implementation, for pickling.";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_7__getstate__(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__getstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_6__getstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_6__getstate__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
PyObject *__pyx_v_d = NULL;
|
|
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("__getstate__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3088
|
|
* def __getstate__(self):
|
|
* """Getstate re-implementation, for pickling."""
|
|
* d = {} # <<<<<<<<<<<<<<
|
|
* d["node_count"] = self.node_count
|
|
* d["nodes"] = self._get_node_ndarray()
|
|
*/
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3088; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_v_d = ((PyObject*)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3089
|
|
* """Getstate re-implementation, for pickling."""
|
|
* d = {}
|
|
* d["node_count"] = self.node_count # <<<<<<<<<<<<<<
|
|
* d["nodes"] = self._get_node_ndarray()
|
|
* d["values"] = self._get_value_ndarray()
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->node_count); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3089; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (unlikely(PyDict_SetItem(__pyx_v_d, __pyx_n_s_node_count, __pyx_t_1) < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3089; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3090
|
|
* d = {}
|
|
* d["node_count"] = self.node_count
|
|
* d["nodes"] = self._get_node_ndarray() # <<<<<<<<<<<<<<
|
|
* d["values"] = self._get_value_ndarray()
|
|
* return d
|
|
*/
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_node_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3090; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (unlikely(PyDict_SetItem(__pyx_v_d, __pyx_n_s_nodes, __pyx_t_1) < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3090; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3091
|
|
* d["node_count"] = self.node_count
|
|
* d["nodes"] = self._get_node_ndarray()
|
|
* d["values"] = self._get_value_ndarray() # <<<<<<<<<<<<<<
|
|
* return d
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_value_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3091; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (unlikely(PyDict_SetItem(__pyx_v_d, __pyx_n_s_values, __pyx_t_1) < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3091; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3092
|
|
* d["nodes"] = self._get_node_ndarray()
|
|
* d["values"] = self._get_value_ndarray()
|
|
* return d # <<<<<<<<<<<<<<
|
|
*
|
|
* def __setstate__(self, d):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(__pyx_v_d);
|
|
__pyx_r = __pyx_v_d;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3086
|
|
* self.n_outputs), self.__getstate__())
|
|
*
|
|
* def __getstate__(self): # <<<<<<<<<<<<<<
|
|
* """Getstate re-implementation, for pickling."""
|
|
* d = {}
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.__getstate__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_d);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3094
|
|
* return d
|
|
*
|
|
* def __setstate__(self, d): # <<<<<<<<<<<<<<
|
|
* """Setstate re-implementation, for unpickling."""
|
|
* self.node_count = d["node_count"]
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_4Tree_8__setstate__[] = "Setstate re-implementation, for unpickling.";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9__setstate__(PyObject *__pyx_v_self, PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__setstate__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_8__setstate__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_d));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_8__setstate__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_d) {
|
|
PyObject *__pyx_v_node_ndarray = NULL;
|
|
PyObject *__pyx_v_value_ndarray = NULL;
|
|
PyObject *__pyx_v_value_shape = NULL;
|
|
CYTHON_UNUSED void *__pyx_v_nodes;
|
|
CYTHON_UNUSED void *__pyx_v_value;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_2;
|
|
int __pyx_t_3;
|
|
int __pyx_t_4;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyObject *__pyx_t_6 = NULL;
|
|
PyObject *__pyx_t_7 = NULL;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
int __pyx_t_10;
|
|
int __pyx_t_11;
|
|
int __pyx_t_12;
|
|
int __pyx_t_13;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c __pyx_t_14;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__setstate__", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3096
|
|
* def __setstate__(self, d):
|
|
* """Setstate re-implementation, for unpickling."""
|
|
* self.node_count = d["node_count"] # <<<<<<<<<<<<<<
|
|
*
|
|
* if 'nodes' not in d:
|
|
*/
|
|
__pyx_t_1 = PyObject_GetItem(__pyx_v_d, __pyx_n_s_node_count); if (unlikely(__pyx_t_1 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3096; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_1); if (unlikely((__pyx_t_2 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3096; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_v_self->node_count = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3098
|
|
* self.node_count = d["node_count"]
|
|
*
|
|
* if 'nodes' not in d: # <<<<<<<<<<<<<<
|
|
* raise ValueError('You have loaded Tree version which '
|
|
* 'cannot be imported')
|
|
*/
|
|
__pyx_t_3 = (__Pyx_PySequence_Contains(__pyx_n_s_nodes, __pyx_v_d, Py_NE)); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3098; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_4 = (__pyx_t_3 != 0);
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3099
|
|
*
|
|
* if 'nodes' not in d:
|
|
* raise ValueError('You have loaded Tree version which ' # <<<<<<<<<<<<<<
|
|
* 'cannot be imported')
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__3, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3099; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_Raise(__pyx_t_1, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3099; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3102
|
|
* 'cannot be imported')
|
|
*
|
|
* node_ndarray = d['nodes'] # <<<<<<<<<<<<<<
|
|
* value_ndarray = d['values']
|
|
*
|
|
*/
|
|
__pyx_t_1 = PyObject_GetItem(__pyx_v_d, __pyx_n_s_nodes); if (unlikely(__pyx_t_1 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3102; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_v_node_ndarray = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3103
|
|
*
|
|
* node_ndarray = d['nodes']
|
|
* value_ndarray = d['values'] # <<<<<<<<<<<<<<
|
|
*
|
|
* value_shape = (node_ndarray.shape[0], self.n_outputs,
|
|
*/
|
|
__pyx_t_1 = PyObject_GetItem(__pyx_v_d, __pyx_n_s_values); if (unlikely(__pyx_t_1 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3103; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_v_value_ndarray = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3105
|
|
* value_ndarray = d['values']
|
|
*
|
|
* value_shape = (node_ndarray.shape[0], self.n_outputs, # <<<<<<<<<<<<<<
|
|
* self.max_n_classes)
|
|
* if (node_ndarray.ndim != 1 or
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_node_ndarray, __pyx_n_s_shape); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3105; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_5 = __Pyx_GetItemInt(__pyx_t_1, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_5 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3105; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->n_outputs); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3105; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3106
|
|
*
|
|
* value_shape = (node_ndarray.shape[0], self.n_outputs,
|
|
* self.max_n_classes) # <<<<<<<<<<<<<<
|
|
* if (node_ndarray.ndim != 1 or
|
|
* node_ndarray.dtype != NODE_DTYPE or
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->max_n_classes); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3106; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3105
|
|
* value_ndarray = d['values']
|
|
*
|
|
* value_shape = (node_ndarray.shape[0], self.n_outputs, # <<<<<<<<<<<<<<
|
|
* self.max_n_classes)
|
|
* if (node_ndarray.ndim != 1 or
|
|
*/
|
|
__pyx_t_7 = PyTuple_New(3); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3105; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_5);
|
|
__Pyx_GIVEREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_7, 2, __pyx_t_6);
|
|
__Pyx_GIVEREF(__pyx_t_6);
|
|
__pyx_t_5 = 0;
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_6 = 0;
|
|
__pyx_v_value_shape = ((PyObject*)__pyx_t_7);
|
|
__pyx_t_7 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3107
|
|
* value_shape = (node_ndarray.shape[0], self.n_outputs,
|
|
* self.max_n_classes)
|
|
* if (node_ndarray.ndim != 1 or # <<<<<<<<<<<<<<
|
|
* node_ndarray.dtype != NODE_DTYPE or
|
|
* not node_ndarray.flags.c_contiguous or
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_node_ndarray, __pyx_n_s_ndim); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3107; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_6 = PyObject_RichCompare(__pyx_t_7, __pyx_int_1, Py_NE); __Pyx_XGOTREF(__pyx_t_6); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3107; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_6); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3107; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
if (!__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3108
|
|
* self.max_n_classes)
|
|
* if (node_ndarray.ndim != 1 or
|
|
* node_ndarray.dtype != NODE_DTYPE or # <<<<<<<<<<<<<<
|
|
* not node_ndarray.flags.c_contiguous or
|
|
* value_ndarray.shape != value_shape or
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyObject_GetAttrStr(__pyx_v_node_ndarray, __pyx_n_s_dtype); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3108; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_t_7 = __Pyx_GetModuleGlobalName(__pyx_n_s_NODE_DTYPE); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3108; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_1 = PyObject_RichCompare(__pyx_t_6, __pyx_t_7, Py_NE); __Pyx_XGOTREF(__pyx_t_1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3108; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3108; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (!__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3109
|
|
* if (node_ndarray.ndim != 1 or
|
|
* node_ndarray.dtype != NODE_DTYPE or
|
|
* not node_ndarray.flags.c_contiguous or # <<<<<<<<<<<<<<
|
|
* value_ndarray.shape != value_shape or
|
|
* not value_ndarray.flags.c_contiguous or
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_node_ndarray, __pyx_n_s_flags); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_c_contiguous); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_8 = __Pyx_PyObject_IsTrue(__pyx_t_7); if (unlikely(__pyx_t_8 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_9 = (!__pyx_t_8);
|
|
if (!__pyx_t_9) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3110
|
|
* node_ndarray.dtype != NODE_DTYPE or
|
|
* not node_ndarray.flags.c_contiguous or
|
|
* value_ndarray.shape != value_shape or # <<<<<<<<<<<<<<
|
|
* not value_ndarray.flags.c_contiguous or
|
|
* value_ndarray.dtype != np.float64):
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_value_ndarray, __pyx_n_s_shape); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3110; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_1 = PyObject_RichCompare(__pyx_t_7, __pyx_v_value_shape, Py_NE); __Pyx_XGOTREF(__pyx_t_1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3110; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_8 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_8 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3110; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (!__pyx_t_8) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3111
|
|
* not node_ndarray.flags.c_contiguous or
|
|
* value_ndarray.shape != value_shape or
|
|
* not value_ndarray.flags.c_contiguous or # <<<<<<<<<<<<<<
|
|
* value_ndarray.dtype != np.float64):
|
|
* raise ValueError('Did not recognise loaded array layout')
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_value_ndarray, __pyx_n_s_flags); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_c_contiguous); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_10 = __Pyx_PyObject_IsTrue(__pyx_t_7); if (unlikely(__pyx_t_10 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__pyx_t_11 = (!__pyx_t_10);
|
|
if (!__pyx_t_11) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3112
|
|
* value_ndarray.shape != value_shape or
|
|
* not value_ndarray.flags.c_contiguous or
|
|
* value_ndarray.dtype != np.float64): # <<<<<<<<<<<<<<
|
|
* raise ValueError('Did not recognise loaded array layout')
|
|
*
|
|
*/
|
|
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_v_value_ndarray, __pyx_n_s_dtype); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3112; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3112; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_6 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_float64); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3112; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyObject_RichCompare(__pyx_t_7, __pyx_t_6, Py_NE); __Pyx_XGOTREF(__pyx_t_1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3112; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__pyx_t_10 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_10 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3112; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_12 = __pyx_t_10;
|
|
} else {
|
|
__pyx_t_12 = __pyx_t_11;
|
|
}
|
|
__pyx_t_11 = __pyx_t_12;
|
|
} else {
|
|
__pyx_t_11 = __pyx_t_8;
|
|
}
|
|
__pyx_t_8 = __pyx_t_11;
|
|
} else {
|
|
__pyx_t_8 = __pyx_t_9;
|
|
}
|
|
__pyx_t_9 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_3;
|
|
}
|
|
__pyx_t_3 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_4;
|
|
}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3113
|
|
* not value_ndarray.flags.c_contiguous or
|
|
* value_ndarray.dtype != np.float64):
|
|
* raise ValueError('Did not recognise loaded array layout') # <<<<<<<<<<<<<<
|
|
*
|
|
* self.capacity = node_ndarray.shape[0]
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__4, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_Raise(__pyx_t_1, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3115
|
|
* raise ValueError('Did not recognise loaded array layout')
|
|
*
|
|
* self.capacity = node_ndarray.shape[0] # <<<<<<<<<<<<<<
|
|
* if self._resize_c(self.capacity) != 0:
|
|
* raise MemoryError("resizing tree to %d" % self.capacity)
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_node_ndarray, __pyx_n_s_shape); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3115; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_6 = __Pyx_GetItemInt(__pyx_t_1, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_6 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3115; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_2 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_6); if (unlikely((__pyx_t_2 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3115; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__pyx_v_self->capacity = __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3116
|
|
*
|
|
* self.capacity = node_ndarray.shape[0]
|
|
* if self._resize_c(self.capacity) != 0: # <<<<<<<<<<<<<<
|
|
* raise MemoryError("resizing tree to %d" % self.capacity)
|
|
* nodes = memcpy(self.nodes, (<np.ndarray> node_ndarray).data,
|
|
*/
|
|
__pyx_t_14.__pyx_n = 1;
|
|
__pyx_t_14.capacity = __pyx_v_self->capacity;
|
|
__pyx_t_13 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_resize_c(__pyx_v_self, &__pyx_t_14);
|
|
__pyx_t_3 = ((__pyx_t_13 != 0) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3117
|
|
* self.capacity = node_ndarray.shape[0]
|
|
* if self._resize_c(self.capacity) != 0:
|
|
* raise MemoryError("resizing tree to %d" % self.capacity) # <<<<<<<<<<<<<<
|
|
* nodes = memcpy(self.nodes, (<np.ndarray> node_ndarray).data,
|
|
* self.capacity * sizeof(Node))
|
|
*/
|
|
__pyx_t_6 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->capacity); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3117; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__pyx_t_1 = __Pyx_PyString_Format(__pyx_kp_s_resizing_tree_to_d, __pyx_t_6); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3117; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3117; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_6, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3117; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
|
|
__Pyx_Raise(__pyx_t_1, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3117; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3118
|
|
* if self._resize_c(self.capacity) != 0:
|
|
* raise MemoryError("resizing tree to %d" % self.capacity)
|
|
* nodes = memcpy(self.nodes, (<np.ndarray> node_ndarray).data, # <<<<<<<<<<<<<<
|
|
* self.capacity * sizeof(Node))
|
|
* value = memcpy(self.value, (<np.ndarray> value_ndarray).data,
|
|
*/
|
|
__pyx_v_nodes = memcpy(__pyx_v_self->nodes, ((PyArrayObject *)__pyx_v_node_ndarray)->data, (__pyx_v_self->capacity * (sizeof(struct __pyx_t_7sklearn_4tree_5_tree_Node))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3120
|
|
* nodes = memcpy(self.nodes, (<np.ndarray> node_ndarray).data,
|
|
* self.capacity * sizeof(Node))
|
|
* value = memcpy(self.value, (<np.ndarray> value_ndarray).data, # <<<<<<<<<<<<<<
|
|
* self.capacity * self.value_stride * sizeof(double))
|
|
*
|
|
*/
|
|
__pyx_v_value = memcpy(__pyx_v_self->value, ((PyArrayObject *)__pyx_v_value_ndarray)->data, ((__pyx_v_self->capacity * __pyx_v_self->value_stride) * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3094
|
|
* return d
|
|
*
|
|
* def __setstate__(self, d): # <<<<<<<<<<<<<<
|
|
* """Setstate re-implementation, for unpickling."""
|
|
* self.node_count = d["node_count"]
|
|
*/
|
|
|
|
/* 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_5);
|
|
__Pyx_XDECREF(__pyx_t_6);
|
|
__Pyx_XDECREF(__pyx_t_7);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.__setstate__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_node_ndarray);
|
|
__Pyx_XDECREF(__pyx_v_value_ndarray);
|
|
__Pyx_XDECREF(__pyx_v_value_shape);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3123
|
|
* self.capacity * self.value_stride * sizeof(double))
|
|
*
|
|
* cdef void _resize(self, SIZE_t capacity) except *: # <<<<<<<<<<<<<<
|
|
* """Resize all inner arrays to `capacity`, if `capacity` == -1, then
|
|
* double the size of the inner arrays."""
|
|
*/
|
|
|
|
static void __pyx_f_7sklearn_4tree_5_tree_4Tree__resize(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_capacity) {
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c __pyx_t_2;
|
|
int __pyx_t_3;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_resize", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3126
|
|
* """Resize all inner arrays to `capacity`, if `capacity` == -1, then
|
|
* double the size of the inner arrays."""
|
|
* if self._resize_c(capacity) != 0: # <<<<<<<<<<<<<<
|
|
* raise MemoryError()
|
|
*
|
|
*/
|
|
__pyx_t_2.__pyx_n = 1;
|
|
__pyx_t_2.capacity = __pyx_v_capacity;
|
|
__pyx_t_1 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_resize_c(__pyx_v_self, &__pyx_t_2);
|
|
__pyx_t_3 = ((__pyx_t_1 != 0) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3127
|
|
* double the size of the inner arrays."""
|
|
* if self._resize_c(capacity) != 0:
|
|
* raise MemoryError() # <<<<<<<<<<<<<<
|
|
*
|
|
* # XXX using (size_t)(-1) is ugly, but SIZE_MAX is not available in C89
|
|
*/
|
|
PyErr_NoMemory(); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3127; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3123
|
|
* self.capacity * self.value_stride * sizeof(double))
|
|
*
|
|
* cdef void _resize(self, SIZE_t capacity) except *: # <<<<<<<<<<<<<<
|
|
* """Resize all inner arrays to `capacity`, if `capacity` == -1, then
|
|
* double the size of the inner arrays."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree._resize", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3131
|
|
* # XXX using (size_t)(-1) is ugly, but SIZE_MAX is not available in C89
|
|
* # (i.e., older MSVC).
|
|
* cdef int _resize_c(self, SIZE_t capacity=<SIZE_t>(-1)) nogil: # <<<<<<<<<<<<<<
|
|
* """Guts of _resize. Returns 0 for success, -1 for error."""
|
|
* if capacity == self.capacity and self.nodes != NULL:
|
|
*/
|
|
|
|
static int __pyx_f_7sklearn_4tree_5_tree_4Tree__resize_c(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c *__pyx_optional_args) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_capacity = __pyx_k__5;
|
|
void *__pyx_v_ptr;
|
|
int __pyx_r;
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
if (__pyx_optional_args) {
|
|
if (__pyx_optional_args->__pyx_n > 0) {
|
|
__pyx_v_capacity = __pyx_optional_args->capacity;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3133
|
|
* cdef int _resize_c(self, SIZE_t capacity=<SIZE_t>(-1)) nogil:
|
|
* """Guts of _resize. Returns 0 for success, -1 for error."""
|
|
* if capacity == self.capacity and self.nodes != NULL: # <<<<<<<<<<<<<<
|
|
* return 0
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_capacity == __pyx_v_self->capacity) != 0);
|
|
if (__pyx_t_1) {
|
|
__pyx_t_2 = ((__pyx_v_self->nodes != NULL) != 0);
|
|
__pyx_t_3 = __pyx_t_2;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_1;
|
|
}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3134
|
|
* """Guts of _resize. Returns 0 for success, -1 for error."""
|
|
* if capacity == self.capacity and self.nodes != NULL:
|
|
* return 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* if capacity == <SIZE_t>(-1):
|
|
*/
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3136
|
|
* return 0
|
|
*
|
|
* if capacity == <SIZE_t>(-1): # <<<<<<<<<<<<<<
|
|
* if self.capacity == 0:
|
|
* capacity = 3 # default initial value
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_capacity == ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)-1)) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3137
|
|
*
|
|
* if capacity == <SIZE_t>(-1):
|
|
* if self.capacity == 0: # <<<<<<<<<<<<<<
|
|
* capacity = 3 # default initial value
|
|
* else:
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_self->capacity == 0) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3138
|
|
* if capacity == <SIZE_t>(-1):
|
|
* if self.capacity == 0:
|
|
* capacity = 3 # default initial value # <<<<<<<<<<<<<<
|
|
* else:
|
|
* capacity = 2 * self.capacity
|
|
*/
|
|
__pyx_v_capacity = 3;
|
|
goto __pyx_L5;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3140
|
|
* capacity = 3 # default initial value
|
|
* else:
|
|
* capacity = 2 * self.capacity # <<<<<<<<<<<<<<
|
|
*
|
|
* # XXX no safe_realloc here because we need to grab the GIL
|
|
*/
|
|
__pyx_v_capacity = (2 * __pyx_v_self->capacity);
|
|
}
|
|
__pyx_L5:;
|
|
goto __pyx_L4;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3143
|
|
*
|
|
* # XXX no safe_realloc here because we need to grab the GIL
|
|
* cdef void* ptr = realloc(self.nodes, capacity * sizeof(Node)) # <<<<<<<<<<<<<<
|
|
* if ptr == NULL:
|
|
* return -1
|
|
*/
|
|
__pyx_v_ptr = realloc(__pyx_v_self->nodes, (__pyx_v_capacity * (sizeof(struct __pyx_t_7sklearn_4tree_5_tree_Node))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3144
|
|
* # XXX no safe_realloc here because we need to grab the GIL
|
|
* cdef void* ptr = realloc(self.nodes, capacity * sizeof(Node))
|
|
* if ptr == NULL: # <<<<<<<<<<<<<<
|
|
* return -1
|
|
* self.nodes = <Node*> ptr
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_ptr == NULL) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3145
|
|
* cdef void* ptr = realloc(self.nodes, capacity * sizeof(Node))
|
|
* if ptr == NULL:
|
|
* return -1 # <<<<<<<<<<<<<<
|
|
* self.nodes = <Node*> ptr
|
|
* ptr = realloc(self.value,
|
|
*/
|
|
__pyx_r = -1;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3146
|
|
* if ptr == NULL:
|
|
* return -1
|
|
* self.nodes = <Node*> ptr # <<<<<<<<<<<<<<
|
|
* ptr = realloc(self.value,
|
|
* capacity * self.value_stride * sizeof(double))
|
|
*/
|
|
__pyx_v_self->nodes = ((struct __pyx_t_7sklearn_4tree_5_tree_Node *)__pyx_v_ptr);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3147
|
|
* return -1
|
|
* self.nodes = <Node*> ptr
|
|
* ptr = realloc(self.value, # <<<<<<<<<<<<<<
|
|
* capacity * self.value_stride * sizeof(double))
|
|
* if ptr == NULL:
|
|
*/
|
|
__pyx_v_ptr = realloc(__pyx_v_self->value, ((__pyx_v_capacity * __pyx_v_self->value_stride) * (sizeof(double))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3149
|
|
* ptr = realloc(self.value,
|
|
* capacity * self.value_stride * sizeof(double))
|
|
* if ptr == NULL: # <<<<<<<<<<<<<<
|
|
* return -1
|
|
* self.value = <double*> ptr
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_ptr == NULL) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3150
|
|
* capacity * self.value_stride * sizeof(double))
|
|
* if ptr == NULL:
|
|
* return -1 # <<<<<<<<<<<<<<
|
|
* self.value = <double*> ptr
|
|
*
|
|
*/
|
|
__pyx_r = -1;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3151
|
|
* if ptr == NULL:
|
|
* return -1
|
|
* self.value = <double*> ptr # <<<<<<<<<<<<<<
|
|
*
|
|
* # value memory is initialised to 0 to enable classifier argmax
|
|
*/
|
|
__pyx_v_self->value = ((double *)__pyx_v_ptr);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3154
|
|
*
|
|
* # value memory is initialised to 0 to enable classifier argmax
|
|
* if capacity > self.capacity: # <<<<<<<<<<<<<<
|
|
* memset(<void*>(self.value + self.capacity * self.value_stride), 0,
|
|
* (capacity - self.capacity) * self.value_stride *
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_capacity > __pyx_v_self->capacity) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3155
|
|
* # value memory is initialised to 0 to enable classifier argmax
|
|
* if capacity > self.capacity:
|
|
* memset(<void*>(self.value + self.capacity * self.value_stride), 0, # <<<<<<<<<<<<<<
|
|
* (capacity - self.capacity) * self.value_stride *
|
|
* sizeof(double))
|
|
*/
|
|
memset(((void *)(__pyx_v_self->value + (__pyx_v_self->capacity * __pyx_v_self->value_stride))), 0, (((__pyx_v_capacity - __pyx_v_self->capacity) * __pyx_v_self->value_stride) * (sizeof(double))));
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3160
|
|
*
|
|
* # if capacity smaller than node_count, adjust the counter
|
|
* if capacity < self.node_count: # <<<<<<<<<<<<<<
|
|
* self.node_count = capacity
|
|
*
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_capacity < __pyx_v_self->node_count) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3161
|
|
* # if capacity smaller than node_count, adjust the counter
|
|
* if capacity < self.node_count:
|
|
* self.node_count = capacity # <<<<<<<<<<<<<<
|
|
*
|
|
* self.capacity = capacity
|
|
*/
|
|
__pyx_v_self->node_count = __pyx_v_capacity;
|
|
goto __pyx_L9;
|
|
}
|
|
__pyx_L9:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3163
|
|
* self.node_count = capacity
|
|
*
|
|
* self.capacity = capacity # <<<<<<<<<<<<<<
|
|
* return 0
|
|
*
|
|
*/
|
|
__pyx_v_self->capacity = __pyx_v_capacity;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3164
|
|
*
|
|
* self.capacity = capacity
|
|
* return 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t _add_node(self, SIZE_t parent, bint is_left, bint is_leaf,
|
|
*/
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3131
|
|
* # XXX using (size_t)(-1) is ugly, but SIZE_MAX is not available in C89
|
|
* # (i.e., older MSVC).
|
|
* cdef int _resize_c(self, SIZE_t capacity=<SIZE_t>(-1)) nogil: # <<<<<<<<<<<<<<
|
|
* """Guts of _resize. Returns 0 for success, -1 for error."""
|
|
* if capacity == self.capacity and self.nodes != NULL:
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3166
|
|
* return 0
|
|
*
|
|
* cdef SIZE_t _add_node(self, SIZE_t parent, bint is_left, bint is_leaf, # <<<<<<<<<<<<<<
|
|
* SIZE_t feature, double threshold, double impurity,
|
|
* SIZE_t n_node_samples, double weighted_n_node_samples) nogil:
|
|
*/
|
|
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_f_7sklearn_4tree_5_tree_4Tree__add_node(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_parent, int __pyx_v_is_left, int __pyx_v_is_leaf, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_feature, double __pyx_v_threshold, double __pyx_v_impurity, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_node_samples, double __pyx_v_weighted_n_node_samples) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_node_id;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_node;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_r;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_t_2;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3175
|
|
* Returns (size_t)(-1) on error.
|
|
* """
|
|
* cdef SIZE_t node_id = self.node_count # <<<<<<<<<<<<<<
|
|
*
|
|
* if node_id >= self.capacity:
|
|
*/
|
|
__pyx_t_1 = __pyx_v_self->node_count;
|
|
__pyx_v_node_id = __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3177
|
|
* cdef SIZE_t node_id = self.node_count
|
|
*
|
|
* if node_id >= self.capacity: # <<<<<<<<<<<<<<
|
|
* if self._resize_c() != 0:
|
|
* return <SIZE_t>(-1)
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_node_id >= __pyx_v_self->capacity) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3178
|
|
*
|
|
* if node_id >= self.capacity:
|
|
* if self._resize_c() != 0: # <<<<<<<<<<<<<<
|
|
* return <SIZE_t>(-1)
|
|
*
|
|
*/
|
|
__pyx_t_2 = ((((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_resize_c(__pyx_v_self, NULL) != 0) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3179
|
|
* if node_id >= self.capacity:
|
|
* if self._resize_c() != 0:
|
|
* return <SIZE_t>(-1) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef Node* node = &self.nodes[node_id]
|
|
*/
|
|
__pyx_r = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)-1);
|
|
goto __pyx_L0;
|
|
}
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3181
|
|
* return <SIZE_t>(-1)
|
|
*
|
|
* cdef Node* node = &self.nodes[node_id] # <<<<<<<<<<<<<<
|
|
* node.impurity = impurity
|
|
* node.n_node_samples = n_node_samples
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_self->nodes[__pyx_v_node_id]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3182
|
|
*
|
|
* cdef Node* node = &self.nodes[node_id]
|
|
* node.impurity = impurity # <<<<<<<<<<<<<<
|
|
* node.n_node_samples = n_node_samples
|
|
* node.weighted_n_node_samples = weighted_n_node_samples
|
|
*/
|
|
__pyx_v_node->impurity = __pyx_v_impurity;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3183
|
|
* cdef Node* node = &self.nodes[node_id]
|
|
* node.impurity = impurity
|
|
* node.n_node_samples = n_node_samples # <<<<<<<<<<<<<<
|
|
* node.weighted_n_node_samples = weighted_n_node_samples
|
|
*
|
|
*/
|
|
__pyx_v_node->n_node_samples = __pyx_v_n_node_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3184
|
|
* node.impurity = impurity
|
|
* node.n_node_samples = n_node_samples
|
|
* node.weighted_n_node_samples = weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* if parent != _TREE_UNDEFINED:
|
|
*/
|
|
__pyx_v_node->weighted_n_node_samples = __pyx_v_weighted_n_node_samples;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3186
|
|
* node.weighted_n_node_samples = weighted_n_node_samples
|
|
*
|
|
* if parent != _TREE_UNDEFINED: # <<<<<<<<<<<<<<
|
|
* if is_left:
|
|
* self.nodes[parent].left_child = node_id
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_parent != __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3187
|
|
*
|
|
* if parent != _TREE_UNDEFINED:
|
|
* if is_left: # <<<<<<<<<<<<<<
|
|
* self.nodes[parent].left_child = node_id
|
|
* else:
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_is_left != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3188
|
|
* if parent != _TREE_UNDEFINED:
|
|
* if is_left:
|
|
* self.nodes[parent].left_child = node_id # <<<<<<<<<<<<<<
|
|
* else:
|
|
* self.nodes[parent].right_child = node_id
|
|
*/
|
|
(__pyx_v_self->nodes[__pyx_v_parent]).left_child = __pyx_v_node_id;
|
|
goto __pyx_L6;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3190
|
|
* self.nodes[parent].left_child = node_id
|
|
* else:
|
|
* self.nodes[parent].right_child = node_id # <<<<<<<<<<<<<<
|
|
*
|
|
* if is_leaf:
|
|
*/
|
|
(__pyx_v_self->nodes[__pyx_v_parent]).right_child = __pyx_v_node_id;
|
|
}
|
|
__pyx_L6:;
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3192
|
|
* self.nodes[parent].right_child = node_id
|
|
*
|
|
* if is_leaf: # <<<<<<<<<<<<<<
|
|
* node.left_child = _TREE_LEAF
|
|
* node.right_child = _TREE_LEAF
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_is_leaf != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3193
|
|
*
|
|
* if is_leaf:
|
|
* node.left_child = _TREE_LEAF # <<<<<<<<<<<<<<
|
|
* node.right_child = _TREE_LEAF
|
|
* node.feature = _TREE_UNDEFINED
|
|
*/
|
|
__pyx_v_node->left_child = __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3194
|
|
* if is_leaf:
|
|
* node.left_child = _TREE_LEAF
|
|
* node.right_child = _TREE_LEAF # <<<<<<<<<<<<<<
|
|
* node.feature = _TREE_UNDEFINED
|
|
* node.threshold = _TREE_UNDEFINED
|
|
*/
|
|
__pyx_v_node->right_child = __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3195
|
|
* node.left_child = _TREE_LEAF
|
|
* node.right_child = _TREE_LEAF
|
|
* node.feature = _TREE_UNDEFINED # <<<<<<<<<<<<<<
|
|
* node.threshold = _TREE_UNDEFINED
|
|
*
|
|
*/
|
|
__pyx_v_node->feature = __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3196
|
|
* node.right_child = _TREE_LEAF
|
|
* node.feature = _TREE_UNDEFINED
|
|
* node.threshold = _TREE_UNDEFINED # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_node->threshold = __pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED;
|
|
goto __pyx_L7;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3200
|
|
* else:
|
|
* # left_child and right_child will be set later
|
|
* node.feature = feature # <<<<<<<<<<<<<<
|
|
* node.threshold = threshold
|
|
*
|
|
*/
|
|
__pyx_v_node->feature = __pyx_v_feature;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3201
|
|
* # left_child and right_child will be set later
|
|
* node.feature = feature
|
|
* node.threshold = threshold # <<<<<<<<<<<<<<
|
|
*
|
|
* self.node_count += 1
|
|
*/
|
|
__pyx_v_node->threshold = __pyx_v_threshold;
|
|
}
|
|
__pyx_L7:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3203
|
|
* node.threshold = threshold
|
|
*
|
|
* self.node_count += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* return node_id
|
|
*/
|
|
__pyx_v_self->node_count = (__pyx_v_self->node_count + 1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3205
|
|
* self.node_count += 1
|
|
*
|
|
* return node_id # <<<<<<<<<<<<<<
|
|
*
|
|
* cpdef np.ndarray predict(self, object X):
|
|
*/
|
|
__pyx_r = __pyx_v_node_id;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3166
|
|
* return 0
|
|
*
|
|
* cdef SIZE_t _add_node(self, SIZE_t parent, bint is_left, bint is_leaf, # <<<<<<<<<<<<<<
|
|
* SIZE_t feature, double threshold, double impurity,
|
|
* SIZE_t n_node_samples, double weighted_n_node_samples) nogil:
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3207
|
|
* return node_id
|
|
*
|
|
* cpdef np.ndarray predict(self, object X): # <<<<<<<<<<<<<<
|
|
* """Predict target for X."""
|
|
* out = self._get_value_ndarray().take(self.apply(X), axis=0,
|
|
*/
|
|
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_11predict(PyObject *__pyx_v_self, PyObject *__pyx_v_X); /*proto*/
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree_predict(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X, int __pyx_skip_dispatch) {
|
|
PyObject *__pyx_v_out = NULL;
|
|
PyArrayObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_t_5;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("predict", 0);
|
|
/* Check if called by wrapper */
|
|
if (unlikely(__pyx_skip_dispatch)) ;
|
|
/* Check if overridden in Python */
|
|
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_predict); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3207; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_11predict)) {
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3207; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3207; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3207; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = ((PyArrayObject *)__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3209
|
|
* cpdef np.ndarray predict(self, object X):
|
|
* """Predict target for X."""
|
|
* out = self._get_value_ndarray().take(self.apply(X), axis=0, # <<<<<<<<<<<<<<
|
|
* mode='clip')
|
|
* if self.n_outputs == 1:
|
|
*/
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->_get_value_ndarray(__pyx_v_self)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_take); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->apply(__pyx_v_self, __pyx_v_X, 0)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_axis, __pyx_int_0) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_mode, __pyx_n_s_clip) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_2, __pyx_t_1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3209; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_v_out = __pyx_t_4;
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3211
|
|
* out = self._get_value_ndarray().take(self.apply(X), axis=0,
|
|
* mode='clip')
|
|
* if self.n_outputs == 1: # <<<<<<<<<<<<<<
|
|
* out = out.reshape(X.shape[0], self.max_n_classes)
|
|
* return out
|
|
*/
|
|
__pyx_t_5 = ((__pyx_v_self->n_outputs == 1) != 0);
|
|
if (__pyx_t_5) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3212
|
|
* mode='clip')
|
|
* if self.n_outputs == 1:
|
|
* out = out.reshape(X.shape[0], self.max_n_classes) # <<<<<<<<<<<<<<
|
|
* return out
|
|
*
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_out, __pyx_n_s_reshape); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3212; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3212; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_GetItemInt(__pyx_t_1, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_2 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3212; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->max_n_classes); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3212; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = PyTuple_New(2); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3212; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 1, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_t_4, __pyx_t_3, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3212; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_out, __pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3213
|
|
* if self.n_outputs == 1:
|
|
* out = out.reshape(X.shape[0], self.max_n_classes)
|
|
* return out # <<<<<<<<<<<<<<
|
|
*
|
|
* cpdef np.ndarray apply(self, object X):
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
if (!(likely(((__pyx_v_out) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_out, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3213; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_INCREF(__pyx_v_out);
|
|
__pyx_r = ((PyArrayObject *)__pyx_v_out);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3207
|
|
* return node_id
|
|
*
|
|
* cpdef np.ndarray predict(self, object X): # <<<<<<<<<<<<<<
|
|
* """Predict target for X."""
|
|
* out = self._get_value_ndarray().take(self.apply(X), axis=0,
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.predict", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF(__pyx_v_out);
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_11predict(PyObject *__pyx_v_self, PyObject *__pyx_v_X); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_4Tree_10predict[] = "Predict target for X.";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_11predict(PyObject *__pyx_v_self, PyObject *__pyx_v_X) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("predict (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_10predict(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_X));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_10predict(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X) {
|
|
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("predict", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->predict(__pyx_v_self, __pyx_v_X, 1)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3207; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.predict", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3215
|
|
* return out
|
|
*
|
|
* cpdef np.ndarray apply(self, object X): # <<<<<<<<<<<<<<
|
|
* """Finds the terminal region (=leaf node) for each sample in X."""
|
|
* if issparse(X):
|
|
*/
|
|
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13apply(PyObject *__pyx_v_self, PyObject *__pyx_v_X); /*proto*/
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree_apply(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X, int __pyx_skip_dispatch) {
|
|
PyArrayObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
int __pyx_t_4;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("apply", 0);
|
|
/* Check if called by wrapper */
|
|
if (unlikely(__pyx_skip_dispatch)) ;
|
|
/* Check if overridden in Python */
|
|
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_apply); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_13apply)) {
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = ((PyArrayObject *)__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3217
|
|
* cpdef np.ndarray apply(self, object X):
|
|
* """Finds the terminal region (=leaf node) for each sample in X."""
|
|
* if issparse(X): # <<<<<<<<<<<<<<
|
|
* return self._apply_sparse_csr(X)
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_issparse); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3217; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3217; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_v_X);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_v_X);
|
|
__Pyx_GIVEREF(__pyx_v_X);
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_3, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3217; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_2); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3217; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (__pyx_t_4) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3218
|
|
* """Finds the terminal region (=leaf node) for each sample in X."""
|
|
* if issparse(X):
|
|
* return self._apply_sparse_csr(X) # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return self._apply_dense(X)
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__pyx_t_2 = ((PyObject *)__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_sparse_csr(__pyx_v_self, __pyx_v_X)); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3218; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_r = ((PyArrayObject *)__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3220
|
|
* return self._apply_sparse_csr(X)
|
|
* else:
|
|
* return self._apply_dense(X) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__pyx_t_2 = ((PyObject *)__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_dense(__pyx_v_self, __pyx_v_X)); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3220; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_r = ((PyArrayObject *)__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3215
|
|
* return out
|
|
*
|
|
* cpdef np.ndarray apply(self, object X): # <<<<<<<<<<<<<<
|
|
* """Finds the terminal region (=leaf node) for each sample in X."""
|
|
* if issparse(X):
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.apply", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13apply(PyObject *__pyx_v_self, PyObject *__pyx_v_X); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_4Tree_12apply[] = "Finds the terminal region (=leaf node) for each sample in X.";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13apply(PyObject *__pyx_v_self, PyObject *__pyx_v_X) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("apply (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_12apply(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_X));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_12apply(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X) {
|
|
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("apply", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->apply(__pyx_v_self, __pyx_v_X, 1)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.apply", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3223
|
|
*
|
|
*
|
|
* cdef inline np.ndarray _apply_dense(self, object X): # <<<<<<<<<<<<<<
|
|
* """Finds the terminal region (=leaf node) for each sample in X."""
|
|
*
|
|
*/
|
|
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_dense(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X) {
|
|
PyArrayObject *__pyx_v_X_ndarray = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_ptr;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_sample_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_X_fx_stride;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_samples;
|
|
PyArrayObject *__pyx_v_out = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_out_ptr;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_node;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_out;
|
|
__Pyx_Buffer __pyx_pybuffer_out;
|
|
PyArrayObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_6;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_7;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
PyObject *__pyx_t_9 = NULL;
|
|
PyArrayObject *__pyx_t_10 = NULL;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_t_11;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_apply_dense", 0);
|
|
__pyx_pybuffer_out.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_out.refcount = 0;
|
|
__pyx_pybuffernd_out.data = NULL;
|
|
__pyx_pybuffernd_out.rcbuffer = &__pyx_pybuffer_out;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3227
|
|
*
|
|
* # Check input
|
|
* if not isinstance(X, np.ndarray): # <<<<<<<<<<<<<<
|
|
* raise ValueError("X should be in np.ndarray format, got %s"
|
|
* % type(X))
|
|
*/
|
|
__pyx_t_1 = __Pyx_TypeCheck(__pyx_v_X, ((PyObject*)__pyx_ptype_5numpy_ndarray));
|
|
__pyx_t_2 = ((!(__pyx_t_1 != 0)) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3229
|
|
* if not isinstance(X, np.ndarray):
|
|
* raise ValueError("X should be in np.ndarray format, got %s"
|
|
* % type(X)) # <<<<<<<<<<<<<<
|
|
*
|
|
* if X.dtype != DTYPE:
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyString_Format(__pyx_kp_s_X_should_be_in_np_ndarray_format, ((PyObject *)Py_TYPE(__pyx_v_X))); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3229; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3228
|
|
* # Check input
|
|
* if not isinstance(X, np.ndarray):
|
|
* raise ValueError("X should be in np.ndarray format, got %s" # <<<<<<<<<<<<<<
|
|
* % type(X))
|
|
*
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3228; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3228; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3228; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3231
|
|
* % type(X))
|
|
*
|
|
* if X.dtype != DTYPE: # <<<<<<<<<<<<<<
|
|
* raise ValueError("X.dtype should be np.float32, got %s" % X.dtype)
|
|
*
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_dtype); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3231; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = __Pyx_GetModuleGlobalName(__pyx_n_s_DTYPE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3231; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_t_3, __pyx_t_4, Py_NE); __Pyx_XGOTREF(__pyx_t_5); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3231; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_2 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_2 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3231; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3232
|
|
*
|
|
* if X.dtype != DTYPE:
|
|
* raise ValueError("X.dtype should be np.float32, got %s" % X.dtype) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Extract input
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_dtype); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3232; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_4 = __Pyx_PyString_Format(__pyx_kp_s_X_dtype_should_be_np_float32_got, __pyx_t_5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3232; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3232; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_t_5, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3232; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3232; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3235
|
|
*
|
|
* # Extract input
|
|
* cdef np.ndarray X_ndarray = X # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* X_ptr = <DTYPE_t*> X_ndarray.data
|
|
* cdef SIZE_t X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize
|
|
*/
|
|
if (!(likely(((__pyx_v_X) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_X, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3235; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_4 = __pyx_v_X;
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
__pyx_v_X_ndarray = ((PyArrayObject *)__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3236
|
|
* # Extract input
|
|
* cdef np.ndarray X_ndarray = X
|
|
* cdef DTYPE_t* X_ptr = <DTYPE_t*> X_ndarray.data # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize
|
|
* cdef SIZE_t X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize
|
|
*/
|
|
__pyx_v_X_ptr = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_X_ndarray->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3237
|
|
* cdef np.ndarray X_ndarray = X
|
|
* cdef DTYPE_t* X_ptr = <DTYPE_t*> X_ndarray.data
|
|
* cdef SIZE_t X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize
|
|
* cdef SIZE_t n_samples = X.shape[0]
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_strides); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3237; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_5 = __Pyx_GetItemInt(__pyx_t_4, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_5 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3237; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_5); if (unlikely((__pyx_t_6 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3237; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_itemsize); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3237; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_7 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_5); if (unlikely((__pyx_t_7 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3237; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_v_X_sample_stride = (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_6) / ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_7));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3238
|
|
* cdef DTYPE_t* X_ptr = <DTYPE_t*> X_ndarray.data
|
|
* cdef SIZE_t X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize
|
|
* cdef SIZE_t X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_samples = X.shape[0]
|
|
*
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_strides); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_4 = __Pyx_GetItemInt(__pyx_t_5, 1, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_4 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3238; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_7 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_4); if (unlikely((__pyx_t_7 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_itemsize); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_6 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_4); if (unlikely((__pyx_t_6 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3238; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_v_X_fx_stride = (((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_7) / ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)__pyx_t_6));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3239
|
|
* cdef SIZE_t X_sample_stride = <SIZE_t> X.strides[0] / <SIZE_t> X.itemsize
|
|
* cdef SIZE_t X_fx_stride = <SIZE_t> X.strides[1] / <SIZE_t> X.itemsize
|
|
* cdef SIZE_t n_samples = X.shape[0] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize output
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3239; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_5 = __Pyx_GetItemInt(__pyx_t_4, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_5 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3239; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_5); if (unlikely((__pyx_t_6 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3239; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_v_n_samples = __pyx_t_6;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3242
|
|
*
|
|
* # Initialize output
|
|
* cdef np.ndarray[SIZE_t] out = np.zeros((n_samples,), dtype=np.intp) # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* out_ptr = <SIZE_t*> out.data
|
|
*
|
|
*/
|
|
__pyx_t_5 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_t_5, __pyx_n_s_zeros); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_n_samples); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_5);
|
|
__Pyx_GIVEREF(__pyx_t_5);
|
|
__pyx_t_5 = 0;
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_8 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_8, __pyx_n_s_intp); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
if (PyDict_SetItem(__pyx_t_3, __pyx_n_s_dtype, __pyx_t_9) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_9); __pyx_t_9 = 0;
|
|
__pyx_t_9 = __Pyx_PyObject_Call(__pyx_t_4, __pyx_t_5, __pyx_t_3); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (!(likely(((__pyx_t_9) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_9, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_10 = ((PyArrayObject *)__pyx_t_9);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_out.rcbuffer->pybuffer, (PyObject*)__pyx_t_10, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_SIZE_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_out = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_out.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3242; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_out.diminfo[0].strides = __pyx_pybuffernd_out.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_out.diminfo[0].shape = __pyx_pybuffernd_out.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_10 = 0;
|
|
__pyx_v_out = ((PyArrayObject *)__pyx_t_9);
|
|
__pyx_t_9 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3243
|
|
* # Initialize output
|
|
* cdef np.ndarray[SIZE_t] out = np.zeros((n_samples,), dtype=np.intp)
|
|
* cdef SIZE_t* out_ptr = <SIZE_t*> out.data # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize auxiliary data-structure
|
|
*/
|
|
__pyx_v_out_ptr = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t *)__pyx_v_out->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3246
|
|
*
|
|
* # Initialize auxiliary data-structure
|
|
* cdef Node* node = NULL # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t i = 0
|
|
*
|
|
*/
|
|
__pyx_v_node = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3247
|
|
* # Initialize auxiliary data-structure
|
|
* cdef Node* node = NULL
|
|
* cdef SIZE_t i = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* with nogil:
|
|
*/
|
|
__pyx_v_i = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3249
|
|
* cdef SIZE_t i = 0
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* for i in range(n_samples):
|
|
* node = self.nodes
|
|
*/
|
|
{
|
|
#ifdef WITH_THREAD
|
|
PyThreadState *_save;
|
|
Py_UNBLOCK_THREADS
|
|
#endif
|
|
/*try:*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3250
|
|
*
|
|
* with nogil:
|
|
* for i in range(n_samples): # <<<<<<<<<<<<<<
|
|
* node = self.nodes
|
|
* # While node not a leaf
|
|
*/
|
|
__pyx_t_6 = __pyx_v_n_samples;
|
|
for (__pyx_t_7 = 0; __pyx_t_7 < __pyx_t_6; __pyx_t_7+=1) {
|
|
__pyx_v_i = __pyx_t_7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3251
|
|
* with nogil:
|
|
* for i in range(n_samples):
|
|
* node = self.nodes # <<<<<<<<<<<<<<
|
|
* # While node not a leaf
|
|
* while node.left_child != _TREE_LEAF:
|
|
*/
|
|
__pyx_t_11 = __pyx_v_self->nodes;
|
|
__pyx_v_node = __pyx_t_11;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3253
|
|
* node = self.nodes
|
|
* # While node not a leaf
|
|
* while node.left_child != _TREE_LEAF: # <<<<<<<<<<<<<<
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* if X_ptr[X_sample_stride * i +
|
|
*/
|
|
while (1) {
|
|
__pyx_t_2 = ((__pyx_v_node->left_child != __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF) != 0);
|
|
if (!__pyx_t_2) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3256
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* if X_ptr[X_sample_stride * i +
|
|
* X_fx_stride * node.feature] <= node.threshold: # <<<<<<<<<<<<<<
|
|
* node = &self.nodes[node.left_child]
|
|
* else:
|
|
*/
|
|
__pyx_t_2 = (((__pyx_v_X_ptr[((__pyx_v_X_sample_stride * __pyx_v_i) + (__pyx_v_X_fx_stride * __pyx_v_node->feature))]) <= __pyx_v_node->threshold) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3257
|
|
* if X_ptr[X_sample_stride * i +
|
|
* X_fx_stride * node.feature] <= node.threshold:
|
|
* node = &self.nodes[node.left_child] # <<<<<<<<<<<<<<
|
|
* else:
|
|
* node = &self.nodes[node.right_child]
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_self->nodes[__pyx_v_node->left_child]));
|
|
goto __pyx_L12;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3259
|
|
* node = &self.nodes[node.left_child]
|
|
* else:
|
|
* node = &self.nodes[node.right_child] # <<<<<<<<<<<<<<
|
|
*
|
|
* out_ptr[i] = <SIZE_t>(node - self.nodes) # node offset
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_self->nodes[__pyx_v_node->right_child]));
|
|
}
|
|
__pyx_L12:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3261
|
|
* node = &self.nodes[node.right_child]
|
|
*
|
|
* out_ptr[i] = <SIZE_t>(node - self.nodes) # node offset # <<<<<<<<<<<<<<
|
|
*
|
|
* return out
|
|
*/
|
|
(__pyx_v_out_ptr[__pyx_v_i]) = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_node - __pyx_v_self->nodes));
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3249
|
|
* cdef SIZE_t i = 0
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* for i in range(n_samples):
|
|
* node = self.nodes
|
|
*/
|
|
/*finally:*/ {
|
|
/*normal exit:*/{
|
|
#ifdef WITH_THREAD
|
|
Py_BLOCK_THREADS
|
|
#endif
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3263
|
|
* out_ptr[i] = <SIZE_t>(node - self.nodes) # node offset
|
|
*
|
|
* return out # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline np.ndarray _apply_sparse_csr(self, object X):
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_out));
|
|
__pyx_r = ((PyArrayObject *)__pyx_v_out);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3223
|
|
*
|
|
*
|
|
* cdef inline np.ndarray _apply_dense(self, object X): # <<<<<<<<<<<<<<
|
|
* """Finds the terminal region (=leaf node) for each sample in X."""
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_XDECREF(__pyx_t_8);
|
|
__Pyx_XDECREF(__pyx_t_9);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_out.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree._apply_dense", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_out.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_X_ndarray);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_out);
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3265
|
|
* return out
|
|
*
|
|
* cdef inline np.ndarray _apply_sparse_csr(self, object X): # <<<<<<<<<<<<<<
|
|
* """Finds the terminal region (=leaf node) for each sample in sparse X.
|
|
*
|
|
*/
|
|
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_sparse_csr(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_X) {
|
|
PyArrayObject *__pyx_v_X_data_ndarray = 0;
|
|
PyArrayObject *__pyx_v_X_indices_ndarray = 0;
|
|
PyArrayObject *__pyx_v_X_indptr_ndarray = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_data;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indices;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t *__pyx_v_X_indptr;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_samples;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_n_features;
|
|
PyArrayObject *__pyx_v_out = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_out_ptr;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_feature_value;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_node;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_sample;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_i;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_v_k;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_feature_to_sample;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_X_data_ndarray;
|
|
__Pyx_Buffer __pyx_pybuffer_X_data_ndarray;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_X_indices_ndarray;
|
|
__Pyx_Buffer __pyx_pybuffer_X_indices_ndarray;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_X_indptr_ndarray;
|
|
__Pyx_Buffer __pyx_pybuffer_X_indptr_ndarray;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_out;
|
|
__Pyx_Buffer __pyx_pybuffer_out;
|
|
PyArrayObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyArrayObject *__pyx_t_6 = NULL;
|
|
PyArrayObject *__pyx_t_7 = NULL;
|
|
PyArrayObject *__pyx_t_8 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_9;
|
|
PyObject *__pyx_t_10 = NULL;
|
|
PyObject *__pyx_t_11 = NULL;
|
|
PyArrayObject *__pyx_t_12 = NULL;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_t_13;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_14;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_15;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_t_16;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_t_17;
|
|
__pyx_t_7sklearn_4tree_5_tree_INT32_t __pyx_t_18;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_apply_sparse_csr", 0);
|
|
__pyx_pybuffer_X_data_ndarray.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_X_data_ndarray.refcount = 0;
|
|
__pyx_pybuffernd_X_data_ndarray.data = NULL;
|
|
__pyx_pybuffernd_X_data_ndarray.rcbuffer = &__pyx_pybuffer_X_data_ndarray;
|
|
__pyx_pybuffer_X_indices_ndarray.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_X_indices_ndarray.refcount = 0;
|
|
__pyx_pybuffernd_X_indices_ndarray.data = NULL;
|
|
__pyx_pybuffernd_X_indices_ndarray.rcbuffer = &__pyx_pybuffer_X_indices_ndarray;
|
|
__pyx_pybuffer_X_indptr_ndarray.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_X_indptr_ndarray.refcount = 0;
|
|
__pyx_pybuffernd_X_indptr_ndarray.data = NULL;
|
|
__pyx_pybuffernd_X_indptr_ndarray.rcbuffer = &__pyx_pybuffer_X_indptr_ndarray;
|
|
__pyx_pybuffer_out.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_out.refcount = 0;
|
|
__pyx_pybuffernd_out.data = NULL;
|
|
__pyx_pybuffernd_out.rcbuffer = &__pyx_pybuffer_out;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3270
|
|
* """
|
|
* # Check input
|
|
* if not isinstance(X, csr_matrix): # <<<<<<<<<<<<<<
|
|
* raise ValueError("X should be in csr_matrix format, got %s"
|
|
* % type(X))
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_csr_matrix); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3270; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyObject_IsInstance(__pyx_v_X, __pyx_t_1); if (unlikely(__pyx_t_2 == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3270; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_3 = ((!(__pyx_t_2 != 0)) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3272
|
|
* if not isinstance(X, csr_matrix):
|
|
* raise ValueError("X should be in csr_matrix format, got %s"
|
|
* % type(X)) # <<<<<<<<<<<<<<
|
|
*
|
|
* if X.dtype != DTYPE:
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyString_Format(__pyx_kp_s_X_should_be_in_csr_matrix_format, ((PyObject *)Py_TYPE(__pyx_v_X))); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3272; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3271
|
|
* # Check input
|
|
* if not isinstance(X, csr_matrix):
|
|
* raise ValueError("X should be in csr_matrix format, got %s" # <<<<<<<<<<<<<<
|
|
* % type(X))
|
|
*
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3271; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3271; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_1, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3271; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3274
|
|
* % type(X))
|
|
*
|
|
* if X.dtype != DTYPE: # <<<<<<<<<<<<<<
|
|
* raise ValueError("X.dtype should be np.float32, got %s" % X.dtype)
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_dtype); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3274; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_4 = __Pyx_GetModuleGlobalName(__pyx_n_s_DTYPE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3274; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_5 = PyObject_RichCompare(__pyx_t_1, __pyx_t_4, Py_NE); __Pyx_XGOTREF(__pyx_t_5); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3274; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_3 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3274; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3275
|
|
*
|
|
* if X.dtype != DTYPE:
|
|
* raise ValueError("X.dtype should be np.float32, got %s" % X.dtype) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Extract input
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_dtype); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_4 = __Pyx_PyString_Format(__pyx_kp_s_X_dtype_should_be_np_float32_got, __pyx_t_5); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_t_5, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3275; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3278
|
|
*
|
|
* # Extract input
|
|
* cdef np.ndarray[ndim=1, dtype=DTYPE_t] X_data_ndarray = X.data # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indices_ndarray = X.indices
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indptr_ndarray = X.indptr
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_data); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3278; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3278; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_6 = ((PyArrayObject *)__pyx_t_4);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_X_data_ndarray.rcbuffer->pybuffer, (PyObject*)__pyx_t_6, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_X_data_ndarray = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_X_data_ndarray.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3278; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_X_data_ndarray.diminfo[0].strides = __pyx_pybuffernd_X_data_ndarray.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_X_data_ndarray.diminfo[0].shape = __pyx_pybuffernd_X_data_ndarray.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_6 = 0;
|
|
__pyx_v_X_data_ndarray = ((PyArrayObject *)__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3279
|
|
* # Extract input
|
|
* cdef np.ndarray[ndim=1, dtype=DTYPE_t] X_data_ndarray = X.data
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indices_ndarray = X.indices # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indptr_ndarray = X.indptr
|
|
*
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_indices); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3279; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3279; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_7 = ((PyArrayObject *)__pyx_t_4);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_X_indices_ndarray.rcbuffer->pybuffer, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_INT32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_X_indices_ndarray = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_X_indices_ndarray.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3279; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_X_indices_ndarray.diminfo[0].strides = __pyx_pybuffernd_X_indices_ndarray.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_X_indices_ndarray.diminfo[0].shape = __pyx_pybuffernd_X_indices_ndarray.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_7 = 0;
|
|
__pyx_v_X_indices_ndarray = ((PyArrayObject *)__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3280
|
|
* cdef np.ndarray[ndim=1, dtype=DTYPE_t] X_data_ndarray = X.data
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indices_ndarray = X.indices
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indptr_ndarray = X.indptr # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t* X_data = <DTYPE_t*>X_data_ndarray.data
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_indptr); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3280; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3280; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_8 = ((PyArrayObject *)__pyx_t_4);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_X_indptr_ndarray.rcbuffer->pybuffer, (PyObject*)__pyx_t_8, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_INT32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_X_indptr_ndarray = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_X_indptr_ndarray.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3280; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_X_indptr_ndarray.diminfo[0].strides = __pyx_pybuffernd_X_indptr_ndarray.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_X_indptr_ndarray.diminfo[0].shape = __pyx_pybuffernd_X_indptr_ndarray.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_8 = 0;
|
|
__pyx_v_X_indptr_ndarray = ((PyArrayObject *)__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3282
|
|
* cdef np.ndarray[ndim=1, dtype=INT32_t] X_indptr_ndarray = X.indptr
|
|
*
|
|
* cdef DTYPE_t* X_data = <DTYPE_t*>X_data_ndarray.data # <<<<<<<<<<<<<<
|
|
* cdef INT32_t* X_indices = <INT32_t*>X_indices_ndarray.data
|
|
* cdef INT32_t* X_indptr = <INT32_t*>X_indptr_ndarray.data
|
|
*/
|
|
__pyx_v_X_data = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_X_data_ndarray->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3283
|
|
*
|
|
* cdef DTYPE_t* X_data = <DTYPE_t*>X_data_ndarray.data
|
|
* cdef INT32_t* X_indices = <INT32_t*>X_indices_ndarray.data # <<<<<<<<<<<<<<
|
|
* cdef INT32_t* X_indptr = <INT32_t*>X_indptr_ndarray.data
|
|
*
|
|
*/
|
|
__pyx_v_X_indices = ((__pyx_t_7sklearn_4tree_5_tree_INT32_t *)__pyx_v_X_indices_ndarray->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3284
|
|
* cdef DTYPE_t* X_data = <DTYPE_t*>X_data_ndarray.data
|
|
* cdef INT32_t* X_indices = <INT32_t*>X_indices_ndarray.data
|
|
* cdef INT32_t* X_indptr = <INT32_t*>X_indptr_ndarray.data # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t n_samples = X.shape[0]
|
|
*/
|
|
__pyx_v_X_indptr = ((__pyx_t_7sklearn_4tree_5_tree_INT32_t *)__pyx_v_X_indptr_ndarray->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3286
|
|
* cdef INT32_t* X_indptr = <INT32_t*>X_indptr_ndarray.data
|
|
*
|
|
* cdef SIZE_t n_samples = X.shape[0] # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t n_features = X.shape[1]
|
|
*
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3286; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_5 = __Pyx_GetItemInt(__pyx_t_4, 0, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_5 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3286; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_9 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_5); if (unlikely((__pyx_t_9 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3286; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_v_n_samples = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3287
|
|
*
|
|
* cdef SIZE_t n_samples = X.shape[0]
|
|
* cdef SIZE_t n_features = X.shape[1] # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize output
|
|
*/
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_v_X, __pyx_n_s_shape); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3287; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__pyx_t_4 = __Pyx_GetItemInt(__pyx_t_5, 1, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(__pyx_t_4 == NULL)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3287; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__pyx_t_9 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_4); if (unlikely((__pyx_t_9 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3287; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_v_n_features = __pyx_t_9;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3290
|
|
*
|
|
* # Initialize output
|
|
* cdef np.ndarray[SIZE_t, ndim=1] out = np.zeros((n_samples,), # <<<<<<<<<<<<<<
|
|
* dtype=np.intp)
|
|
* cdef SIZE_t* out_ptr = <SIZE_t*> out.data
|
|
*/
|
|
__pyx_t_4 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_t_4, __pyx_n_s_zeros); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_n_samples); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3291
|
|
* # Initialize output
|
|
* cdef np.ndarray[SIZE_t, ndim=1] out = np.zeros((n_samples,),
|
|
* dtype=np.intp) # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* out_ptr = <SIZE_t*> out.data
|
|
*
|
|
*/
|
|
__pyx_t_10 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_10)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3291; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_10);
|
|
__pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_10, __pyx_n_s_intp); if (unlikely(!__pyx_t_11)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3291; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_11);
|
|
__Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_11) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3290
|
|
*
|
|
* # Initialize output
|
|
* cdef np.ndarray[SIZE_t, ndim=1] out = np.zeros((n_samples,), # <<<<<<<<<<<<<<
|
|
* dtype=np.intp)
|
|
* cdef SIZE_t* out_ptr = <SIZE_t*> out.data
|
|
*/
|
|
__pyx_t_11 = __Pyx_PyObject_Call(__pyx_t_5, __pyx_t_4, __pyx_t_1); if (unlikely(!__pyx_t_11)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_11);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (!(likely(((__pyx_t_11) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_11, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_12 = ((PyArrayObject *)__pyx_t_11);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_out.rcbuffer->pybuffer, (PyObject*)__pyx_t_12, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_SIZE_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
__pyx_v_out = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_pybuffernd_out.rcbuffer->pybuffer.buf = NULL;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3290; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
} else {__pyx_pybuffernd_out.diminfo[0].strides = __pyx_pybuffernd_out.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_out.diminfo[0].shape = __pyx_pybuffernd_out.rcbuffer->pybuffer.shape[0];
|
|
}
|
|
}
|
|
__pyx_t_12 = 0;
|
|
__pyx_v_out = ((PyArrayObject *)__pyx_t_11);
|
|
__pyx_t_11 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3292
|
|
* cdef np.ndarray[SIZE_t, ndim=1] out = np.zeros((n_samples,),
|
|
* dtype=np.intp)
|
|
* cdef SIZE_t* out_ptr = <SIZE_t*> out.data # <<<<<<<<<<<<<<
|
|
*
|
|
* # Initialize auxiliary data-structure
|
|
*/
|
|
__pyx_v_out_ptr = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t *)__pyx_v_out->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3295
|
|
*
|
|
* # Initialize auxiliary data-structure
|
|
* cdef DTYPE_t feature_value = 0. # <<<<<<<<<<<<<<
|
|
* cdef Node* node = NULL
|
|
* cdef DTYPE_t* X_sample = NULL
|
|
*/
|
|
__pyx_v_feature_value = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3296
|
|
* # Initialize auxiliary data-structure
|
|
* cdef DTYPE_t feature_value = 0.
|
|
* cdef Node* node = NULL # <<<<<<<<<<<<<<
|
|
* cdef DTYPE_t* X_sample = NULL
|
|
* cdef SIZE_t i = 0
|
|
*/
|
|
__pyx_v_node = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3297
|
|
* cdef DTYPE_t feature_value = 0.
|
|
* cdef Node* node = NULL
|
|
* cdef DTYPE_t* X_sample = NULL # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t i = 0
|
|
* cdef INT32_t k = 0
|
|
*/
|
|
__pyx_v_X_sample = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3298
|
|
* cdef Node* node = NULL
|
|
* cdef DTYPE_t* X_sample = NULL
|
|
* cdef SIZE_t i = 0 # <<<<<<<<<<<<<<
|
|
* cdef INT32_t k = 0
|
|
*
|
|
*/
|
|
__pyx_v_i = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3299
|
|
* cdef DTYPE_t* X_sample = NULL
|
|
* cdef SIZE_t i = 0
|
|
* cdef INT32_t k = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* # feature_to_sample as a data structure records the last seen sample
|
|
*/
|
|
__pyx_v_k = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3304
|
|
* # for each feature; functionally, it is an efficient way to identify
|
|
* # which features are nonzero in the present sample.
|
|
* cdef SIZE_t* feature_to_sample = NULL # <<<<<<<<<<<<<<
|
|
*
|
|
* safe_realloc(&X_sample, n_features * sizeof(DTYPE_t))
|
|
*/
|
|
__pyx_v_feature_to_sample = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3306
|
|
* cdef SIZE_t* feature_to_sample = NULL
|
|
*
|
|
* safe_realloc(&X_sample, n_features * sizeof(DTYPE_t)) # <<<<<<<<<<<<<<
|
|
* safe_realloc(&feature_to_sample, n_features * sizeof(SIZE_t))
|
|
*
|
|
*/
|
|
__pyx_t_13 = __pyx_fuse_0__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_X_sample), (__pyx_v_n_features * (sizeof(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t)))); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3306; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3307
|
|
*
|
|
* safe_realloc(&X_sample, n_features * sizeof(DTYPE_t))
|
|
* safe_realloc(&feature_to_sample, n_features * sizeof(SIZE_t)) # <<<<<<<<<<<<<<
|
|
*
|
|
* with nogil:
|
|
*/
|
|
__pyx_t_14 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_feature_to_sample), (__pyx_v_n_features * (sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t)))); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3307; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3309
|
|
* safe_realloc(&feature_to_sample, n_features * sizeof(SIZE_t))
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* memset(feature_to_sample, -1, n_features * sizeof(SIZE_t))
|
|
*
|
|
*/
|
|
{
|
|
#ifdef WITH_THREAD
|
|
PyThreadState *_save;
|
|
Py_UNBLOCK_THREADS
|
|
#endif
|
|
/*try:*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3310
|
|
*
|
|
* with nogil:
|
|
* memset(feature_to_sample, -1, n_features * sizeof(SIZE_t)) # <<<<<<<<<<<<<<
|
|
*
|
|
* for i in range(n_samples):
|
|
*/
|
|
memset(__pyx_v_feature_to_sample, -1, (__pyx_v_n_features * (sizeof(__pyx_t_7sklearn_4tree_5_tree_SIZE_t))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3312
|
|
* memset(feature_to_sample, -1, n_features * sizeof(SIZE_t))
|
|
*
|
|
* for i in range(n_samples): # <<<<<<<<<<<<<<
|
|
* node = self.nodes
|
|
*
|
|
*/
|
|
__pyx_t_9 = __pyx_v_n_samples;
|
|
for (__pyx_t_15 = 0; __pyx_t_15 < __pyx_t_9; __pyx_t_15+=1) {
|
|
__pyx_v_i = __pyx_t_15;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3313
|
|
*
|
|
* for i in range(n_samples):
|
|
* node = self.nodes # <<<<<<<<<<<<<<
|
|
*
|
|
* for k in range(X_indptr[i], X_indptr[i + 1]):
|
|
*/
|
|
__pyx_t_16 = __pyx_v_self->nodes;
|
|
__pyx_v_node = __pyx_t_16;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3315
|
|
* node = self.nodes
|
|
*
|
|
* for k in range(X_indptr[i], X_indptr[i + 1]): # <<<<<<<<<<<<<<
|
|
* feature_to_sample[X_indices[k]] = i
|
|
* X_sample[X_indices[k]] = X_data[k]
|
|
*/
|
|
__pyx_t_17 = (__pyx_v_X_indptr[(__pyx_v_i + 1)]);
|
|
for (__pyx_t_18 = (__pyx_v_X_indptr[__pyx_v_i]); __pyx_t_18 < __pyx_t_17; __pyx_t_18+=1) {
|
|
__pyx_v_k = __pyx_t_18;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3316
|
|
*
|
|
* for k in range(X_indptr[i], X_indptr[i + 1]):
|
|
* feature_to_sample[X_indices[k]] = i # <<<<<<<<<<<<<<
|
|
* X_sample[X_indices[k]] = X_data[k]
|
|
*
|
|
*/
|
|
(__pyx_v_feature_to_sample[(__pyx_v_X_indices[__pyx_v_k])]) = __pyx_v_i;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3317
|
|
* for k in range(X_indptr[i], X_indptr[i + 1]):
|
|
* feature_to_sample[X_indices[k]] = i
|
|
* X_sample[X_indices[k]] = X_data[k] # <<<<<<<<<<<<<<
|
|
*
|
|
* # While node not a leaf
|
|
*/
|
|
(__pyx_v_X_sample[(__pyx_v_X_indices[__pyx_v_k])]) = (__pyx_v_X_data[__pyx_v_k]);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3320
|
|
*
|
|
* # While node not a leaf
|
|
* while node.left_child != _TREE_LEAF: # <<<<<<<<<<<<<<
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* if feature_to_sample[node.feature] == i:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_3 = ((__pyx_v_node->left_child != __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF) != 0);
|
|
if (!__pyx_t_3) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3322
|
|
* while node.left_child != _TREE_LEAF:
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* if feature_to_sample[node.feature] == i: # <<<<<<<<<<<<<<
|
|
* feature_value = X_sample[node.feature]
|
|
*
|
|
*/
|
|
__pyx_t_3 = (((__pyx_v_feature_to_sample[__pyx_v_node->feature]) == __pyx_v_i) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3323
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* if feature_to_sample[node.feature] == i:
|
|
* feature_value = X_sample[node.feature] # <<<<<<<<<<<<<<
|
|
*
|
|
* else:
|
|
*/
|
|
__pyx_v_feature_value = (__pyx_v_X_sample[__pyx_v_node->feature]);
|
|
goto __pyx_L14;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3326
|
|
*
|
|
* else:
|
|
* feature_value = 0. # <<<<<<<<<<<<<<
|
|
*
|
|
* if feature_value <= node.threshold:
|
|
*/
|
|
__pyx_v_feature_value = 0.;
|
|
}
|
|
__pyx_L14:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3328
|
|
* feature_value = 0.
|
|
*
|
|
* if feature_value <= node.threshold: # <<<<<<<<<<<<<<
|
|
* node = &self.nodes[node.left_child]
|
|
* else:
|
|
*/
|
|
__pyx_t_3 = ((__pyx_v_feature_value <= __pyx_v_node->threshold) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3329
|
|
*
|
|
* if feature_value <= node.threshold:
|
|
* node = &self.nodes[node.left_child] # <<<<<<<<<<<<<<
|
|
* else:
|
|
* node = &self.nodes[node.right_child]
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_self->nodes[__pyx_v_node->left_child]));
|
|
goto __pyx_L15;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3331
|
|
* node = &self.nodes[node.left_child]
|
|
* else:
|
|
* node = &self.nodes[node.right_child] # <<<<<<<<<<<<<<
|
|
*
|
|
* out_ptr[i] = <SIZE_t>(node - self.nodes) # node offset
|
|
*/
|
|
__pyx_v_node = (&(__pyx_v_self->nodes[__pyx_v_node->right_child]));
|
|
}
|
|
__pyx_L15:;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3333
|
|
* node = &self.nodes[node.right_child]
|
|
*
|
|
* out_ptr[i] = <SIZE_t>(node - self.nodes) # node offset # <<<<<<<<<<<<<<
|
|
*
|
|
* # Free auxiliary arrays
|
|
*/
|
|
(__pyx_v_out_ptr[__pyx_v_i]) = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)(__pyx_v_node - __pyx_v_self->nodes));
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3336
|
|
*
|
|
* # Free auxiliary arrays
|
|
* free(X_sample) # <<<<<<<<<<<<<<
|
|
* free(feature_to_sample)
|
|
*
|
|
*/
|
|
free(__pyx_v_X_sample);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3337
|
|
* # Free auxiliary arrays
|
|
* free(X_sample)
|
|
* free(feature_to_sample) # <<<<<<<<<<<<<<
|
|
*
|
|
* return out
|
|
*/
|
|
free(__pyx_v_feature_to_sample);
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3309
|
|
* safe_realloc(&feature_to_sample, n_features * sizeof(SIZE_t))
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* memset(feature_to_sample, -1, n_features * sizeof(SIZE_t))
|
|
*
|
|
*/
|
|
/*finally:*/ {
|
|
/*normal exit:*/{
|
|
#ifdef WITH_THREAD
|
|
Py_BLOCK_THREADS
|
|
#endif
|
|
goto __pyx_L7;
|
|
}
|
|
__pyx_L7:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3339
|
|
* free(feature_to_sample)
|
|
*
|
|
* return out # <<<<<<<<<<<<<<
|
|
*
|
|
* cpdef compute_feature_importances(self, normalize=True):
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_out));
|
|
__pyx_r = ((PyArrayObject *)__pyx_v_out);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3265
|
|
* return out
|
|
*
|
|
* cdef inline np.ndarray _apply_sparse_csr(self, object X): # <<<<<<<<<<<<<<
|
|
* """Finds the terminal region (=leaf node) for each sample in sparse X.
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_XDECREF(__pyx_t_10);
|
|
__Pyx_XDECREF(__pyx_t_11);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_X_data_ndarray.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_X_indices_ndarray.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_X_indptr_ndarray.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_out.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree._apply_sparse_csr", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_X_data_ndarray.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_X_indices_ndarray.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_X_indptr_ndarray.rcbuffer->pybuffer);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_out.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_X_data_ndarray);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_X_indices_ndarray);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_X_indptr_ndarray);
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_out);
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3341
|
|
* return out
|
|
*
|
|
* cpdef compute_feature_importances(self, normalize=True): # <<<<<<<<<<<<<<
|
|
* """Computes the importance of each feature (aka variable)."""
|
|
* cdef Node* left
|
|
*/
|
|
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_15compute_feature_importances(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static PyObject *__pyx_f_7sklearn_4tree_5_tree_4Tree_compute_feature_importances(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree_compute_feature_importances *__pyx_optional_args) {
|
|
PyObject *__pyx_v_normalize = ((PyObject *)Py_True);
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_left;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_right;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_nodes;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_node;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_v_end_node;
|
|
double __pyx_v_normalizer;
|
|
PyArrayObject *__pyx_v_importances = 0;
|
|
__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *__pyx_v_importance_data;
|
|
__Pyx_LocalBuf_ND __pyx_pybuffernd_importances;
|
|
__Pyx_Buffer __pyx_pybuffer_importances;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
struct __pyx_t_7sklearn_4tree_5_tree_Node *__pyx_t_4;
|
|
PyArrayObject *__pyx_t_5 = NULL;
|
|
int __pyx_t_6;
|
|
PyObject *__pyx_t_7 = NULL;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
PyObject *__pyx_t_9 = NULL;
|
|
int __pyx_t_10;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_11;
|
|
double __pyx_t_12;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("compute_feature_importances", 0);
|
|
if (__pyx_optional_args) {
|
|
if (__pyx_optional_args->__pyx_n > 0) {
|
|
__pyx_v_normalize = __pyx_optional_args->normalize;
|
|
}
|
|
}
|
|
__pyx_pybuffer_importances.pybuffer.buf = NULL;
|
|
__pyx_pybuffer_importances.refcount = 0;
|
|
__pyx_pybuffernd_importances.data = NULL;
|
|
__pyx_pybuffernd_importances.rcbuffer = &__pyx_pybuffer_importances;
|
|
/* Check if called by wrapper */
|
|
if (unlikely(__pyx_skip_dispatch)) ;
|
|
/* Check if overridden in Python */
|
|
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_compute_feature_importances); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3341; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_15compute_feature_importances)) {
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3341; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_v_normalize);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_v_normalize);
|
|
__Pyx_GIVEREF(__pyx_v_normalize);
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_1, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3341; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_r = __pyx_t_3;
|
|
__pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3345
|
|
* cdef Node* left
|
|
* cdef Node* right
|
|
* cdef Node* nodes = self.nodes # <<<<<<<<<<<<<<
|
|
* cdef Node* node = nodes
|
|
* cdef Node* end_node = node + self.node_count
|
|
*/
|
|
__pyx_t_4 = __pyx_v_self->nodes;
|
|
__pyx_v_nodes = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3346
|
|
* cdef Node* right
|
|
* cdef Node* nodes = self.nodes
|
|
* cdef Node* node = nodes # <<<<<<<<<<<<<<
|
|
* cdef Node* end_node = node + self.node_count
|
|
*
|
|
*/
|
|
__pyx_v_node = __pyx_v_nodes;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3347
|
|
* cdef Node* nodes = self.nodes
|
|
* cdef Node* node = nodes
|
|
* cdef Node* end_node = node + self.node_count # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double normalizer = 0.
|
|
*/
|
|
__pyx_v_end_node = (__pyx_v_node + __pyx_v_self->node_count);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3349
|
|
* cdef Node* end_node = node + self.node_count
|
|
*
|
|
* cdef double normalizer = 0. # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef np.ndarray[np.float64_t, ndim=1] importances
|
|
*/
|
|
__pyx_v_normalizer = 0.;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3352
|
|
*
|
|
* cdef np.ndarray[np.float64_t, ndim=1] importances
|
|
* importances = np.zeros((self.n_features,)) # <<<<<<<<<<<<<<
|
|
* cdef DOUBLE_t* importance_data = <DOUBLE_t*>importances.data
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_zeros); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->n_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_PyObject_Call(__pyx_t_3, __pyx_t_1, NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_5 = ((PyArrayObject *)__pyx_t_2);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_importances.rcbuffer->pybuffer);
|
|
__pyx_t_6 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_importances.rcbuffer->pybuffer, (PyObject*)__pyx_t_5, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_6 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_7, &__pyx_t_8, &__pyx_t_9);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_importances.rcbuffer->pybuffer, (PyObject*)__pyx_v_importances, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_7); Py_XDECREF(__pyx_t_8); Py_XDECREF(__pyx_t_9);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_7, __pyx_t_8, __pyx_t_9);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_importances.diminfo[0].strides = __pyx_pybuffernd_importances.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_importances.diminfo[0].shape = __pyx_pybuffernd_importances.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3352; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_5 = 0;
|
|
__pyx_v_importances = ((PyArrayObject *)__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3353
|
|
* cdef np.ndarray[np.float64_t, ndim=1] importances
|
|
* importances = np.zeros((self.n_features,))
|
|
* cdef DOUBLE_t* importance_data = <DOUBLE_t*>importances.data # <<<<<<<<<<<<<<
|
|
*
|
|
* with nogil:
|
|
*/
|
|
__pyx_v_importance_data = ((__pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *)__pyx_v_importances->data);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3355
|
|
* cdef DOUBLE_t* importance_data = <DOUBLE_t*>importances.data
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* while node != end_node:
|
|
* if node.left_child != _TREE_LEAF:
|
|
*/
|
|
{
|
|
#ifdef WITH_THREAD
|
|
PyThreadState *_save;
|
|
Py_UNBLOCK_THREADS
|
|
#endif
|
|
/*try:*/ {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3356
|
|
*
|
|
* with nogil:
|
|
* while node != end_node: # <<<<<<<<<<<<<<
|
|
* if node.left_child != _TREE_LEAF:
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
*/
|
|
while (1) {
|
|
__pyx_t_10 = ((__pyx_v_node != __pyx_v_end_node) != 0);
|
|
if (!__pyx_t_10) break;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3357
|
|
* with nogil:
|
|
* while node != end_node:
|
|
* if node.left_child != _TREE_LEAF: # <<<<<<<<<<<<<<
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* left = &nodes[node.left_child]
|
|
*/
|
|
__pyx_t_10 = ((__pyx_v_node->left_child != __pyx_v_7sklearn_4tree_5_tree__TREE_LEAF) != 0);
|
|
if (__pyx_t_10) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3359
|
|
* if node.left_child != _TREE_LEAF:
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* left = &nodes[node.left_child] # <<<<<<<<<<<<<<
|
|
* right = &nodes[node.right_child]
|
|
*
|
|
*/
|
|
__pyx_v_left = (&(__pyx_v_nodes[__pyx_v_node->left_child]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3360
|
|
* # ... and node.right_child != _TREE_LEAF:
|
|
* left = &nodes[node.left_child]
|
|
* right = &nodes[node.right_child] # <<<<<<<<<<<<<<
|
|
*
|
|
* importance_data[node.feature] += (
|
|
*/
|
|
__pyx_v_right = (&(__pyx_v_nodes[__pyx_v_node->right_child]));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3362
|
|
* right = &nodes[node.right_child]
|
|
*
|
|
* importance_data[node.feature] += ( # <<<<<<<<<<<<<<
|
|
* node.weighted_n_node_samples * node.impurity -
|
|
* left.weighted_n_node_samples * left.impurity -
|
|
*/
|
|
__pyx_t_11 = __pyx_v_node->feature;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3364
|
|
* importance_data[node.feature] += (
|
|
* node.weighted_n_node_samples * node.impurity -
|
|
* left.weighted_n_node_samples * left.impurity - # <<<<<<<<<<<<<<
|
|
* right.weighted_n_node_samples * right.impurity)
|
|
* node += 1
|
|
*/
|
|
(__pyx_v_importance_data[__pyx_t_11]) = ((__pyx_v_importance_data[__pyx_t_11]) + (((__pyx_v_node->weighted_n_node_samples * __pyx_v_node->impurity) - (__pyx_v_left->weighted_n_node_samples * __pyx_v_left->impurity)) - (__pyx_v_right->weighted_n_node_samples * __pyx_v_right->impurity)));
|
|
goto __pyx_L8;
|
|
}
|
|
__pyx_L8:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3366
|
|
* left.weighted_n_node_samples * left.impurity -
|
|
* right.weighted_n_node_samples * right.impurity)
|
|
* node += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* importances /= nodes[0].weighted_n_node_samples
|
|
*/
|
|
__pyx_v_node = (__pyx_v_node + 1);
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3355
|
|
* cdef DOUBLE_t* importance_data = <DOUBLE_t*>importances.data
|
|
*
|
|
* with nogil: # <<<<<<<<<<<<<<
|
|
* while node != end_node:
|
|
* if node.left_child != _TREE_LEAF:
|
|
*/
|
|
/*finally:*/ {
|
|
/*normal exit:*/{
|
|
#ifdef WITH_THREAD
|
|
Py_BLOCK_THREADS
|
|
#endif
|
|
goto __pyx_L5;
|
|
}
|
|
__pyx_L5:;
|
|
}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3368
|
|
* node += 1
|
|
*
|
|
* importances /= nodes[0].weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
*
|
|
* if normalize:
|
|
*/
|
|
__pyx_t_2 = PyFloat_FromDouble((__pyx_v_nodes[0]).weighted_n_node_samples); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = __Pyx_PyNumber_InPlaceDivide(((PyObject *)__pyx_v_importances), __pyx_t_2); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_5 = ((PyArrayObject *)__pyx_t_1);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_importances.rcbuffer->pybuffer);
|
|
__pyx_t_6 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_importances.rcbuffer->pybuffer, (PyObject*)__pyx_t_5, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_6 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_9, &__pyx_t_8, &__pyx_t_7);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_importances.rcbuffer->pybuffer, (PyObject*)__pyx_v_importances, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_9); Py_XDECREF(__pyx_t_8); Py_XDECREF(__pyx_t_7);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_9, __pyx_t_8, __pyx_t_7);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_importances.diminfo[0].strides = __pyx_pybuffernd_importances.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_importances.diminfo[0].shape = __pyx_pybuffernd_importances.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_5 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_importances, ((PyArrayObject *)__pyx_t_1));
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3370
|
|
* importances /= nodes[0].weighted_n_node_samples
|
|
*
|
|
* if normalize: # <<<<<<<<<<<<<<
|
|
* normalizer = np.sum(importances)
|
|
*
|
|
*/
|
|
__pyx_t_10 = __Pyx_PyObject_IsTrue(__pyx_v_normalize); if (unlikely(__pyx_t_10 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3370; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__pyx_t_10) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3371
|
|
*
|
|
* if normalize:
|
|
* normalizer = np.sum(importances) # <<<<<<<<<<<<<<
|
|
*
|
|
* if normalizer > 0.0:
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3371; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_sum); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3371; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3371; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_importances));
|
|
PyTuple_SET_ITEM(__pyx_t_1, 0, ((PyObject *)__pyx_v_importances));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_importances));
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_t_1, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3371; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_12 = __pyx_PyFloat_AsDouble(__pyx_t_3); if (unlikely((__pyx_t_12 == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3371; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_v_normalizer = __pyx_t_12;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3373
|
|
* normalizer = np.sum(importances)
|
|
*
|
|
* if normalizer > 0.0: # <<<<<<<<<<<<<<
|
|
* # Avoid dividing by zero (e.g., when root is pure)
|
|
* importances /= normalizer
|
|
*/
|
|
__pyx_t_10 = ((__pyx_v_normalizer > 0.0) != 0);
|
|
if (__pyx_t_10) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3375
|
|
* if normalizer > 0.0:
|
|
* # Avoid dividing by zero (e.g., when root is pure)
|
|
* importances /= normalizer # <<<<<<<<<<<<<<
|
|
*
|
|
* return importances
|
|
*/
|
|
__pyx_t_3 = PyFloat_FromDouble(__pyx_v_normalizer); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3375; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_1 = __Pyx_PyNumber_InPlaceDivide(((PyObject *)__pyx_v_importances), __pyx_t_3); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3375; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (!(likely(((__pyx_t_1) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_1, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3375; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_t_5 = ((PyArrayObject *)__pyx_t_1);
|
|
{
|
|
__Pyx_BufFmt_StackElem __pyx_stack[1];
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_importances.rcbuffer->pybuffer);
|
|
__pyx_t_6 = __Pyx_GetBufferAndValidate(&__pyx_pybuffernd_importances.rcbuffer->pybuffer, (PyObject*)__pyx_t_5, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack);
|
|
if (unlikely(__pyx_t_6 < 0)) {
|
|
PyErr_Fetch(&__pyx_t_7, &__pyx_t_8, &__pyx_t_9);
|
|
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_pybuffernd_importances.rcbuffer->pybuffer, (PyObject*)__pyx_v_importances, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
|
|
Py_XDECREF(__pyx_t_7); Py_XDECREF(__pyx_t_8); Py_XDECREF(__pyx_t_9);
|
|
__Pyx_RaiseBufferFallbackError();
|
|
} else {
|
|
PyErr_Restore(__pyx_t_7, __pyx_t_8, __pyx_t_9);
|
|
}
|
|
}
|
|
__pyx_pybuffernd_importances.diminfo[0].strides = __pyx_pybuffernd_importances.rcbuffer->pybuffer.strides[0]; __pyx_pybuffernd_importances.diminfo[0].shape = __pyx_pybuffernd_importances.rcbuffer->pybuffer.shape[0];
|
|
if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3375; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_5 = 0;
|
|
__Pyx_DECREF_SET(__pyx_v_importances, ((PyArrayObject *)__pyx_t_1));
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L10;
|
|
}
|
|
__pyx_L10:;
|
|
goto __pyx_L9;
|
|
}
|
|
__pyx_L9:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3377
|
|
* importances /= normalizer
|
|
*
|
|
* return importances # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef np.ndarray _get_value_ndarray(self):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_importances));
|
|
__pyx_r = ((PyObject *)__pyx_v_importances);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3341
|
|
* return out
|
|
*
|
|
* cpdef compute_feature_importances(self, normalize=True): # <<<<<<<<<<<<<<
|
|
* """Computes the importance of each feature (aka variable)."""
|
|
* cdef Node* left
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
|
|
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_importances.rcbuffer->pybuffer);
|
|
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.compute_feature_importances", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
__Pyx_SafeReleaseBuffer(&__pyx_pybuffernd_importances.rcbuffer->pybuffer);
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_importances);
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_15compute_feature_importances(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
|
|
static char __pyx_doc_7sklearn_4tree_5_tree_4Tree_14compute_feature_importances[] = "Computes the importance of each feature (aka variable).";
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_15compute_feature_importances(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
|
|
PyObject *__pyx_v_normalize = 0;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("compute_feature_importances (wrapper)", 0);
|
|
{
|
|
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s_normalize,0};
|
|
PyObject* values[1] = {0};
|
|
values[0] = ((PyObject *)Py_True);
|
|
if (unlikely(__pyx_kwds)) {
|
|
Py_ssize_t kw_args;
|
|
const Py_ssize_t pos_args = PyTuple_GET_SIZE(__pyx_args);
|
|
switch (pos_args) {
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
kw_args = PyDict_Size(__pyx_kwds);
|
|
switch (pos_args) {
|
|
case 0:
|
|
if (kw_args > 0) {
|
|
PyObject* value = PyDict_GetItem(__pyx_kwds, __pyx_n_s_normalize);
|
|
if (value) { values[0] = value; kw_args--; }
|
|
}
|
|
}
|
|
if (unlikely(kw_args > 0)) {
|
|
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, pos_args, "compute_feature_importances") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3341; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
}
|
|
} else {
|
|
switch (PyTuple_GET_SIZE(__pyx_args)) {
|
|
case 1: values[0] = PyTuple_GET_ITEM(__pyx_args, 0);
|
|
case 0: break;
|
|
default: goto __pyx_L5_argtuple_error;
|
|
}
|
|
}
|
|
__pyx_v_normalize = values[0];
|
|
}
|
|
goto __pyx_L4_argument_unpacking_done;
|
|
__pyx_L5_argtuple_error:;
|
|
__Pyx_RaiseArgtupleInvalid("compute_feature_importances", 0, 0, 1, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3341; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
|
|
__pyx_L3_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.compute_feature_importances", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__Pyx_RefNannyFinishContext();
|
|
return NULL;
|
|
__pyx_L4_argument_unpacking_done:;
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_14compute_feature_importances(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), __pyx_v_normalize);
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_14compute_feature_importances(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_normalize) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree_compute_feature_importances __pyx_t_2;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("compute_feature_importances", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_2.__pyx_n = 1;
|
|
__pyx_t_2.normalize = __pyx_v_normalize;
|
|
__pyx_t_1 = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree *)__pyx_v_self->__pyx_vtab)->compute_feature_importances(__pyx_v_self, 1, &__pyx_t_2); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3341; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.compute_feature_importances", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3379
|
|
* return importances
|
|
*
|
|
* cdef np.ndarray _get_value_ndarray(self): # <<<<<<<<<<<<<<
|
|
* """Wraps value as a 3-d NumPy array
|
|
*
|
|
*/
|
|
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree__get_value_ndarray(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
npy_intp __pyx_v_shape[3];
|
|
PyArrayObject *__pyx_v_arr = 0;
|
|
PyArrayObject *__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("_get_value_ndarray", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3386
|
|
* """
|
|
* cdef np.npy_intp shape[3]
|
|
* shape[0] = <np.npy_intp> self.node_count # <<<<<<<<<<<<<<
|
|
* shape[1] = <np.npy_intp> self.n_outputs
|
|
* shape[2] = <np.npy_intp> self.max_n_classes
|
|
*/
|
|
(__pyx_v_shape[0]) = ((npy_intp)__pyx_v_self->node_count);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3387
|
|
* cdef np.npy_intp shape[3]
|
|
* shape[0] = <np.npy_intp> self.node_count
|
|
* shape[1] = <np.npy_intp> self.n_outputs # <<<<<<<<<<<<<<
|
|
* shape[2] = <np.npy_intp> self.max_n_classes
|
|
* cdef np.ndarray arr
|
|
*/
|
|
(__pyx_v_shape[1]) = ((npy_intp)__pyx_v_self->n_outputs);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3388
|
|
* shape[0] = <np.npy_intp> self.node_count
|
|
* shape[1] = <np.npy_intp> self.n_outputs
|
|
* shape[2] = <np.npy_intp> self.max_n_classes # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray arr
|
|
* arr = np.PyArray_SimpleNewFromData(3, shape, np.NPY_DOUBLE, self.value)
|
|
*/
|
|
(__pyx_v_shape[2]) = ((npy_intp)__pyx_v_self->max_n_classes);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3390
|
|
* shape[2] = <np.npy_intp> self.max_n_classes
|
|
* cdef np.ndarray arr
|
|
* arr = np.PyArray_SimpleNewFromData(3, shape, np.NPY_DOUBLE, self.value) # <<<<<<<<<<<<<<
|
|
* Py_INCREF(self)
|
|
* arr.base = <PyObject*> self
|
|
*/
|
|
__pyx_t_1 = PyArray_SimpleNewFromData(3, __pyx_v_shape, NPY_DOUBLE, __pyx_v_self->value); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3390; __pyx_clineno = __LINE__; goto __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_filename = __pyx_f[0]; __pyx_lineno = 3390; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_arr = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3391
|
|
* cdef np.ndarray arr
|
|
* arr = np.PyArray_SimpleNewFromData(3, shape, np.NPY_DOUBLE, self.value)
|
|
* Py_INCREF(self) # <<<<<<<<<<<<<<
|
|
* arr.base = <PyObject*> self
|
|
* return arr
|
|
*/
|
|
Py_INCREF(((PyObject *)__pyx_v_self));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3392
|
|
* arr = np.PyArray_SimpleNewFromData(3, shape, np.NPY_DOUBLE, self.value)
|
|
* Py_INCREF(self)
|
|
* arr.base = <PyObject*> self # <<<<<<<<<<<<<<
|
|
* return arr
|
|
*
|
|
*/
|
|
__pyx_v_arr->base = ((PyObject *)__pyx_v_self);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3393
|
|
* Py_INCREF(self)
|
|
* arr.base = <PyObject*> self
|
|
* return arr # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef np.ndarray _get_node_ndarray(self):
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_arr));
|
|
__pyx_r = __pyx_v_arr;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3379
|
|
* return importances
|
|
*
|
|
* cdef np.ndarray _get_value_ndarray(self): # <<<<<<<<<<<<<<
|
|
* """Wraps value as a 3-d NumPy array
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree._get_value_ndarray", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_arr);
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3395
|
|
* return arr
|
|
*
|
|
* cdef np.ndarray _get_node_ndarray(self): # <<<<<<<<<<<<<<
|
|
* """Wraps nodes as a NumPy struct array
|
|
*
|
|
*/
|
|
|
|
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_4Tree__get_node_ndarray(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
npy_intp __pyx_v_shape[1];
|
|
npy_intp __pyx_v_strides[1];
|
|
PyArrayObject *__pyx_v_arr = 0;
|
|
PyArrayObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_get_node_ndarray", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3403
|
|
* """
|
|
* cdef np.npy_intp shape[1]
|
|
* shape[0] = <np.npy_intp> self.node_count # <<<<<<<<<<<<<<
|
|
* cdef np.npy_intp strides[1]
|
|
* strides[0] = sizeof(Node)
|
|
*/
|
|
(__pyx_v_shape[0]) = ((npy_intp)__pyx_v_self->node_count);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3405
|
|
* shape[0] = <np.npy_intp> self.node_count
|
|
* cdef np.npy_intp strides[1]
|
|
* strides[0] = sizeof(Node) # <<<<<<<<<<<<<<
|
|
* cdef np.ndarray arr
|
|
* Py_INCREF(NODE_DTYPE)
|
|
*/
|
|
(__pyx_v_strides[0]) = (sizeof(struct __pyx_t_7sklearn_4tree_5_tree_Node));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3407
|
|
* strides[0] = sizeof(Node)
|
|
* cdef np.ndarray arr
|
|
* Py_INCREF(NODE_DTYPE) # <<<<<<<<<<<<<<
|
|
* arr = PyArray_NewFromDescr(np.ndarray, <np.dtype> NODE_DTYPE, 1, shape,
|
|
* strides, <void*> self.nodes,
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_NODE_DTYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3407; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
Py_INCREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3408
|
|
* cdef np.ndarray arr
|
|
* Py_INCREF(NODE_DTYPE)
|
|
* arr = PyArray_NewFromDescr(np.ndarray, <np.dtype> NODE_DTYPE, 1, shape, # <<<<<<<<<<<<<<
|
|
* strides, <void*> self.nodes,
|
|
* np.NPY_DEFAULT, None)
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_NODE_DTYPE); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3408; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3410
|
|
* arr = PyArray_NewFromDescr(np.ndarray, <np.dtype> NODE_DTYPE, 1, shape,
|
|
* strides, <void*> self.nodes,
|
|
* np.NPY_DEFAULT, None) # <<<<<<<<<<<<<<
|
|
* Py_INCREF(self)
|
|
* arr.base = <PyObject*> self
|
|
*/
|
|
__pyx_t_2 = PyArray_NewFromDescr(((PyObject *)((PyObject*)__pyx_ptype_5numpy_ndarray)), ((PyArray_Descr *)__pyx_t_1), 1, __pyx_v_shape, __pyx_v_strides, ((void *)__pyx_v_self->nodes), NPY_DEFAULT, Py_None); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3408; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3408
|
|
* cdef np.ndarray arr
|
|
* Py_INCREF(NODE_DTYPE)
|
|
* arr = PyArray_NewFromDescr(np.ndarray, <np.dtype> NODE_DTYPE, 1, shape, # <<<<<<<<<<<<<<
|
|
* strides, <void*> self.nodes,
|
|
* np.NPY_DEFAULT, None)
|
|
*/
|
|
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3408; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_arr = ((PyArrayObject *)__pyx_t_2);
|
|
__pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3411
|
|
* strides, <void*> self.nodes,
|
|
* np.NPY_DEFAULT, None)
|
|
* Py_INCREF(self) # <<<<<<<<<<<<<<
|
|
* arr.base = <PyObject*> self
|
|
* return arr
|
|
*/
|
|
Py_INCREF(((PyObject *)__pyx_v_self));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3412
|
|
* np.NPY_DEFAULT, None)
|
|
* Py_INCREF(self)
|
|
* arr.base = <PyObject*> self # <<<<<<<<<<<<<<
|
|
* return arr
|
|
*
|
|
*/
|
|
__pyx_v_arr->base = ((PyObject *)__pyx_v_self);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3413
|
|
* Py_INCREF(self)
|
|
* arr.base = <PyObject*> self
|
|
* return arr # <<<<<<<<<<<<<<
|
|
*
|
|
* # =============================================================================
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_arr));
|
|
__pyx_r = __pyx_v_arr;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3395
|
|
* return arr
|
|
*
|
|
* cdef np.ndarray _get_node_ndarray(self): # <<<<<<<<<<<<<<
|
|
* """Wraps nodes as a NumPy struct array
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree._get_node_ndarray", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_arr);
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":163
|
|
*
|
|
* # Input/Output layout
|
|
* cdef public SIZE_t n_features # Number of features in X # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t* n_classes # Number of classes in y[:, k]
|
|
* cdef public SIZE_t n_outputs # Number of outputs in y
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_10n_features_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_10n_features_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_10n_features___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_10n_features___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->n_features); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 163; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.n_features.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_10n_features_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_10n_features_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_10n_features_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_10n_features_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 163; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->n_features = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.n_features.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":165
|
|
* cdef public SIZE_t n_features # Number of features in X
|
|
* cdef SIZE_t* n_classes # Number of classes in y[:, k]
|
|
* cdef public SIZE_t n_outputs # Number of outputs in y # <<<<<<<<<<<<<<
|
|
* cdef public SIZE_t max_n_classes # max(n_classes)
|
|
*
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_outputs_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_outputs_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_outputs___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_outputs___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->n_outputs); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 165; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.n_outputs.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_outputs_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_outputs_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_outputs_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_9n_outputs_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 165; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->n_outputs = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.n_outputs.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":166
|
|
* cdef SIZE_t* n_classes # Number of classes in y[:, k]
|
|
* cdef public SIZE_t n_outputs # Number of outputs in y
|
|
* cdef public SIZE_t max_n_classes # max(n_classes) # <<<<<<<<<<<<<<
|
|
*
|
|
* # Inner structures: values are stored separately from node structure,
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13max_n_classes_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_13max_n_classes_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_13max_n_classes___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_13max_n_classes___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->max_n_classes); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 166; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.max_n_classes.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_13max_n_classes_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_13max_n_classes_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_13max_n_classes_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_13max_n_classes_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 166; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->max_n_classes = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.max_n_classes.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":170
|
|
* # Inner structures: values are stored separately from node structure,
|
|
* # since size is determined at runtime.
|
|
* cdef public SIZE_t max_depth # Max depth of the tree # <<<<<<<<<<<<<<
|
|
* cdef public SIZE_t node_count # Counter for node IDs
|
|
* cdef public SIZE_t capacity # Capacity of tree, in terms of nodes
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9max_depth_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_9max_depth_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_9max_depth___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_9max_depth___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->max_depth); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 170; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.max_depth.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_9max_depth_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_9max_depth_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_9max_depth_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_9max_depth_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 170; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->max_depth = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.max_depth.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":171
|
|
* # since size is determined at runtime.
|
|
* cdef public SIZE_t max_depth # Max depth of the tree
|
|
* cdef public SIZE_t node_count # Counter for node IDs # <<<<<<<<<<<<<<
|
|
* cdef public SIZE_t capacity # Capacity of tree, in terms of nodes
|
|
* cdef Node* nodes # Array of nodes
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_10node_count_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_10node_count_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_10node_count___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_10node_count___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->node_count); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 171; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.node_count.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_10node_count_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_10node_count_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_10node_count_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_10node_count_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 171; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->node_count = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.node_count.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pxd":172
|
|
* cdef public SIZE_t max_depth # Max depth of the tree
|
|
* cdef public SIZE_t node_count # Counter for node IDs
|
|
* cdef public SIZE_t capacity # Capacity of tree, in terms of nodes # <<<<<<<<<<<<<<
|
|
* cdef Node* nodes # Array of nodes
|
|
* cdef double* value # (capacity, n_outputs, max_n_classes) array of values
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_8capacity_1__get__(PyObject *__pyx_v_self); /*proto*/
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_4Tree_8capacity_1__get__(PyObject *__pyx_v_self) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_8capacity___get__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4Tree_8capacity___get__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self) {
|
|
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("__get__", 0);
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = __Pyx_PyInt_From_Py_intptr_t(__pyx_v_self->capacity); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 172; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.capacity.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* Python wrapper */
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_8capacity_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
|
|
static int __pyx_pw_7sklearn_4tree_5_tree_4Tree_8capacity_3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree_4Tree_8capacity_2__set__(((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)__pyx_v_self), ((PyObject *)__pyx_v_value));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_7sklearn_4tree_5_tree_4Tree_8capacity_2__set__(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *__pyx_v_self, PyObject *__pyx_v_value) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_1;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__set__", 0);
|
|
__pyx_t_1 = __Pyx_PyInt_As_Py_intptr_t(__pyx_v_value); if (unlikely((__pyx_t_1 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 172; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_self->capacity = __pyx_t_1;
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree.Tree.capacity.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3430
|
|
* (unsigned char*)
|
|
*
|
|
* cdef realloc_ptr safe_realloc(realloc_ptr* p, size_t nelems) except *: # <<<<<<<<<<<<<<
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
*/
|
|
|
|
static __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_fuse_0__pyx_f_7sklearn_4tree_5_tree_safe_realloc(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t **__pyx_v_p, size_t __pyx_v_nelems) {
|
|
size_t __pyx_v_nbytes;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__pyx_fuse_0safe_realloc", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3433
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
* cdef size_t nbytes = nelems * sizeof(p[0][0]) # <<<<<<<<<<<<<<
|
|
* if nbytes / sizeof(p[0][0]) != nelems:
|
|
* # Overflow in the multiplication
|
|
*/
|
|
__pyx_v_nbytes = (__pyx_v_nelems * (sizeof(((__pyx_v_p[0])[0]))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3434
|
|
* # 0.20.1 to crash.
|
|
* cdef size_t nbytes = nelems * sizeof(p[0][0])
|
|
* if nbytes / sizeof(p[0][0]) != nelems: # <<<<<<<<<<<<<<
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_nbytes / (sizeof(((__pyx_v_p[0])[0])))) != __pyx_v_nelems) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3437
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
* % (nelems, sizeof(p[0][0]))) # <<<<<<<<<<<<<<
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL:
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_FromSize_t(__pyx_v_nelems); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = __Pyx_PyInt_FromSize_t((sizeof(((__pyx_v_p[0])[0])))); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyTuple_New(2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyString_Format(__pyx_kp_s_could_not_allocate_d_d_bytes, __pyx_t_4); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3436
|
|
* if nbytes / sizeof(p[0][0]) != nelems:
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes" # <<<<<<<<<<<<<<
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3438
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes) # <<<<<<<<<<<<<<
|
|
* if tmp == NULL:
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*/
|
|
__pyx_v_tmp = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)realloc((__pyx_v_p[0]), __pyx_v_nbytes));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3439
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL: # <<<<<<<<<<<<<<
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_tmp == NULL) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3440
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL:
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes) # <<<<<<<<<<<<<<
|
|
*
|
|
* p[0] = tmp
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyInt_FromSize_t(__pyx_v_nbytes); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = __Pyx_PyString_Format(__pyx_kp_s_could_not_allocate_d_bytes, __pyx_t_3); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_3, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3442
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*
|
|
* p[0] = tmp # <<<<<<<<<<<<<<
|
|
* return tmp # for convenience
|
|
*
|
|
*/
|
|
(__pyx_v_p[0]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3443
|
|
*
|
|
* p[0] = tmp
|
|
* return tmp # for convenience # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_tmp;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3430
|
|
* (unsigned char*)
|
|
*
|
|
* cdef realloc_ptr safe_realloc(realloc_ptr* p, size_t nelems) except *: # <<<<<<<<<<<<<<
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.safe_realloc", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static __pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc(__pyx_t_7sklearn_4tree_5_tree_SIZE_t **__pyx_v_p, size_t __pyx_v_nelems) {
|
|
size_t __pyx_v_nbytes;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_tmp;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__pyx_fuse_1safe_realloc", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3433
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
* cdef size_t nbytes = nelems * sizeof(p[0][0]) # <<<<<<<<<<<<<<
|
|
* if nbytes / sizeof(p[0][0]) != nelems:
|
|
* # Overflow in the multiplication
|
|
*/
|
|
__pyx_v_nbytes = (__pyx_v_nelems * (sizeof(((__pyx_v_p[0])[0]))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3434
|
|
* # 0.20.1 to crash.
|
|
* cdef size_t nbytes = nelems * sizeof(p[0][0])
|
|
* if nbytes / sizeof(p[0][0]) != nelems: # <<<<<<<<<<<<<<
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_nbytes / (sizeof(((__pyx_v_p[0])[0])))) != __pyx_v_nelems) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3437
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
* % (nelems, sizeof(p[0][0]))) # <<<<<<<<<<<<<<
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL:
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_FromSize_t(__pyx_v_nelems); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = __Pyx_PyInt_FromSize_t((sizeof(((__pyx_v_p[0])[0])))); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyTuple_New(2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyString_Format(__pyx_kp_s_could_not_allocate_d_d_bytes, __pyx_t_4); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3436
|
|
* if nbytes / sizeof(p[0][0]) != nelems:
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes" # <<<<<<<<<<<<<<
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3438
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes) # <<<<<<<<<<<<<<
|
|
* if tmp == NULL:
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*/
|
|
__pyx_v_tmp = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t *)realloc((__pyx_v_p[0]), __pyx_v_nbytes));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3439
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL: # <<<<<<<<<<<<<<
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_tmp == NULL) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3440
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL:
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes) # <<<<<<<<<<<<<<
|
|
*
|
|
* p[0] = tmp
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyInt_FromSize_t(__pyx_v_nbytes); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = __Pyx_PyString_Format(__pyx_kp_s_could_not_allocate_d_bytes, __pyx_t_3); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_3, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3442
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*
|
|
* p[0] = tmp # <<<<<<<<<<<<<<
|
|
* return tmp # for convenience
|
|
*
|
|
*/
|
|
(__pyx_v_p[0]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3443
|
|
*
|
|
* p[0] = tmp
|
|
* return tmp # for convenience # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_tmp;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3430
|
|
* (unsigned char*)
|
|
*
|
|
* cdef realloc_ptr safe_realloc(realloc_ptr* p, size_t nelems) except *: # <<<<<<<<<<<<<<
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.safe_realloc", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static unsigned char *__pyx_fuse_2__pyx_f_7sklearn_4tree_5_tree_safe_realloc(unsigned char **__pyx_v_p, size_t __pyx_v_nelems) {
|
|
size_t __pyx_v_nbytes;
|
|
unsigned char *__pyx_v_tmp;
|
|
unsigned char *__pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__pyx_fuse_2safe_realloc", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3433
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
* cdef size_t nbytes = nelems * sizeof(p[0][0]) # <<<<<<<<<<<<<<
|
|
* if nbytes / sizeof(p[0][0]) != nelems:
|
|
* # Overflow in the multiplication
|
|
*/
|
|
__pyx_v_nbytes = (__pyx_v_nelems * (sizeof(((__pyx_v_p[0])[0]))));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3434
|
|
* # 0.20.1 to crash.
|
|
* cdef size_t nbytes = nelems * sizeof(p[0][0])
|
|
* if nbytes / sizeof(p[0][0]) != nelems: # <<<<<<<<<<<<<<
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_nbytes / (sizeof(((__pyx_v_p[0])[0])))) != __pyx_v_nelems) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3437
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
* % (nelems, sizeof(p[0][0]))) # <<<<<<<<<<<<<<
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL:
|
|
*/
|
|
__pyx_t_2 = __Pyx_PyInt_FromSize_t(__pyx_v_nelems); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_3 = __Pyx_PyInt_FromSize_t((sizeof(((__pyx_v_p[0])[0])))); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyTuple_New(2); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 1, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyString_Format(__pyx_kp_s_could_not_allocate_d_d_bytes, __pyx_t_4); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3437; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3436
|
|
* if nbytes / sizeof(p[0][0]) != nelems:
|
|
* # Overflow in the multiplication
|
|
* raise MemoryError("could not allocate (%d * %d) bytes" # <<<<<<<<<<<<<<
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
*/
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3436; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3438
|
|
* raise MemoryError("could not allocate (%d * %d) bytes"
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes) # <<<<<<<<<<<<<<
|
|
* if tmp == NULL:
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*/
|
|
__pyx_v_tmp = ((unsigned char *)realloc((__pyx_v_p[0]), __pyx_v_nbytes));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3439
|
|
* % (nelems, sizeof(p[0][0])))
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL: # <<<<<<<<<<<<<<
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_tmp == NULL) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3440
|
|
* cdef realloc_ptr tmp = <realloc_ptr>realloc(p[0], nbytes)
|
|
* if tmp == NULL:
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes) # <<<<<<<<<<<<<<
|
|
*
|
|
* p[0] = tmp
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyInt_FromSize_t(__pyx_v_nbytes); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = __Pyx_PyString_Format(__pyx_kp_s_could_not_allocate_d_bytes, __pyx_t_3); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_4);
|
|
__Pyx_GIVEREF(__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_MemoryError, __pyx_t_3, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3440; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3442
|
|
* raise MemoryError("could not allocate %d bytes" % nbytes)
|
|
*
|
|
* p[0] = tmp # <<<<<<<<<<<<<<
|
|
* return tmp # for convenience
|
|
*
|
|
*/
|
|
(__pyx_v_p[0]) = __pyx_v_tmp;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3443
|
|
*
|
|
* p[0] = tmp
|
|
* return tmp # for convenience # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_tmp;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3430
|
|
* (unsigned char*)
|
|
*
|
|
* cdef realloc_ptr safe_realloc(realloc_ptr* p, size_t nelems) except *: # <<<<<<<<<<<<<<
|
|
* # sizeof(realloc_ptr[0]) would be more like idiomatic C, but causes Cython
|
|
* # 0.20.1 to crash.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.safe_realloc", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3446
|
|
*
|
|
*
|
|
* def _realloc_test(): # <<<<<<<<<<<<<<
|
|
* # Helper for tests. Tries to allocate <size_t>(-1) / 2 * sizeof(size_t)
|
|
* # bytes, which will always overflow.
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_1_realloc_test(PyObject *__pyx_self, CYTHON_UNUSED PyObject *unused); /*proto*/
|
|
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_1_realloc_test = {__Pyx_NAMESTR("_realloc_test"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_1_realloc_test, METH_NOARGS, __Pyx_DOCSTR(0)};
|
|
static PyObject *__pyx_pw_7sklearn_4tree_5_tree_1_realloc_test(PyObject *__pyx_self, CYTHON_UNUSED PyObject *unused) {
|
|
PyObject *__pyx_r = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("_realloc_test (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_7sklearn_4tree_5_tree__realloc_test(__pyx_self);
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static PyObject *__pyx_pf_7sklearn_4tree_5_tree__realloc_test(CYTHON_UNUSED PyObject *__pyx_self) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_p;
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_realloc_test", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3449
|
|
* # Helper for tests. Tries to allocate <size_t>(-1) / 2 * sizeof(size_t)
|
|
* # bytes, which will always overflow.
|
|
* cdef SIZE_t* p = NULL # <<<<<<<<<<<<<<
|
|
* safe_realloc(&p, <size_t>(-1) / 2)
|
|
* if p != NULL:
|
|
*/
|
|
__pyx_v_p = NULL;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3450
|
|
* # bytes, which will always overflow.
|
|
* cdef SIZE_t* p = NULL
|
|
* safe_realloc(&p, <size_t>(-1) / 2) # <<<<<<<<<<<<<<
|
|
* if p != NULL:
|
|
* free(p)
|
|
*/
|
|
__pyx_t_1 = __pyx_fuse_1__pyx_f_7sklearn_4tree_5_tree_safe_realloc((&__pyx_v_p), (((size_t)-1) / 2)); if (unlikely(PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3450; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3451
|
|
* cdef SIZE_t* p = NULL
|
|
* safe_realloc(&p, <size_t>(-1) / 2)
|
|
* if p != NULL: # <<<<<<<<<<<<<<
|
|
* free(p)
|
|
* assert False
|
|
*/
|
|
__pyx_t_2 = ((__pyx_v_p != NULL) != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "sklearn/tree/_tree.pyx":3452
|
|
* safe_realloc(&p, <size_t>(-1) / 2)
|
|
* if p != NULL:
|
|
* free(p) # <<<<<<<<<<<<<<
|
|
* assert False
|
|
*
|
|
*/
|
|
free(__pyx_v_p);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3453
|
|
* if p != NULL:
|
|
* free(p)
|
|
* assert False # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
#ifndef CYTHON_WITHOUT_ASSERTIONS
|
|
if (unlikely(!Py_OptimizeFlag)) {
|
|
if (unlikely(!0)) {
|
|
PyErr_SetNone(PyExc_AssertionError);
|
|
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 3453; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
}
|
|
#endif
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3446
|
|
*
|
|
*
|
|
* def _realloc_test(): # <<<<<<<<<<<<<<
|
|
* # Helper for tests. Tries to allocate <size_t>(-1) / 2 * sizeof(size_t)
|
|
* # bytes, which will always overflow.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_AddTraceback("sklearn.tree._tree._realloc_test", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3458
|
|
* # rand_r replacement using a 32bit XorShift generator
|
|
* # See http://www.jstatsoft.org/v08/i14/paper for details
|
|
* cdef inline UINT32_t our_rand_r(UINT32_t* seed) nogil: # <<<<<<<<<<<<<<
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 13)
|
|
* seed[0] ^= <UINT32_t>(seed[0] >> 17)
|
|
*/
|
|
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_UINT32_t __pyx_f_7sklearn_4tree_5_tree_our_rand_r(__pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_seed) {
|
|
__pyx_t_7sklearn_4tree_5_tree_UINT32_t __pyx_r;
|
|
long __pyx_t_1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3459
|
|
* # See http://www.jstatsoft.org/v08/i14/paper for details
|
|
* cdef inline UINT32_t our_rand_r(UINT32_t* seed) nogil:
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 13) # <<<<<<<<<<<<<<
|
|
* seed[0] ^= <UINT32_t>(seed[0] >> 17)
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 5)
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
(__pyx_v_seed[__pyx_t_1]) = ((__pyx_v_seed[__pyx_t_1]) ^ ((__pyx_t_7sklearn_4tree_5_tree_UINT32_t)((__pyx_v_seed[0]) << 13)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3460
|
|
* cdef inline UINT32_t our_rand_r(UINT32_t* seed) nogil:
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 13)
|
|
* seed[0] ^= <UINT32_t>(seed[0] >> 17) # <<<<<<<<<<<<<<
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 5)
|
|
*
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
(__pyx_v_seed[__pyx_t_1]) = ((__pyx_v_seed[__pyx_t_1]) ^ ((__pyx_t_7sklearn_4tree_5_tree_UINT32_t)((__pyx_v_seed[0]) >> 17)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3461
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 13)
|
|
* seed[0] ^= <UINT32_t>(seed[0] >> 17)
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 5) # <<<<<<<<<<<<<<
|
|
*
|
|
* return seed[0] % (<UINT32_t>RAND_R_MAX + 1)
|
|
*/
|
|
__pyx_t_1 = 0;
|
|
(__pyx_v_seed[__pyx_t_1]) = ((__pyx_v_seed[__pyx_t_1]) ^ ((__pyx_t_7sklearn_4tree_5_tree_UINT32_t)((__pyx_v_seed[0]) << 5)));
|
|
|
|
/* "sklearn/tree/_tree.pyx":3463
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 5)
|
|
*
|
|
* return seed[0] % (<UINT32_t>RAND_R_MAX + 1) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline np.ndarray sizet_ptr_to_ndarray(SIZE_t* data, SIZE_t size):
|
|
*/
|
|
__pyx_r = ((__pyx_v_seed[0]) % (((__pyx_t_7sklearn_4tree_5_tree_UINT32_t)__pyx_e_7sklearn_4tree_5_tree_RAND_R_MAX) + 1));
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3458
|
|
* # rand_r replacement using a 32bit XorShift generator
|
|
* # See http://www.jstatsoft.org/v08/i14/paper for details
|
|
* cdef inline UINT32_t our_rand_r(UINT32_t* seed) nogil: # <<<<<<<<<<<<<<
|
|
* seed[0] ^= <UINT32_t>(seed[0] << 13)
|
|
* seed[0] ^= <UINT32_t>(seed[0] >> 17)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3465
|
|
* return seed[0] % (<UINT32_t>RAND_R_MAX + 1)
|
|
*
|
|
* cdef inline np.ndarray sizet_ptr_to_ndarray(SIZE_t* data, SIZE_t size): # <<<<<<<<<<<<<<
|
|
* """Encapsulate data into a 1D numpy array of intp's."""
|
|
* cdef np.npy_intp shape[1]
|
|
*/
|
|
|
|
static CYTHON_INLINE PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_sizet_ptr_to_ndarray(__pyx_t_7sklearn_4tree_5_tree_SIZE_t *__pyx_v_data, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_size) {
|
|
npy_intp __pyx_v_shape[1];
|
|
PyArrayObject *__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("sizet_ptr_to_ndarray", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3468
|
|
* """Encapsulate data into a 1D numpy array of intp's."""
|
|
* cdef np.npy_intp shape[1]
|
|
* shape[0] = <np.npy_intp> size # <<<<<<<<<<<<<<
|
|
* return np.PyArray_SimpleNewFromData(1, shape, np.NPY_INTP, data)
|
|
*
|
|
*/
|
|
(__pyx_v_shape[0]) = ((npy_intp)__pyx_v_size);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3469
|
|
* cdef np.npy_intp shape[1]
|
|
* shape[0] = <np.npy_intp> size
|
|
* return np.PyArray_SimpleNewFromData(1, shape, np.NPY_INTP, data) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline SIZE_t rand_int(SIZE_t low, SIZE_t high,
|
|
*/
|
|
__Pyx_XDECREF(((PyObject *)__pyx_r));
|
|
__pyx_t_1 = PyArray_SimpleNewFromData(1, __pyx_v_shape, NPY_INTP, __pyx_v_data); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3469; __pyx_clineno = __LINE__; goto __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_filename = __pyx_f[0]; __pyx_lineno = 3469; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_r = ((PyArrayObject *)__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3465
|
|
* return seed[0] % (<UINT32_t>RAND_R_MAX + 1)
|
|
*
|
|
* cdef inline np.ndarray sizet_ptr_to_ndarray(SIZE_t* data, SIZE_t size): # <<<<<<<<<<<<<<
|
|
* """Encapsulate data into a 1D numpy array of intp's."""
|
|
* cdef np.npy_intp shape[1]
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("sklearn.tree._tree.sizet_ptr_to_ndarray", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF((PyObject *)__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3471
|
|
* return np.PyArray_SimpleNewFromData(1, shape, np.NPY_INTP, data)
|
|
*
|
|
* cdef inline SIZE_t rand_int(SIZE_t low, SIZE_t high, # <<<<<<<<<<<<<<
|
|
* UINT32_t* random_state) nogil:
|
|
* """Generate a random integer in [0; end)."""
|
|
*/
|
|
|
|
static CYTHON_INLINE __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_f_7sklearn_4tree_5_tree_rand_int(__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_low, __pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_v_high, __pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state) {
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3474
|
|
* UINT32_t* random_state) nogil:
|
|
* """Generate a random integer in [0; end)."""
|
|
* return low + our_rand_r(random_state) % (high - low) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline double rand_uniform(double low, double high,
|
|
*/
|
|
__pyx_r = (__pyx_v_low + (__pyx_f_7sklearn_4tree_5_tree_our_rand_r(__pyx_v_random_state) % (__pyx_v_high - __pyx_v_low)));
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3471
|
|
* return np.PyArray_SimpleNewFromData(1, shape, np.NPY_INTP, data)
|
|
*
|
|
* cdef inline SIZE_t rand_int(SIZE_t low, SIZE_t high, # <<<<<<<<<<<<<<
|
|
* UINT32_t* random_state) nogil:
|
|
* """Generate a random integer in [0; end)."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3476
|
|
* return low + our_rand_r(random_state) % (high - low)
|
|
*
|
|
* cdef inline double rand_uniform(double low, double high, # <<<<<<<<<<<<<<
|
|
* UINT32_t* random_state) nogil:
|
|
* """Generate a random double in [low; high)."""
|
|
*/
|
|
|
|
static CYTHON_INLINE double __pyx_f_7sklearn_4tree_5_tree_rand_uniform(double __pyx_v_low, double __pyx_v_high, __pyx_t_7sklearn_4tree_5_tree_UINT32_t *__pyx_v_random_state) {
|
|
double __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3480
|
|
* """Generate a random double in [low; high)."""
|
|
* return ((high - low) * <double> our_rand_r(random_state) /
|
|
* <double> RAND_R_MAX) + low # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline double log(double x) nogil:
|
|
*/
|
|
__pyx_r = ((((__pyx_v_high - __pyx_v_low) * ((double)__pyx_f_7sklearn_4tree_5_tree_our_rand_r(__pyx_v_random_state))) / ((double)__pyx_e_7sklearn_4tree_5_tree_RAND_R_MAX)) + __pyx_v_low);
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3476
|
|
* return low + our_rand_r(random_state) % (high - low)
|
|
*
|
|
* cdef inline double rand_uniform(double low, double high, # <<<<<<<<<<<<<<
|
|
* UINT32_t* random_state) nogil:
|
|
* """Generate a random double in [low; high)."""
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "sklearn/tree/_tree.pyx":3482
|
|
* <double> RAND_R_MAX) + low
|
|
*
|
|
* cdef inline double log(double x) nogil: # <<<<<<<<<<<<<<
|
|
* return ln(x) / ln(2.0)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE double __pyx_f_7sklearn_4tree_5_tree_log(double __pyx_v_x) {
|
|
double __pyx_r;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3483
|
|
*
|
|
* cdef inline double log(double x) nogil:
|
|
* return ln(x) / ln(2.0) # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = (log(__pyx_v_x) / log(2.0));
|
|
goto __pyx_L0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3482
|
|
* <double> RAND_R_MAX) + low
|
|
*
|
|
* cdef inline double log(double x) nogil: # <<<<<<<<<<<<<<
|
|
* return ln(x) / ln(2.0)
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":194
|
|
* # experimental exception made for __getbuffer__ and __releasebuffer__
|
|
* # -- the details of this may change.
|
|
* def __getbuffer__(ndarray self, Py_buffer* info, int flags): # <<<<<<<<<<<<<<
|
|
* # This implementation of getbuffer is geared towards Cython
|
|
* # requirements, and does not yet fullfill the PEP.
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static CYTHON_UNUSED int __pyx_pw_5numpy_7ndarray_1__getbuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags); /*proto*/
|
|
static CYTHON_UNUSED int __pyx_pw_5numpy_7ndarray_1__getbuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags) {
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__getbuffer__ (wrapper)", 0);
|
|
__pyx_r = __pyx_pf_5numpy_7ndarray___getbuffer__(((PyArrayObject *)__pyx_v_self), ((Py_buffer *)__pyx_v_info), ((int)__pyx_v_flags));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
static int __pyx_pf_5numpy_7ndarray___getbuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags) {
|
|
int __pyx_v_copy_shape;
|
|
int __pyx_v_i;
|
|
int __pyx_v_ndim;
|
|
int __pyx_v_endian_detector;
|
|
int __pyx_v_little_endian;
|
|
int __pyx_v_t;
|
|
char *__pyx_v_f;
|
|
PyArray_Descr *__pyx_v_descr = 0;
|
|
int __pyx_v_offset;
|
|
int __pyx_v_hasfields;
|
|
int __pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
int __pyx_t_3;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
char *__pyx_t_9;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("__getbuffer__", 0);
|
|
if (__pyx_v_info != NULL) {
|
|
__pyx_v_info->obj = Py_None; __Pyx_INCREF(Py_None);
|
|
__Pyx_GIVEREF(__pyx_v_info->obj);
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":200
|
|
* # of flags
|
|
*
|
|
* if info == NULL: return # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef int copy_shape, i, ndim
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_info == NULL) != 0);
|
|
if (__pyx_t_1) {
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":203
|
|
*
|
|
* cdef int copy_shape, i, ndim
|
|
* cdef int endian_detector = 1 # <<<<<<<<<<<<<<
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
|
|
*
|
|
*/
|
|
__pyx_v_endian_detector = 1;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":204
|
|
* cdef int copy_shape, i, ndim
|
|
* cdef int endian_detector = 1
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0) # <<<<<<<<<<<<<<
|
|
*
|
|
* ndim = PyArray_NDIM(self)
|
|
*/
|
|
__pyx_v_little_endian = ((((char *)(&__pyx_v_endian_detector))[0]) != 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":206
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
|
|
*
|
|
* ndim = PyArray_NDIM(self) # <<<<<<<<<<<<<<
|
|
*
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
*/
|
|
__pyx_v_ndim = PyArray_NDIM(__pyx_v_self);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":208
|
|
* ndim = PyArray_NDIM(self)
|
|
*
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t): # <<<<<<<<<<<<<<
|
|
* copy_shape = 1
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = (((sizeof(npy_intp)) != (sizeof(Py_ssize_t))) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":209
|
|
*
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
* copy_shape = 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* copy_shape = 0
|
|
*/
|
|
__pyx_v_copy_shape = 1;
|
|
goto __pyx_L4;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":211
|
|
* copy_shape = 1
|
|
* else:
|
|
* copy_shape = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
*/
|
|
__pyx_v_copy_shape = 0;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":213
|
|
* copy_shape = 0
|
|
*
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS) # <<<<<<<<<<<<<<
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not C contiguous")
|
|
*/
|
|
__pyx_t_1 = (((__pyx_v_flags & PyBUF_C_CONTIGUOUS) == PyBUF_C_CONTIGUOUS) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":214
|
|
*
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"ndarray is not C contiguous")
|
|
*
|
|
*/
|
|
__pyx_t_2 = ((!(PyArray_CHKFLAGS(__pyx_v_self, NPY_C_CONTIGUOUS) != 0)) != 0);
|
|
__pyx_t_3 = __pyx_t_2;
|
|
} else {
|
|
__pyx_t_3 = __pyx_t_1;
|
|
}
|
|
if (__pyx_t_3) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":215
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not C contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__6, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":217
|
|
* raise ValueError(u"ndarray is not C contiguous")
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS) # <<<<<<<<<<<<<<
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not Fortran contiguous")
|
|
*/
|
|
__pyx_t_3 = (((__pyx_v_flags & PyBUF_F_CONTIGUOUS) == PyBUF_F_CONTIGUOUS) != 0);
|
|
if (__pyx_t_3) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":218
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"ndarray is not Fortran contiguous")
|
|
*
|
|
*/
|
|
__pyx_t_1 = ((!(PyArray_CHKFLAGS(__pyx_v_self, NPY_F_CONTIGUOUS) != 0)) != 0);
|
|
__pyx_t_2 = __pyx_t_1;
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_3;
|
|
}
|
|
if (__pyx_t_2) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":219
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not Fortran contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* info.buf = PyArray_DATA(self)
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__7, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 219; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 219; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":221
|
|
* raise ValueError(u"ndarray is not Fortran contiguous")
|
|
*
|
|
* info.buf = PyArray_DATA(self) # <<<<<<<<<<<<<<
|
|
* info.ndim = ndim
|
|
* if copy_shape:
|
|
*/
|
|
__pyx_v_info->buf = PyArray_DATA(__pyx_v_self);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":222
|
|
*
|
|
* info.buf = PyArray_DATA(self)
|
|
* info.ndim = ndim # <<<<<<<<<<<<<<
|
|
* if copy_shape:
|
|
* # Allocate new buffer for strides and shape info.
|
|
*/
|
|
__pyx_v_info->ndim = __pyx_v_ndim;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":223
|
|
* info.buf = PyArray_DATA(self)
|
|
* info.ndim = ndim
|
|
* if copy_shape: # <<<<<<<<<<<<<<
|
|
* # Allocate new buffer for strides and shape info.
|
|
* # This is allocated as one block, strides first.
|
|
*/
|
|
__pyx_t_2 = (__pyx_v_copy_shape != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":226
|
|
* # Allocate new buffer for strides and shape info.
|
|
* # This is allocated as one block, strides first.
|
|
* info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * <size_t>ndim * 2) # <<<<<<<<<<<<<<
|
|
* info.shape = info.strides + ndim
|
|
* for i in range(ndim):
|
|
*/
|
|
__pyx_v_info->strides = ((Py_ssize_t *)malloc((((sizeof(Py_ssize_t)) * ((size_t)__pyx_v_ndim)) * 2)));
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":227
|
|
* # This is allocated as one block, strides first.
|
|
* info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * <size_t>ndim * 2)
|
|
* info.shape = info.strides + ndim # <<<<<<<<<<<<<<
|
|
* for i in range(ndim):
|
|
* info.strides[i] = PyArray_STRIDES(self)[i]
|
|
*/
|
|
__pyx_v_info->shape = (__pyx_v_info->strides + __pyx_v_ndim);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":228
|
|
* info.strides = <Py_ssize_t*>stdlib.malloc(sizeof(Py_ssize_t) * <size_t>ndim * 2)
|
|
* info.shape = info.strides + ndim
|
|
* for i in range(ndim): # <<<<<<<<<<<<<<
|
|
* info.strides[i] = PyArray_STRIDES(self)[i]
|
|
* info.shape[i] = PyArray_DIMS(self)[i]
|
|
*/
|
|
__pyx_t_5 = __pyx_v_ndim;
|
|
for (__pyx_t_6 = 0; __pyx_t_6 < __pyx_t_5; __pyx_t_6+=1) {
|
|
__pyx_v_i = __pyx_t_6;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":229
|
|
* info.shape = info.strides + ndim
|
|
* for i in range(ndim):
|
|
* info.strides[i] = PyArray_STRIDES(self)[i] # <<<<<<<<<<<<<<
|
|
* info.shape[i] = PyArray_DIMS(self)[i]
|
|
* else:
|
|
*/
|
|
(__pyx_v_info->strides[__pyx_v_i]) = (PyArray_STRIDES(__pyx_v_self)[__pyx_v_i]);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":230
|
|
* for i in range(ndim):
|
|
* info.strides[i] = PyArray_STRIDES(self)[i]
|
|
* info.shape[i] = PyArray_DIMS(self)[i] # <<<<<<<<<<<<<<
|
|
* else:
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
|
|
*/
|
|
(__pyx_v_info->shape[__pyx_v_i]) = (PyArray_DIMS(__pyx_v_self)[__pyx_v_i]);
|
|
}
|
|
goto __pyx_L7;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":232
|
|
* info.shape[i] = PyArray_DIMS(self)[i]
|
|
* else:
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self) # <<<<<<<<<<<<<<
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self)
|
|
* info.suboffsets = NULL
|
|
*/
|
|
__pyx_v_info->strides = ((Py_ssize_t *)PyArray_STRIDES(__pyx_v_self));
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":233
|
|
* else:
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self) # <<<<<<<<<<<<<<
|
|
* info.suboffsets = NULL
|
|
* info.itemsize = PyArray_ITEMSIZE(self)
|
|
*/
|
|
__pyx_v_info->shape = ((Py_ssize_t *)PyArray_DIMS(__pyx_v_self));
|
|
}
|
|
__pyx_L7:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":234
|
|
* info.strides = <Py_ssize_t*>PyArray_STRIDES(self)
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self)
|
|
* info.suboffsets = NULL # <<<<<<<<<<<<<<
|
|
* info.itemsize = PyArray_ITEMSIZE(self)
|
|
* info.readonly = not PyArray_ISWRITEABLE(self)
|
|
*/
|
|
__pyx_v_info->suboffsets = NULL;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":235
|
|
* info.shape = <Py_ssize_t*>PyArray_DIMS(self)
|
|
* info.suboffsets = NULL
|
|
* info.itemsize = PyArray_ITEMSIZE(self) # <<<<<<<<<<<<<<
|
|
* info.readonly = not PyArray_ISWRITEABLE(self)
|
|
*
|
|
*/
|
|
__pyx_v_info->itemsize = PyArray_ITEMSIZE(__pyx_v_self);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":236
|
|
* info.suboffsets = NULL
|
|
* info.itemsize = PyArray_ITEMSIZE(self)
|
|
* info.readonly = not PyArray_ISWRITEABLE(self) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef int t
|
|
*/
|
|
__pyx_v_info->readonly = (!(PyArray_ISWRITEABLE(__pyx_v_self) != 0));
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":239
|
|
*
|
|
* cdef int t
|
|
* cdef char* f = NULL # <<<<<<<<<<<<<<
|
|
* cdef dtype descr = self.descr
|
|
* cdef list stack
|
|
*/
|
|
__pyx_v_f = NULL;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":240
|
|
* cdef int t
|
|
* cdef char* f = NULL
|
|
* cdef dtype descr = self.descr # <<<<<<<<<<<<<<
|
|
* cdef list stack
|
|
* cdef int offset
|
|
*/
|
|
__pyx_t_4 = ((PyObject *)__pyx_v_self->descr);
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
__pyx_v_descr = ((PyArray_Descr *)__pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":244
|
|
* cdef int offset
|
|
*
|
|
* cdef bint hasfields = PyDataType_HASFIELDS(descr) # <<<<<<<<<<<<<<
|
|
*
|
|
* if not hasfields and not copy_shape:
|
|
*/
|
|
__pyx_v_hasfields = PyDataType_HASFIELDS(__pyx_v_descr);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":246
|
|
* cdef bint hasfields = PyDataType_HASFIELDS(descr)
|
|
*
|
|
* if not hasfields and not copy_shape: # <<<<<<<<<<<<<<
|
|
* # do not call releasebuffer
|
|
* info.obj = None
|
|
*/
|
|
__pyx_t_2 = ((!(__pyx_v_hasfields != 0)) != 0);
|
|
if (__pyx_t_2) {
|
|
__pyx_t_3 = ((!(__pyx_v_copy_shape != 0)) != 0);
|
|
__pyx_t_1 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_1 = __pyx_t_2;
|
|
}
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":248
|
|
* if not hasfields and not copy_shape:
|
|
* # do not call releasebuffer
|
|
* info.obj = None # <<<<<<<<<<<<<<
|
|
* else:
|
|
* # need to call releasebuffer
|
|
*/
|
|
__Pyx_INCREF(Py_None);
|
|
__Pyx_GIVEREF(Py_None);
|
|
__Pyx_GOTREF(__pyx_v_info->obj);
|
|
__Pyx_DECREF(__pyx_v_info->obj);
|
|
__pyx_v_info->obj = Py_None;
|
|
goto __pyx_L10;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":251
|
|
* else:
|
|
* # need to call releasebuffer
|
|
* info.obj = self # <<<<<<<<<<<<<<
|
|
*
|
|
* if not hasfields:
|
|
*/
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_self));
|
|
__Pyx_GIVEREF(((PyObject *)__pyx_v_self));
|
|
__Pyx_GOTREF(__pyx_v_info->obj);
|
|
__Pyx_DECREF(__pyx_v_info->obj);
|
|
__pyx_v_info->obj = ((PyObject *)__pyx_v_self);
|
|
}
|
|
__pyx_L10:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":253
|
|
* info.obj = self
|
|
*
|
|
* if not hasfields: # <<<<<<<<<<<<<<
|
|
* t = descr.type_num
|
|
* if ((descr.byteorder == c'>' and little_endian) or
|
|
*/
|
|
__pyx_t_1 = ((!(__pyx_v_hasfields != 0)) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":254
|
|
*
|
|
* if not hasfields:
|
|
* t = descr.type_num # <<<<<<<<<<<<<<
|
|
* if ((descr.byteorder == c'>' and little_endian) or
|
|
* (descr.byteorder == c'<' and not little_endian)):
|
|
*/
|
|
__pyx_t_5 = __pyx_v_descr->type_num;
|
|
__pyx_v_t = __pyx_t_5;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":255
|
|
* if not hasfields:
|
|
* t = descr.type_num
|
|
* if ((descr.byteorder == c'>' and little_endian) or # <<<<<<<<<<<<<<
|
|
* (descr.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_descr->byteorder == '>') != 0);
|
|
if (__pyx_t_1) {
|
|
__pyx_t_2 = (__pyx_v_little_endian != 0);
|
|
} else {
|
|
__pyx_t_2 = __pyx_t_1;
|
|
}
|
|
if (!__pyx_t_2) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":256
|
|
* t = descr.type_num
|
|
* if ((descr.byteorder == c'>' and little_endian) or
|
|
* (descr.byteorder == c'<' and not little_endian)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* if t == NPY_BYTE: f = "b"
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_descr->byteorder == '<') != 0);
|
|
if (__pyx_t_1) {
|
|
__pyx_t_3 = ((!(__pyx_v_little_endian != 0)) != 0);
|
|
__pyx_t_7 = __pyx_t_3;
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_1;
|
|
}
|
|
__pyx_t_1 = __pyx_t_7;
|
|
} else {
|
|
__pyx_t_1 = __pyx_t_2;
|
|
}
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":257
|
|
* if ((descr.byteorder == c'>' and little_endian) or
|
|
* (descr.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__8, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 257; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 257; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":274
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
* elif t == NPY_OBJECT: f = "O" # <<<<<<<<<<<<<<
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
*/
|
|
switch (__pyx_v_t) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":258
|
|
* (descr.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* if t == NPY_BYTE: f = "b" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
* elif t == NPY_SHORT: f = "h"
|
|
*/
|
|
case NPY_BYTE:
|
|
__pyx_v_f = __pyx_k_b;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":259
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_SHORT: f = "h"
|
|
* elif t == NPY_USHORT: f = "H"
|
|
*/
|
|
case NPY_UBYTE:
|
|
__pyx_v_f = __pyx_k_B;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":260
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
* elif t == NPY_SHORT: f = "h" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_USHORT: f = "H"
|
|
* elif t == NPY_INT: f = "i"
|
|
*/
|
|
case NPY_SHORT:
|
|
__pyx_v_f = __pyx_k_h;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":261
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
* elif t == NPY_SHORT: f = "h"
|
|
* elif t == NPY_USHORT: f = "H" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_INT: f = "i"
|
|
* elif t == NPY_UINT: f = "I"
|
|
*/
|
|
case NPY_USHORT:
|
|
__pyx_v_f = __pyx_k_H;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":262
|
|
* elif t == NPY_SHORT: f = "h"
|
|
* elif t == NPY_USHORT: f = "H"
|
|
* elif t == NPY_INT: f = "i" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UINT: f = "I"
|
|
* elif t == NPY_LONG: f = "l"
|
|
*/
|
|
case NPY_INT:
|
|
__pyx_v_f = __pyx_k_i;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":263
|
|
* elif t == NPY_USHORT: f = "H"
|
|
* elif t == NPY_INT: f = "i"
|
|
* elif t == NPY_UINT: f = "I" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONG: f = "l"
|
|
* elif t == NPY_ULONG: f = "L"
|
|
*/
|
|
case NPY_UINT:
|
|
__pyx_v_f = __pyx_k_I;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":264
|
|
* elif t == NPY_INT: f = "i"
|
|
* elif t == NPY_UINT: f = "I"
|
|
* elif t == NPY_LONG: f = "l" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONG: f = "L"
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
*/
|
|
case NPY_LONG:
|
|
__pyx_v_f = __pyx_k_l;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":265
|
|
* elif t == NPY_UINT: f = "I"
|
|
* elif t == NPY_LONG: f = "l"
|
|
* elif t == NPY_ULONG: f = "L" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
*/
|
|
case NPY_ULONG:
|
|
__pyx_v_f = __pyx_k_L;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":266
|
|
* elif t == NPY_LONG: f = "l"
|
|
* elif t == NPY_ULONG: f = "L"
|
|
* elif t == NPY_LONGLONG: f = "q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
*/
|
|
case NPY_LONGLONG:
|
|
__pyx_v_f = __pyx_k_q;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":267
|
|
* elif t == NPY_ULONG: f = "L"
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
* elif t == NPY_ULONGLONG: f = "Q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
*/
|
|
case NPY_ULONGLONG:
|
|
__pyx_v_f = __pyx_k_Q;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":268
|
|
* elif t == NPY_LONGLONG: f = "q"
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
* elif t == NPY_FLOAT: f = "f" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
*/
|
|
case NPY_FLOAT:
|
|
__pyx_v_f = __pyx_k_f;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":269
|
|
* elif t == NPY_ULONGLONG: f = "Q"
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
* elif t == NPY_DOUBLE: f = "d" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
*/
|
|
case NPY_DOUBLE:
|
|
__pyx_v_f = __pyx_k_d;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":270
|
|
* elif t == NPY_FLOAT: f = "f"
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
* elif t == NPY_LONGDOUBLE: f = "g" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
*/
|
|
case NPY_LONGDOUBLE:
|
|
__pyx_v_f = __pyx_k_g;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":271
|
|
* elif t == NPY_DOUBLE: f = "d"
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
* elif t == NPY_CFLOAT: f = "Zf" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
*/
|
|
case NPY_CFLOAT:
|
|
__pyx_v_f = __pyx_k_Zf;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":272
|
|
* elif t == NPY_LONGDOUBLE: f = "g"
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
* elif t == NPY_CDOUBLE: f = "Zd" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
* elif t == NPY_OBJECT: f = "O"
|
|
*/
|
|
case NPY_CDOUBLE:
|
|
__pyx_v_f = __pyx_k_Zd;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":273
|
|
* elif t == NPY_CFLOAT: f = "Zf"
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_OBJECT: f = "O"
|
|
* else:
|
|
*/
|
|
case NPY_CLONGDOUBLE:
|
|
__pyx_v_f = __pyx_k_Zg;
|
|
break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":274
|
|
* elif t == NPY_CDOUBLE: f = "Zd"
|
|
* elif t == NPY_CLONGDOUBLE: f = "Zg"
|
|
* elif t == NPY_OBJECT: f = "O" # <<<<<<<<<<<<<<
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
*/
|
|
case NPY_OBJECT:
|
|
__pyx_v_f = __pyx_k_O;
|
|
break;
|
|
default:
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":276
|
|
* elif t == NPY_OBJECT: f = "O"
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t) # <<<<<<<<<<<<<<
|
|
* info.format = f
|
|
* return
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyInt_From_int(__pyx_v_t); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 276; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_8 = PyUnicode_Format(__pyx_kp_u_unknown_dtype_code_in_numpy_pxd, __pyx_t_4); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 276; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 276; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_8);
|
|
__Pyx_GIVEREF(__pyx_t_8);
|
|
__pyx_t_8 = 0;
|
|
__pyx_t_8 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 276; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_8, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 276; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
break;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":277
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
* info.format = f # <<<<<<<<<<<<<<
|
|
* return
|
|
* else:
|
|
*/
|
|
__pyx_v_info->format = __pyx_v_f;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":278
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
* info.format = f
|
|
* return # <<<<<<<<<<<<<<
|
|
* else:
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
|
|
*/
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":280
|
|
* return
|
|
* else:
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len) # <<<<<<<<<<<<<<
|
|
* info.format[0] = c'^' # Native data types, manual alignment
|
|
* offset = 0
|
|
*/
|
|
__pyx_v_info->format = ((char *)malloc(255));
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":281
|
|
* else:
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
|
|
* info.format[0] = c'^' # Native data types, manual alignment # <<<<<<<<<<<<<<
|
|
* offset = 0
|
|
* f = _util_dtypestring(descr, info.format + 1,
|
|
*/
|
|
(__pyx_v_info->format[0]) = '^';
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":282
|
|
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
|
|
* info.format[0] = c'^' # Native data types, manual alignment
|
|
* offset = 0 # <<<<<<<<<<<<<<
|
|
* f = _util_dtypestring(descr, info.format + 1,
|
|
* info.format + _buffer_format_string_len,
|
|
*/
|
|
__pyx_v_offset = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":283
|
|
* info.format[0] = c'^' # Native data types, manual alignment
|
|
* offset = 0
|
|
* f = _util_dtypestring(descr, info.format + 1, # <<<<<<<<<<<<<<
|
|
* info.format + _buffer_format_string_len,
|
|
* &offset)
|
|
*/
|
|
__pyx_t_9 = __pyx_f_5numpy__util_dtypestring(__pyx_v_descr, (__pyx_v_info->format + 1), (__pyx_v_info->format + 255), (&__pyx_v_offset)); if (unlikely(__pyx_t_9 == NULL)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 283; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_f = __pyx_t_9;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":286
|
|
* info.format + _buffer_format_string_len,
|
|
* &offset)
|
|
* f[0] = c'\0' # Terminate format string # <<<<<<<<<<<<<<
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info):
|
|
*/
|
|
(__pyx_v_f[0]) = '\x00';
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":194
|
|
* # experimental exception made for __getbuffer__ and __releasebuffer__
|
|
* # -- the details of this may change.
|
|
* def __getbuffer__(ndarray self, Py_buffer* info, int flags): # <<<<<<<<<<<<<<
|
|
* # This implementation of getbuffer is geared towards Cython
|
|
* # requirements, and does not yet fullfill the PEP.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_r = 0;
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_XDECREF(__pyx_t_8);
|
|
__Pyx_AddTraceback("numpy.ndarray.__getbuffer__", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = -1;
|
|
if (__pyx_v_info != NULL && __pyx_v_info->obj != NULL) {
|
|
__Pyx_GOTREF(__pyx_v_info->obj);
|
|
__Pyx_DECREF(__pyx_v_info->obj); __pyx_v_info->obj = NULL;
|
|
}
|
|
goto __pyx_L2;
|
|
__pyx_L0:;
|
|
if (__pyx_v_info != NULL && __pyx_v_info->obj == Py_None) {
|
|
__Pyx_GOTREF(Py_None);
|
|
__Pyx_DECREF(Py_None); __pyx_v_info->obj = NULL;
|
|
}
|
|
__pyx_L2:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_descr);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":288
|
|
* f[0] = c'\0' # Terminate format string
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info): # <<<<<<<<<<<<<<
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format)
|
|
*/
|
|
|
|
/* Python wrapper */
|
|
static CYTHON_UNUSED void __pyx_pw_5numpy_7ndarray_3__releasebuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info); /*proto*/
|
|
static CYTHON_UNUSED void __pyx_pw_5numpy_7ndarray_3__releasebuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__releasebuffer__ (wrapper)", 0);
|
|
__pyx_pf_5numpy_7ndarray_2__releasebuffer__(((PyArrayObject *)__pyx_v_self), ((Py_buffer *)__pyx_v_info));
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
static void __pyx_pf_5numpy_7ndarray_2__releasebuffer__(PyArrayObject *__pyx_v_self, Py_buffer *__pyx_v_info) {
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
__Pyx_RefNannySetupContext("__releasebuffer__", 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":289
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info):
|
|
* if PyArray_HASFIELDS(self): # <<<<<<<<<<<<<<
|
|
* stdlib.free(info.format)
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
*/
|
|
__pyx_t_1 = (PyArray_HASFIELDS(__pyx_v_self) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":290
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info):
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format) # <<<<<<<<<<<<<<
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
* stdlib.free(info.strides)
|
|
*/
|
|
free(__pyx_v_info->format);
|
|
goto __pyx_L3;
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":291
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format)
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t): # <<<<<<<<<<<<<<
|
|
* stdlib.free(info.strides)
|
|
* # info.shape was stored after info.strides in the same block
|
|
*/
|
|
__pyx_t_1 = (((sizeof(npy_intp)) != (sizeof(Py_ssize_t))) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":292
|
|
* stdlib.free(info.format)
|
|
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
|
|
* stdlib.free(info.strides) # <<<<<<<<<<<<<<
|
|
* # info.shape was stored after info.strides in the same block
|
|
*
|
|
*/
|
|
free(__pyx_v_info->strides);
|
|
goto __pyx_L4;
|
|
}
|
|
__pyx_L4:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":288
|
|
* f[0] = c'\0' # Terminate format string
|
|
*
|
|
* def __releasebuffer__(ndarray self, Py_buffer* info): # <<<<<<<<<<<<<<
|
|
* if PyArray_HASFIELDS(self):
|
|
* stdlib.free(info.format)
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":768
|
|
* ctypedef npy_cdouble complex_t
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew1(a): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(1, <void*>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);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":769
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew1(a):
|
|
* return PyArray_MultiIterNew(1, <void*>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_filename = __pyx_f[2]; __pyx_lineno = 769; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":768
|
|
* ctypedef npy_cdouble complex_t
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew1(a): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(1, <void*>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;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":771
|
|
* return PyArray_MultiIterNew(1, <void*>a)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew2(a, b): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>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);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":772
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew2(a, b):
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>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_filename = __pyx_f[2]; __pyx_lineno = 772; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":771
|
|
* return PyArray_MultiIterNew(1, <void*>a)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew2(a, b): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>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;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":774
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>b)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew3(a, b, c): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> 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);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":775
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew3(a, b, c):
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(3, ((void *)__pyx_v_a), ((void *)__pyx_v_b), ((void *)__pyx_v_c)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 775; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":774
|
|
* return PyArray_MultiIterNew(2, <void*>a, <void*>b)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew3(a, b, c): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c)
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew3", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":777
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew4(PyObject *__pyx_v_a, PyObject *__pyx_v_b, PyObject *__pyx_v_c, PyObject *__pyx_v_d) {
|
|
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_MultiIterNew4", 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":778
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d):
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e):
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(4, ((void *)__pyx_v_a), ((void *)__pyx_v_b), ((void *)__pyx_v_c), ((void *)__pyx_v_d)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 778; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":777
|
|
* return PyArray_MultiIterNew(3, <void*>a, <void*>b, <void*> c)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew4(a, b, c, d): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew4", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":780
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e)
|
|
*
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew5(PyObject *__pyx_v_a, PyObject *__pyx_v_b, PyObject *__pyx_v_c, PyObject *__pyx_v_d, PyObject *__pyx_v_e) {
|
|
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_MultiIterNew5", 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":781
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e):
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e) # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline char* _util_dtypestring(dtype descr, char* f, char* end, int* offset) except NULL:
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__pyx_t_1 = PyArray_MultiIterNew(5, ((void *)__pyx_v_a), ((void *)__pyx_v_b), ((void *)__pyx_v_c), ((void *)__pyx_v_d), ((void *)__pyx_v_e)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 781; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_r = __pyx_t_1;
|
|
__pyx_t_1 = 0;
|
|
goto __pyx_L0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":780
|
|
* return PyArray_MultiIterNew(4, <void*>a, <void*>b, <void*>c, <void*> d)
|
|
*
|
|
* cdef inline object PyArray_MultiIterNew5(a, b, c, d, e): # <<<<<<<<<<<<<<
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e)
|
|
*
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_AddTraceback("numpy.PyArray_MultiIterNew5", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = 0;
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":783
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e)
|
|
*
|
|
* cdef inline char* _util_dtypestring(dtype descr, char* f, char* end, int* offset) except NULL: # <<<<<<<<<<<<<<
|
|
* # Recursive utility function used in __getbuffer__ to get format
|
|
* # string. The new location in the format string is returned.
|
|
*/
|
|
|
|
static CYTHON_INLINE char *__pyx_f_5numpy__util_dtypestring(PyArray_Descr *__pyx_v_descr, char *__pyx_v_f, char *__pyx_v_end, int *__pyx_v_offset) {
|
|
PyArray_Descr *__pyx_v_child = 0;
|
|
int __pyx_v_endian_detector;
|
|
int __pyx_v_little_endian;
|
|
PyObject *__pyx_v_fields = 0;
|
|
PyObject *__pyx_v_childname = NULL;
|
|
PyObject *__pyx_v_new_offset = NULL;
|
|
PyObject *__pyx_v_t = NULL;
|
|
char *__pyx_r;
|
|
__Pyx_RefNannyDeclarations
|
|
PyObject *__pyx_t_1 = NULL;
|
|
Py_ssize_t __pyx_t_2;
|
|
PyObject *__pyx_t_3 = NULL;
|
|
PyObject *__pyx_t_4 = NULL;
|
|
int __pyx_t_5;
|
|
int __pyx_t_6;
|
|
int __pyx_t_7;
|
|
int __pyx_t_8;
|
|
int __pyx_t_9;
|
|
long __pyx_t_10;
|
|
char *__pyx_t_11;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannySetupContext("_util_dtypestring", 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":790
|
|
* cdef int delta_offset
|
|
* cdef tuple i
|
|
* cdef int endian_detector = 1 # <<<<<<<<<<<<<<
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0)
|
|
* cdef tuple fields
|
|
*/
|
|
__pyx_v_endian_detector = 1;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":791
|
|
* cdef tuple i
|
|
* cdef int endian_detector = 1
|
|
* cdef bint little_endian = ((<char*>&endian_detector)[0] != 0) # <<<<<<<<<<<<<<
|
|
* cdef tuple fields
|
|
*
|
|
*/
|
|
__pyx_v_little_endian = ((((char *)(&__pyx_v_endian_detector))[0]) != 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":794
|
|
* cdef tuple fields
|
|
*
|
|
* for childname in descr.names: # <<<<<<<<<<<<<<
|
|
* fields = descr.fields[childname]
|
|
* child, new_offset = fields
|
|
*/
|
|
if (unlikely(__pyx_v_descr->names == Py_None)) {
|
|
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable");
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_t_1 = __pyx_v_descr->names; __Pyx_INCREF(__pyx_t_1); __pyx_t_2 = 0;
|
|
for (;;) {
|
|
if (__pyx_t_2 >= PyTuple_GET_SIZE(__pyx_t_1)) break;
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
__pyx_t_3 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_2); __Pyx_INCREF(__pyx_t_3); __pyx_t_2++; if (unlikely(0 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#else
|
|
__pyx_t_3 = PySequence_ITEM(__pyx_t_1, __pyx_t_2); __pyx_t_2++; if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
__Pyx_XDECREF_SET(__pyx_v_childname, __pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":795
|
|
*
|
|
* for childname in descr.names:
|
|
* fields = descr.fields[childname] # <<<<<<<<<<<<<<
|
|
* child, new_offset = fields
|
|
*
|
|
*/
|
|
__pyx_t_3 = PyObject_GetItem(__pyx_v_descr->fields, __pyx_v_childname); if (unlikely(__pyx_t_3 == NULL)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
if (!(likely(PyTuple_CheckExact(__pyx_t_3))||((__pyx_t_3) == Py_None)||(PyErr_Format(PyExc_TypeError, "Expected %.16s, got %.200s", "tuple", Py_TYPE(__pyx_t_3)->tp_name), 0))) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_XDECREF_SET(__pyx_v_fields, ((PyObject*)__pyx_t_3));
|
|
__pyx_t_3 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":796
|
|
* for childname in descr.names:
|
|
* fields = descr.fields[childname]
|
|
* child, new_offset = fields # <<<<<<<<<<<<<<
|
|
*
|
|
* if (end - f) - <int>(new_offset - offset[0]) < 15:
|
|
*/
|
|
if (likely(__pyx_v_fields != Py_None)) {
|
|
PyObject* sequence = __pyx_v_fields;
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
Py_ssize_t size = Py_SIZE(sequence);
|
|
#else
|
|
Py_ssize_t size = PySequence_Size(sequence);
|
|
#endif
|
|
if (unlikely(size != 2)) {
|
|
if (size > 2) __Pyx_RaiseTooManyValuesError(2);
|
|
else if (size >= 0) __Pyx_RaiseNeedMoreValuesError(size);
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 796; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
__pyx_t_3 = PyTuple_GET_ITEM(sequence, 0);
|
|
__pyx_t_4 = PyTuple_GET_ITEM(sequence, 1);
|
|
__Pyx_INCREF(__pyx_t_3);
|
|
__Pyx_INCREF(__pyx_t_4);
|
|
#else
|
|
__pyx_t_3 = PySequence_ITEM(sequence, 0); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 796; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PySequence_ITEM(sequence, 1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 796; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
#endif
|
|
} else {
|
|
__Pyx_RaiseNoneNotIterableError(); {__pyx_filename = __pyx_f[2]; __pyx_lineno = 796; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
if (!(likely(((__pyx_t_3) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_3, __pyx_ptype_5numpy_dtype))))) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 796; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_XDECREF_SET(__pyx_v_child, ((PyArray_Descr *)__pyx_t_3));
|
|
__pyx_t_3 = 0;
|
|
__Pyx_XDECREF_SET(__pyx_v_new_offset, __pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":798
|
|
* child, new_offset = fields
|
|
*
|
|
* if (end - f) - <int>(new_offset - offset[0]) < 15: # <<<<<<<<<<<<<<
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
|
|
*
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyInt_From_int((__pyx_v_offset[0])); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyNumber_Subtract(__pyx_v_new_offset, __pyx_t_4); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_5 = __Pyx_PyInt_As_int(__pyx_t_3); if (unlikely((__pyx_t_5 == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 798; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = ((((__pyx_v_end - __pyx_v_f) - ((int)__pyx_t_5)) < 15) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":799
|
|
*
|
|
* if (end - f) - <int>(new_offset - offset[0]) < 15:
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((child.byteorder == c'>' and little_endian) or
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_RuntimeError, __pyx_tuple__9, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 799; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 799; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":801
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
|
|
*
|
|
* if ((child.byteorder == c'>' and little_endian) or # <<<<<<<<<<<<<<
|
|
* (child.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_child->byteorder == '>') != 0);
|
|
if (__pyx_t_6) {
|
|
__pyx_t_7 = (__pyx_v_little_endian != 0);
|
|
} else {
|
|
__pyx_t_7 = __pyx_t_6;
|
|
}
|
|
if (!__pyx_t_7) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":802
|
|
*
|
|
* if ((child.byteorder == c'>' and little_endian) or
|
|
* (child.byteorder == c'<' and not little_endian)): # <<<<<<<<<<<<<<
|
|
* raise ValueError(u"Non-native byte order not supported")
|
|
* # One could encode it in the format string and have Cython
|
|
*/
|
|
__pyx_t_6 = ((__pyx_v_child->byteorder == '<') != 0);
|
|
if (__pyx_t_6) {
|
|
__pyx_t_8 = ((!(__pyx_v_little_endian != 0)) != 0);
|
|
__pyx_t_9 = __pyx_t_8;
|
|
} else {
|
|
__pyx_t_9 = __pyx_t_6;
|
|
}
|
|
__pyx_t_6 = __pyx_t_9;
|
|
} else {
|
|
__pyx_t_6 = __pyx_t_7;
|
|
}
|
|
if (__pyx_t_6) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":803
|
|
* if ((child.byteorder == c'>' and little_endian) or
|
|
* (child.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* # One could encode it in the format string and have Cython
|
|
* # complain instead, BUT: < and > in format strings also imply
|
|
*/
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_tuple__10, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 803; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 803; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":813
|
|
*
|
|
* # Output padding bytes
|
|
* while offset[0] < new_offset: # <<<<<<<<<<<<<<
|
|
* f[0] = 120 # "x"; pad byte
|
|
* f += 1
|
|
*/
|
|
while (1) {
|
|
__pyx_t_3 = __Pyx_PyInt_From_int((__pyx_v_offset[0])); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 813; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_t_3, __pyx_v_new_offset, Py_LT); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 813; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 813; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (!__pyx_t_6) break;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":814
|
|
* # Output padding bytes
|
|
* while offset[0] < new_offset:
|
|
* f[0] = 120 # "x"; pad byte # <<<<<<<<<<<<<<
|
|
* f += 1
|
|
* offset[0] += 1
|
|
*/
|
|
(__pyx_v_f[0]) = 120;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":815
|
|
* while offset[0] < new_offset:
|
|
* f[0] = 120 # "x"; pad byte
|
|
* f += 1 # <<<<<<<<<<<<<<
|
|
* offset[0] += 1
|
|
*
|
|
*/
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":816
|
|
* f[0] = 120 # "x"; pad byte
|
|
* f += 1
|
|
* offset[0] += 1 # <<<<<<<<<<<<<<
|
|
*
|
|
* offset[0] += child.itemsize
|
|
*/
|
|
__pyx_t_10 = 0;
|
|
(__pyx_v_offset[__pyx_t_10]) = ((__pyx_v_offset[__pyx_t_10]) + 1);
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":818
|
|
* offset[0] += 1
|
|
*
|
|
* offset[0] += child.itemsize # <<<<<<<<<<<<<<
|
|
*
|
|
* if not PyDataType_HASFIELDS(child):
|
|
*/
|
|
__pyx_t_10 = 0;
|
|
(__pyx_v_offset[__pyx_t_10]) = ((__pyx_v_offset[__pyx_t_10]) + __pyx_v_child->elsize);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":820
|
|
* offset[0] += child.itemsize
|
|
*
|
|
* if not PyDataType_HASFIELDS(child): # <<<<<<<<<<<<<<
|
|
* t = child.type_num
|
|
* if end - f < 5:
|
|
*/
|
|
__pyx_t_6 = ((!(PyDataType_HASFIELDS(__pyx_v_child) != 0)) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":821
|
|
*
|
|
* if not PyDataType_HASFIELDS(child):
|
|
* t = child.type_num # <<<<<<<<<<<<<<
|
|
* if end - f < 5:
|
|
* raise RuntimeError(u"Format string allocated too short.")
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyInt_From_int(__pyx_v_child->type_num); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 821; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_XDECREF_SET(__pyx_v_t, __pyx_t_4);
|
|
__pyx_t_4 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":822
|
|
* if not PyDataType_HASFIELDS(child):
|
|
* t = child.type_num
|
|
* if end - f < 5: # <<<<<<<<<<<<<<
|
|
* raise RuntimeError(u"Format string allocated too short.")
|
|
*
|
|
*/
|
|
__pyx_t_6 = (((__pyx_v_end - __pyx_v_f) < 5) != 0);
|
|
if (__pyx_t_6) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":823
|
|
* t = child.type_num
|
|
* if end - f < 5:
|
|
* raise RuntimeError(u"Format string allocated too short.") # <<<<<<<<<<<<<<
|
|
*
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
*/
|
|
__pyx_t_4 = __Pyx_PyObject_Call(__pyx_builtin_RuntimeError, __pyx_tuple__11, NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__Pyx_Raise(__pyx_t_4, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":826
|
|
*
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
* if t == NPY_BYTE: f[0] = 98 #"b" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B"
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_BYTE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 98;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":827
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
* if t == NPY_BYTE: f[0] = 98 #"b"
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_UBYTE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 66;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":828
|
|
* if t == NPY_BYTE: f[0] = 98 #"b"
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B"
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_SHORT); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 104;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":829
|
|
* elif t == NPY_UBYTE: f[0] = 66 #"B"
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_USHORT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 72;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":830
|
|
* elif t == NPY_SHORT: f[0] = 104 #"h"
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
* elif t == NPY_INT: f[0] = 105 #"i" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_INT); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 105;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":831
|
|
* elif t == NPY_USHORT: f[0] = 72 #"H"
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
* elif t == NPY_UINT: f[0] = 73 #"I" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_UINT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 73;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":832
|
|
* elif t == NPY_INT: f[0] = 105 #"i"
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
* elif t == NPY_LONG: f[0] = 108 #"l" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_LONG); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 108;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":833
|
|
* elif t == NPY_UINT: f[0] = 73 #"I"
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_ULONG); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 76;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":834
|
|
* elif t == NPY_LONG: f[0] = 108 #"l"
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_LONGLONG); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 113;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":835
|
|
* elif t == NPY_ULONG: f[0] = 76 #"L"
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_ULONGLONG); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 81;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":836
|
|
* elif t == NPY_LONGLONG: f[0] = 113 #"q"
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_FLOAT); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 102;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":837
|
|
* elif t == NPY_ULONGLONG: f[0] = 81 #"Q"
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_DOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 100;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":838
|
|
* elif t == NPY_FLOAT: f[0] = 102 #"f"
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g" # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_LONGDOUBLE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 103;
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":839
|
|
* elif t == NPY_DOUBLE: f[0] = 100 #"d"
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_CFLOAT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 839; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 839; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 839; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 90;
|
|
(__pyx_v_f[1]) = 102;
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":840
|
|
* elif t == NPY_LONGDOUBLE: f[0] = 103 #"g"
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd # <<<<<<<<<<<<<<
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O"
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_CDOUBLE); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 90;
|
|
(__pyx_v_f[1]) = 100;
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":841
|
|
* elif t == NPY_CFLOAT: f[0] = 90; f[1] = 102; f += 1 # Zf
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg # <<<<<<<<<<<<<<
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O"
|
|
* else:
|
|
*/
|
|
__pyx_t_3 = PyInt_FromLong(NPY_CLONGDOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); __Pyx_XGOTREF(__pyx_t_4); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_4); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 841; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 90;
|
|
(__pyx_v_f[1]) = 103;
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L11;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":842
|
|
* elif t == NPY_CDOUBLE: f[0] = 90; f[1] = 100; f += 1 # Zd
|
|
* elif t == NPY_CLONGDOUBLE: f[0] = 90; f[1] = 103; f += 1 # Zg
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O" # <<<<<<<<<<<<<<
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
*/
|
|
__pyx_t_4 = PyInt_FromLong(NPY_OBJECT); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 842; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_4, Py_EQ); __Pyx_XGOTREF(__pyx_t_3); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 842; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_3); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 842; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
if (__pyx_t_6) {
|
|
(__pyx_v_f[0]) = 79;
|
|
goto __pyx_L11;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":844
|
|
* elif t == NPY_OBJECT: f[0] = 79 #"O"
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t) # <<<<<<<<<<<<<<
|
|
* f += 1
|
|
* else:
|
|
*/
|
|
__pyx_t_3 = PyUnicode_Format(__pyx_kp_u_unknown_dtype_code_in_numpy_pxd, __pyx_v_t); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 844; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 844; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_4);
|
|
PyTuple_SET_ITEM(__pyx_t_4, 0, __pyx_t_3);
|
|
__Pyx_GIVEREF(__pyx_t_3);
|
|
__pyx_t_3 = 0;
|
|
__pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_ValueError, __pyx_t_4, NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 844; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_3);
|
|
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
|
|
__Pyx_Raise(__pyx_t_3, 0, 0, 0);
|
|
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
|
|
{__pyx_filename = __pyx_f[2]; __pyx_lineno = 844; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
__pyx_L11:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":845
|
|
* else:
|
|
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
|
|
* f += 1 # <<<<<<<<<<<<<<
|
|
* else:
|
|
* # Cython ignores struct boundary information ("T{...}"),
|
|
*/
|
|
__pyx_v_f = (__pyx_v_f + 1);
|
|
goto __pyx_L9;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":849
|
|
* # Cython ignores struct boundary information ("T{...}"),
|
|
* # so don't output it
|
|
* f = _util_dtypestring(child, f, end, offset) # <<<<<<<<<<<<<<
|
|
* return f
|
|
*
|
|
*/
|
|
__pyx_t_11 = __pyx_f_5numpy__util_dtypestring(__pyx_v_child, __pyx_v_f, __pyx_v_end, __pyx_v_offset); if (unlikely(__pyx_t_11 == NULL)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 849; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_v_f = __pyx_t_11;
|
|
}
|
|
__pyx_L9:;
|
|
}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":850
|
|
* # so don't output it
|
|
* f = _util_dtypestring(child, f, end, offset)
|
|
* return f # <<<<<<<<<<<<<<
|
|
*
|
|
*
|
|
*/
|
|
__pyx_r = __pyx_v_f;
|
|
goto __pyx_L0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":783
|
|
* return PyArray_MultiIterNew(5, <void*>a, <void*>b, <void*>c, <void*> d, <void*> e)
|
|
*
|
|
* cdef inline char* _util_dtypestring(dtype descr, char* f, char* end, int* offset) except NULL: # <<<<<<<<<<<<<<
|
|
* # Recursive utility function used in __getbuffer__ to get format
|
|
* # string. The new location in the format string is returned.
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_3);
|
|
__Pyx_XDECREF(__pyx_t_4);
|
|
__Pyx_AddTraceback("numpy._util_dtypestring", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
__pyx_r = NULL;
|
|
__pyx_L0:;
|
|
__Pyx_XDECREF((PyObject *)__pyx_v_child);
|
|
__Pyx_XDECREF(__pyx_v_fields);
|
|
__Pyx_XDECREF(__pyx_v_childname);
|
|
__Pyx_XDECREF(__pyx_v_new_offset);
|
|
__Pyx_XDECREF(__pyx_v_t);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":966
|
|
*
|
|
*
|
|
* cdef inline void set_array_base(ndarray arr, object base): # <<<<<<<<<<<<<<
|
|
* cdef PyObject* baseptr
|
|
* if base is None:
|
|
*/
|
|
|
|
static CYTHON_INLINE void __pyx_f_5numpy_set_array_base(PyArrayObject *__pyx_v_arr, PyObject *__pyx_v_base) {
|
|
PyObject *__pyx_v_baseptr;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
int __pyx_t_2;
|
|
__Pyx_RefNannySetupContext("set_array_base", 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":968
|
|
* cdef inline void set_array_base(ndarray arr, object base):
|
|
* cdef PyObject* baseptr
|
|
* if base is None: # <<<<<<<<<<<<<<
|
|
* baseptr = NULL
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = (__pyx_v_base == Py_None);
|
|
__pyx_t_2 = (__pyx_t_1 != 0);
|
|
if (__pyx_t_2) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":969
|
|
* cdef PyObject* baseptr
|
|
* if base is None:
|
|
* baseptr = NULL # <<<<<<<<<<<<<<
|
|
* else:
|
|
* Py_INCREF(base) # important to do this before decref below!
|
|
*/
|
|
__pyx_v_baseptr = NULL;
|
|
goto __pyx_L3;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":971
|
|
* baseptr = NULL
|
|
* else:
|
|
* Py_INCREF(base) # important to do this before decref below! # <<<<<<<<<<<<<<
|
|
* baseptr = <PyObject*>base
|
|
* Py_XDECREF(arr.base)
|
|
*/
|
|
Py_INCREF(__pyx_v_base);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":972
|
|
* else:
|
|
* Py_INCREF(base) # important to do this before decref below!
|
|
* baseptr = <PyObject*>base # <<<<<<<<<<<<<<
|
|
* Py_XDECREF(arr.base)
|
|
* arr.base = baseptr
|
|
*/
|
|
__pyx_v_baseptr = ((PyObject *)__pyx_v_base);
|
|
}
|
|
__pyx_L3:;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":973
|
|
* Py_INCREF(base) # important to do this before decref below!
|
|
* baseptr = <PyObject*>base
|
|
* Py_XDECREF(arr.base) # <<<<<<<<<<<<<<
|
|
* arr.base = baseptr
|
|
*
|
|
*/
|
|
Py_XDECREF(__pyx_v_arr->base);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":974
|
|
* baseptr = <PyObject*>base
|
|
* Py_XDECREF(arr.base)
|
|
* arr.base = baseptr # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr):
|
|
*/
|
|
__pyx_v_arr->base = __pyx_v_baseptr;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":966
|
|
*
|
|
*
|
|
* cdef inline void set_array_base(ndarray arr, object base): # <<<<<<<<<<<<<<
|
|
* cdef PyObject* baseptr
|
|
* if base is None:
|
|
*/
|
|
|
|
/* function exit code */
|
|
__Pyx_RefNannyFinishContext();
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":976
|
|
* arr.base = baseptr
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr): # <<<<<<<<<<<<<<
|
|
* if arr.base is NULL:
|
|
* return None
|
|
*/
|
|
|
|
static CYTHON_INLINE PyObject *__pyx_f_5numpy_get_array_base(PyArrayObject *__pyx_v_arr) {
|
|
PyObject *__pyx_r = NULL;
|
|
__Pyx_RefNannyDeclarations
|
|
int __pyx_t_1;
|
|
__Pyx_RefNannySetupContext("get_array_base", 0);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":977
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr):
|
|
* if arr.base is NULL: # <<<<<<<<<<<<<<
|
|
* return None
|
|
* else:
|
|
*/
|
|
__pyx_t_1 = ((__pyx_v_arr->base == NULL) != 0);
|
|
if (__pyx_t_1) {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":978
|
|
* cdef inline object get_array_base(ndarray arr):
|
|
* if arr.base is NULL:
|
|
* return None # <<<<<<<<<<<<<<
|
|
* else:
|
|
* return <object>arr.base
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(Py_None);
|
|
__pyx_r = Py_None;
|
|
goto __pyx_L0;
|
|
}
|
|
/*else*/ {
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":980
|
|
* return None
|
|
* else:
|
|
* return <object>arr.base # <<<<<<<<<<<<<<
|
|
*/
|
|
__Pyx_XDECREF(__pyx_r);
|
|
__Pyx_INCREF(((PyObject *)__pyx_v_arr->base));
|
|
__pyx_r = ((PyObject *)__pyx_v_arr->base);
|
|
goto __pyx_L0;
|
|
}
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":976
|
|
* arr.base = baseptr
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr): # <<<<<<<<<<<<<<
|
|
* if arr.base is NULL:
|
|
* return None
|
|
*/
|
|
|
|
/* function exit code */
|
|
__pyx_L0:;
|
|
__Pyx_XGIVEREF(__pyx_r);
|
|
__Pyx_RefNannyFinishContext();
|
|
return __pyx_r;
|
|
}
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion __pyx_vtable_7sklearn_4tree_5_tree_Criterion;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Criterion(PyTypeObject *t, CYTHON_UNUSED PyObject *a, CYTHON_UNUSED PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *p;
|
|
PyObject *o;
|
|
if (likely((t->tp_flags & Py_TPFLAGS_IS_ABSTRACT) == 0)) {
|
|
o = (*t->tp_alloc)(t, 0);
|
|
} else {
|
|
o = (PyObject *) PyBaseObject_Type.tp_new(t, __pyx_empty_tuple, 0);
|
|
}
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)o);
|
|
p->__pyx_vtab = __pyx_vtabptr_7sklearn_4tree_5_tree_Criterion;
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_Criterion(PyObject *o) {
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && (!PyType_IS_GC(Py_TYPE(o)) || !_PyGC_FINALIZED(o))) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
(*Py_TYPE(o)->tp_free)(o);
|
|
}
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_Criterion = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.Criterion"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Criterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Interface for impurity criteria."), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_Criterion, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter __pyx_vtable_7sklearn_4tree_5_tree_Splitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Splitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *p;
|
|
PyObject *o;
|
|
if (likely((t->tp_flags & Py_TPFLAGS_IS_ABSTRACT) == 0)) {
|
|
o = (*t->tp_alloc)(t, 0);
|
|
} else {
|
|
o = (PyObject *) PyBaseObject_Type.tp_new(t, __pyx_empty_tuple, 0);
|
|
}
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)o);
|
|
p->__pyx_vtab = __pyx_vtabptr_7sklearn_4tree_5_tree_Splitter;
|
|
p->criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)Py_None); Py_INCREF(Py_None);
|
|
p->random_state = Py_None; Py_INCREF(Py_None);
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_8Splitter_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter(PyObject *o) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *p = (struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)o;
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && !_PyGC_FINALIZED(o)) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
PyObject_GC_UnTrack(o);
|
|
{
|
|
PyObject *etype, *eval, *etb;
|
|
PyErr_Fetch(&etype, &eval, &etb);
|
|
++Py_REFCNT(o);
|
|
__pyx_pw_7sklearn_4tree_5_tree_8Splitter_3__dealloc__(o);
|
|
--Py_REFCNT(o);
|
|
PyErr_Restore(etype, eval, etb);
|
|
}
|
|
Py_CLEAR(p->criterion);
|
|
Py_CLEAR(p->random_state);
|
|
(*Py_TYPE(o)->tp_free)(o);
|
|
}
|
|
|
|
static int __pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter(PyObject *o, visitproc v, void *a) {
|
|
int e;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *p = (struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)o;
|
|
if (p->criterion) {
|
|
e = (*v)(((PyObject*)p->criterion), a); if (e) return e;
|
|
}
|
|
if (p->random_state) {
|
|
e = (*v)(p->random_state, a); if (e) return e;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int __pyx_tp_clear_7sklearn_4tree_5_tree_Splitter(PyObject *o) {
|
|
PyObject* tmp;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *p = (struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)o;
|
|
tmp = ((PyObject*)p->criterion);
|
|
p->criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)Py_None); Py_INCREF(Py_None);
|
|
Py_XDECREF(tmp);
|
|
tmp = ((PyObject*)p->random_state);
|
|
p->random_state = Py_None; Py_INCREF(Py_None);
|
|
Py_XDECREF(tmp);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_8Splitter_criterion(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_criterion(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_3__set__(o, v);
|
|
}
|
|
else {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_9criterion_5__del__(o);
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_8Splitter_max_features(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_12max_features_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_max_features(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_12max_features_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_8Splitter_min_samples_leaf(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_min_samples_leaf(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_16min_samples_leaf_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_8Splitter_min_weight_leaf(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_min_weight_leaf(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_8Splitter_15min_weight_leaf_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_Splitter[] = {
|
|
{__Pyx_NAMESTR("__getstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_8Splitter_5__getstate__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{__Pyx_NAMESTR("__setstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_8Splitter_7__setstate__, METH_O, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static struct PyGetSetDef __pyx_getsets_7sklearn_4tree_5_tree_Splitter[] = {
|
|
{(char *)"criterion", __pyx_getprop_7sklearn_4tree_5_tree_8Splitter_criterion, __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_criterion, 0, 0},
|
|
{(char *)"max_features", __pyx_getprop_7sklearn_4tree_5_tree_8Splitter_max_features, __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_max_features, 0, 0},
|
|
{(char *)"min_samples_leaf", __pyx_getprop_7sklearn_4tree_5_tree_8Splitter_min_samples_leaf, __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_min_samples_leaf, 0, 0},
|
|
{(char *)"min_weight_leaf", __pyx_getprop_7sklearn_4tree_5_tree_8Splitter_min_weight_leaf, __pyx_setprop_7sklearn_4tree_5_tree_8Splitter_min_weight_leaf, 0, 0},
|
|
{0, 0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_Splitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.Splitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_Splitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
__pyx_getsets_7sklearn_4tree_5_tree_Splitter, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_Splitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Tree __pyx_vtable_7sklearn_4tree_5_tree_Tree;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Tree(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Tree *p;
|
|
PyObject *o;
|
|
if (likely((t->tp_flags & Py_TPFLAGS_IS_ABSTRACT) == 0)) {
|
|
o = (*t->tp_alloc)(t, 0);
|
|
} else {
|
|
o = (PyObject *) PyBaseObject_Type.tp_new(t, __pyx_empty_tuple, 0);
|
|
}
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_Tree *)o);
|
|
p->__pyx_vtab = __pyx_vtabptr_7sklearn_4tree_5_tree_Tree;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_4Tree_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_Tree(PyObject *o) {
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && (!PyType_IS_GC(Py_TYPE(o)) || !_PyGC_FINALIZED(o))) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
{
|
|
PyObject *etype, *eval, *etb;
|
|
PyErr_Fetch(&etype, &eval, &etb);
|
|
++Py_REFCNT(o);
|
|
__pyx_pw_7sklearn_4tree_5_tree_4Tree_3__dealloc__(o);
|
|
--Py_REFCNT(o);
|
|
PyErr_Restore(etype, eval, etb);
|
|
}
|
|
(*Py_TYPE(o)->tp_free)(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_classes(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_classes_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_children_left(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_13children_left_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_children_right(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_14children_right_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_feature(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_7feature_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_threshold(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_9threshold_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_impurity(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_8impurity_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_node_samples(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_14n_node_samples_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_weighted_n_node_samples(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_23weighted_n_node_samples_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_value(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_5value_1__get__(o);
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_features(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_10n_features_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_4Tree_n_features(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_10n_features_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_outputs(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_outputs_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_4Tree_n_outputs(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_9n_outputs_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_max_n_classes(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_13max_n_classes_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_4Tree_max_n_classes(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_13max_n_classes_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_max_depth(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_9max_depth_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_4Tree_max_depth(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_9max_depth_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_node_count(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_10node_count_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_4Tree_node_count(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_10node_count_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyObject *__pyx_getprop_7sklearn_4tree_5_tree_4Tree_capacity(PyObject *o, CYTHON_UNUSED void *x) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_8capacity_1__get__(o);
|
|
}
|
|
|
|
static int __pyx_setprop_7sklearn_4tree_5_tree_4Tree_capacity(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
|
|
if (v) {
|
|
return __pyx_pw_7sklearn_4tree_5_tree_4Tree_8capacity_3__set__(o, v);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_NotImplementedError, "__del__");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_Tree[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_5__reduce__, METH_NOARGS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4Tree_4__reduce__)},
|
|
{__Pyx_NAMESTR("__getstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_7__getstate__, METH_NOARGS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4Tree_6__getstate__)},
|
|
{__Pyx_NAMESTR("__setstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_9__setstate__, METH_O, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4Tree_8__setstate__)},
|
|
{__Pyx_NAMESTR("predict"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_11predict, METH_O, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4Tree_10predict)},
|
|
{__Pyx_NAMESTR("apply"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_13apply, METH_O, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4Tree_12apply)},
|
|
{__Pyx_NAMESTR("compute_feature_importances"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_4Tree_15compute_feature_importances, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4Tree_14compute_feature_importances)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static struct PyGetSetDef __pyx_getsets_7sklearn_4tree_5_tree_Tree[] = {
|
|
{(char *)"n_classes", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_classes, 0, 0, 0},
|
|
{(char *)"children_left", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_children_left, 0, 0, 0},
|
|
{(char *)"children_right", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_children_right, 0, 0, 0},
|
|
{(char *)"feature", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_feature, 0, 0, 0},
|
|
{(char *)"threshold", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_threshold, 0, 0, 0},
|
|
{(char *)"impurity", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_impurity, 0, 0, 0},
|
|
{(char *)"n_node_samples", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_node_samples, 0, 0, 0},
|
|
{(char *)"weighted_n_node_samples", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_weighted_n_node_samples, 0, 0, 0},
|
|
{(char *)"value", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_value, 0, 0, 0},
|
|
{(char *)"n_features", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_features, __pyx_setprop_7sklearn_4tree_5_tree_4Tree_n_features, 0, 0},
|
|
{(char *)"n_outputs", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_n_outputs, __pyx_setprop_7sklearn_4tree_5_tree_4Tree_n_outputs, 0, 0},
|
|
{(char *)"max_n_classes", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_max_n_classes, __pyx_setprop_7sklearn_4tree_5_tree_4Tree_max_n_classes, 0, 0},
|
|
{(char *)"max_depth", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_max_depth, __pyx_setprop_7sklearn_4tree_5_tree_4Tree_max_depth, 0, 0},
|
|
{(char *)"node_count", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_node_count, __pyx_setprop_7sklearn_4tree_5_tree_4Tree_node_count, 0, 0},
|
|
{(char *)"capacity", __pyx_getprop_7sklearn_4tree_5_tree_4Tree_capacity, __pyx_setprop_7sklearn_4tree_5_tree_4Tree_capacity, 0, 0},
|
|
{0, 0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_Tree = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.Tree"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_Tree), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Tree, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Array-based representation of a binary decision tree.\n\n The binary tree is represented as a number of parallel arrays. The i-th\n element of each array holds information about the node `i`. Node 0 is the\n tree's root. You can find a detailed description of all arrays in\n `_tree.pxd`. NOTE: Some of the arrays only apply to either leaves or split\n nodes, resp. In this case the values of nodes of the other type are\n arbitrary!\n\n Attributes\n ----------\n node_count : int\n The number of nodes (internal nodes + leaves) in the tree.\n\n capacity : int\n The current capacity (i.e., size) of the arrays, which is at least as\n great as `node_count`.\n\n max_depth : int\n The maximal depth of the tree.\n\n children_left : array of int, shape [node_count]\n children_left[i] holds the node id of the left child of node i.\n For leaves, children_left[i] == TREE_LEAF. Otherwise,\n children_left[i] > i. This child handles the case where\n X[:, feature[i]] <= threshold[i].\n\n children_right : array of int, shape [node_count]\n children_right[i] holds the node id of the right child of node i.\n For leaves, children_right[i] == TREE_LEAF. Otherwise,\n children_right[i] > i. This child handles the case where\n X[:, feature[i]] > threshold[i].\n\n feature : array of int, shape [node_count]\n feature[i] holds the feature to split on, for the internal node i.\n\n threshold : array of double, shape [node_count]\n threshold[i] holds the threshold for the internal node i.\n\n value : array of double, shape [node_count, n_outputs, max_n_classes]\n Contains the constant prediction value of each node.\n\n impurity : array of double, shape [node_count]\n impurity[i] holds the impurity (i.e., the value of the splitting\n criterion) at node i.\n\n n_node_samples : array of int, shape [node_count]\n n_node_samples[i] holds the number of training samples reaching node i.\n\n weighted_n_node_samples : array of int, shape [node_count]\n weighted_n_node_samples[i] holds the weighted number of training samples\n reaching node i.\n "), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_Tree, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
__pyx_getsets_7sklearn_4tree_5_tree_Tree, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_Tree, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder __pyx_vtable_7sklearn_4tree_5_tree_TreeBuilder;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_TreeBuilder(PyTypeObject *t, CYTHON_UNUSED PyObject *a, CYTHON_UNUSED PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *p;
|
|
PyObject *o;
|
|
if (likely((t->tp_flags & Py_TPFLAGS_IS_ABSTRACT) == 0)) {
|
|
o = (*t->tp_alloc)(t, 0);
|
|
} else {
|
|
o = (PyObject *) PyBaseObject_Type.tp_new(t, __pyx_empty_tuple, 0);
|
|
}
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)o);
|
|
p->__pyx_vtab = __pyx_vtabptr_7sklearn_4tree_5_tree_TreeBuilder;
|
|
p->splitter = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)Py_None); Py_INCREF(Py_None);
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_TreeBuilder(PyObject *o) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *p = (struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)o;
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && !_PyGC_FINALIZED(o)) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
PyObject_GC_UnTrack(o);
|
|
Py_CLEAR(p->splitter);
|
|
(*Py_TYPE(o)->tp_free)(o);
|
|
}
|
|
|
|
static int __pyx_tp_traverse_7sklearn_4tree_5_tree_TreeBuilder(PyObject *o, visitproc v, void *a) {
|
|
int e;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *p = (struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)o;
|
|
if (p->splitter) {
|
|
e = (*v)(((PyObject*)p->splitter), a); if (e) return e;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int __pyx_tp_clear_7sklearn_4tree_5_tree_TreeBuilder(PyObject *o) {
|
|
PyObject* tmp;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *p = (struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *)o;
|
|
tmp = ((PyObject*)p->splitter);
|
|
p->splitter = ((struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *)Py_None); Py_INCREF(Py_None);
|
|
Py_XDECREF(tmp);
|
|
return 0;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_TreeBuilder[] = {
|
|
{__Pyx_NAMESTR("build"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_11TreeBuilder_1build, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_11TreeBuilder_build)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_TreeBuilder = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.TreeBuilder"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_TreeBuilder, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Interface for different tree building strategies. "), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_TreeBuilder, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_TreeBuilder, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_TreeBuilder, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_TreeBuilder, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_ClassificationCriterion(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_Criterion(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)o);
|
|
p->__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion*)__pyx_vtabptr_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_ClassificationCriterion(PyObject *o) {
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && (!PyType_IS_GC(Py_TYPE(o)) || !_PyGC_FINALIZED(o))) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
{
|
|
PyObject *etype, *eval, *etb;
|
|
PyErr_Fetch(&etype, &eval, &etb);
|
|
++Py_REFCNT(o);
|
|
__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_3__dealloc__(o);
|
|
--Py_REFCNT(o);
|
|
PyErr_Restore(etype, eval, etb);
|
|
}
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Criterion(o);
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_ClassificationCriterion[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_5__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{__Pyx_NAMESTR("__getstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_7__getstate__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{__Pyx_NAMESTR("__setstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_23ClassificationCriterion_9__setstate__, METH_O, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.ClassificationCriterion"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Abstract criterion for classification."), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Entropy __pyx_vtable_7sklearn_4tree_5_tree_Entropy;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Entropy(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Entropy *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_ClassificationCriterion(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_Entropy *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion*)__pyx_vtabptr_7sklearn_4tree_5_tree_Entropy;
|
|
return o;
|
|
}
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_Entropy = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.Entropy"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_Entropy), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Cross Entropy impurity criteria.\n\n Let the target be a classification outcome taking values in 0, 1, ..., K-1.\n If node m represents a region Rm with Nm observations, then let\n\n pmk = 1/ Nm \\sum_{x_i in Rm} I(yi = k)\n\n be the proportion of class k observations in node m.\n\n The cross-entropy is then defined as\n\n cross-entropy = - \\sum_{k=0}^{K-1} pmk log(pmk)\n "), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_Entropy, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Gini __pyx_vtable_7sklearn_4tree_5_tree_Gini;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_Gini(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_Gini *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_ClassificationCriterion(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_Gini *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion*)__pyx_vtabptr_7sklearn_4tree_5_tree_Gini;
|
|
return o;
|
|
}
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_Gini = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.Gini"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_Gini), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Gini Index impurity criteria.\n\n Let the target be a classification outcome taking values in 0, 1, ..., K-1.\n If node m represents a region Rm with Nm observations, then let\n\n pmk = 1/ Nm \\sum_{x_i in Rm} I(yi = k)\n\n be the proportion of class k observations in node m.\n\n The Gini Index is then defined as:\n\n index = \\sum_{k=0}^{K-1} pmk (1 - pmk)\n = 1 - \\sum_{k=0}^{K-1} pmk ** 2\n "), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_Gini, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RegressionCriterion __pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_RegressionCriterion(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_Criterion(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)o);
|
|
p->__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion*)__pyx_vtabptr_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_RegressionCriterion(PyObject *o) {
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && (!PyType_IS_GC(Py_TYPE(o)) || !_PyGC_FINALIZED(o))) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
{
|
|
PyObject *etype, *eval, *etb;
|
|
PyErr_Fetch(&etype, &eval, &etb);
|
|
++Py_REFCNT(o);
|
|
__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_3__dealloc__(o);
|
|
--Py_REFCNT(o);
|
|
PyErr_Restore(etype, eval, etb);
|
|
}
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Criterion(o);
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_RegressionCriterion[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_5__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{__Pyx_NAMESTR("__getstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_7__getstate__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{__Pyx_NAMESTR("__setstate__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_19RegressionCriterion_9__setstate__, METH_O, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_RegressionCriterion = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.RegressionCriterion"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_RegressionCriterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Abstract criterion for regression.\n\n Computes variance of the target values left and right of the split point.\n Computation is linear in `n_samples` by using ::\n\n var = \\sum_i^n (y_i - y_bar) ** 2\n = (\\sum_i^n y_i ** 2) - n_samples y_bar ** 2\n "), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_RegressionCriterion, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_RegressionCriterion, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_MSE __pyx_vtable_7sklearn_4tree_5_tree_MSE;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_MSE(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_MSE *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_RegressionCriterion(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_MSE *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion*)__pyx_vtabptr_7sklearn_4tree_5_tree_MSE;
|
|
return o;
|
|
}
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_MSE = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.MSE"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_MSE), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_RegressionCriterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Mean squared error impurity criterion.\n\n MSE = var_left + var_right\n "), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_MSE, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_FriedmanMSE __pyx_vtable_7sklearn_4tree_5_tree_FriedmanMSE;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_FriedmanMSE(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_FriedmanMSE *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_MSE(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_FriedmanMSE *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion*)__pyx_vtabptr_7sklearn_4tree_5_tree_FriedmanMSE;
|
|
return o;
|
|
}
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_FriedmanMSE = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.FriedmanMSE"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_FriedmanMSE), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_RegressionCriterion, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
|
|
__Pyx_DOCSTR("Mean squared error impurity criterion with improvement score by Friedman\n\n Uses the formula (35) in Friedmans original Gradient Boosting paper:\n\n diff = mean_left - mean_right\n improvement = n_left * n_right * diff^2 / (n_left + n_right)\n "), /*tp_doc*/
|
|
0, /*tp_traverse*/
|
|
0, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
0, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_FriedmanMSE, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseDenseSplitter __pyx_vtable_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BaseDenseSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_Splitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter *)o);
|
|
p->__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_17BaseDenseSplitter_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_BaseDenseSplitter[] = {
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.BaseDenseSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_BaseDenseSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_BaseDenseSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_BaseDenseSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestSplitter __pyx_vtable_7sklearn_4tree_5_tree_BestSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BestSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_BaseDenseSplitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_BestSplitter;
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_BestSplitter[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_12BestSplitter_1__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_BestSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.BestSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_BestSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Splitter for finding the best split."), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_BestSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_BestSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RandomSplitter __pyx_vtable_7sklearn_4tree_5_tree_RandomSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_RandomSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_BaseDenseSplitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_RandomSplitter;
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_RandomSplitter[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_14RandomSplitter_1__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_RandomSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.RandomSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Splitter for finding the best random split."), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_RandomSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_RandomSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_PresortBestSplitter __pyx_vtable_7sklearn_4tree_5_tree_PresortBestSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_PresortBestSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_BaseDenseSplitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_PresortBestSplitter;
|
|
p->X_argsorted = ((PyArrayObject *)Py_None); Py_INCREF(Py_None);
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_PresortBestSplitter(PyObject *o) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *p = (struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)o;
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && !_PyGC_FINALIZED(o)) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
PyObject_GC_UnTrack(o);
|
|
{
|
|
PyObject *etype, *eval, *etb;
|
|
PyErr_Fetch(&etype, &eval, &etb);
|
|
++Py_REFCNT(o);
|
|
__pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_3__dealloc__(o);
|
|
--Py_REFCNT(o);
|
|
PyErr_Restore(etype, eval, etb);
|
|
}
|
|
Py_CLEAR(p->X_argsorted);
|
|
PyObject_GC_Track(o);
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter(o);
|
|
}
|
|
|
|
static int __pyx_tp_traverse_7sklearn_4tree_5_tree_PresortBestSplitter(PyObject *o, visitproc v, void *a) {
|
|
int e;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *p = (struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)o;
|
|
e = __pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter(o, v, a); if (e) return e;
|
|
if (p->X_argsorted) {
|
|
e = (*v)(((PyObject*)p->X_argsorted), a); if (e) return e;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int __pyx_tp_clear_7sklearn_4tree_5_tree_PresortBestSplitter(PyObject *o) {
|
|
PyObject* tmp;
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *p = (struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter *)o;
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter(o);
|
|
tmp = ((PyObject*)p->X_argsorted);
|
|
p->X_argsorted = ((PyArrayObject *)Py_None); Py_INCREF(Py_None);
|
|
Py_XDECREF(tmp);
|
|
return 0;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_PresortBestSplitter[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_19PresortBestSplitter_5__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.PresortBestSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_PresortBestSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_PresortBestSplitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Splitter for finding the best split, using presorting."), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_PresortBestSplitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_PresortBestSplitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_PresortBestSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_PresortBestSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BaseSparseSplitter __pyx_vtable_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BaseSparseSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_Splitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *)o);
|
|
p->__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_18BaseSparseSplitter_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_BaseSparseSplitter(PyObject *o) {
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
if (unlikely(Py_TYPE(o)->tp_finalize) && !_PyGC_FINALIZED(o)) {
|
|
if (PyObject_CallFinalizerFromDealloc(o)) return;
|
|
}
|
|
#endif
|
|
PyObject_GC_UnTrack(o);
|
|
{
|
|
PyObject *etype, *eval, *etb;
|
|
PyErr_Fetch(&etype, &eval, &etb);
|
|
++Py_REFCNT(o);
|
|
__pyx_pw_7sklearn_4tree_5_tree_18BaseSparseSplitter_3__dealloc__(o);
|
|
--Py_REFCNT(o);
|
|
PyErr_Restore(etype, eval, etb);
|
|
}
|
|
PyObject_GC_Track(o);
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Splitter(o);
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_BaseSparseSplitter[] = {
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.BaseSparseSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_BaseSparseSplitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
0, /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_BaseSparseSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_BaseSparseSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestSparseSplitter __pyx_vtable_7sklearn_4tree_5_tree_BestSparseSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BestSparseSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_BaseSparseSplitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_BestSparseSplitter;
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_BestSparseSplitter[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_18BestSparseSplitter_1__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.BestSparseSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_BestSparseSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_BaseSparseSplitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Splitter for finding the best split, using the sparse data."), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_BestSparseSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_BestSparseSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RandomSparseSplitter __pyx_vtable_7sklearn_4tree_5_tree_RandomSparseSplitter;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_RandomSparseSplitter(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_BaseSparseSplitter(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter *)o);
|
|
p->__pyx_base.__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Splitter*)__pyx_vtabptr_7sklearn_4tree_5_tree_RandomSparseSplitter;
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_RandomSparseSplitter[] = {
|
|
{__Pyx_NAMESTR("__reduce__"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_20RandomSparseSplitter_1__reduce__, METH_NOARGS, __Pyx_DOCSTR(0)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.RandomSparseSplitter"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_RandomSparseSplitter), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_BaseSparseSplitter, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Splitter for finding a random split, using the sparse data."), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_Splitter, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_Splitter, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_RandomSparseSplitter, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_RandomSparseSplitter, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_DepthFirstTreeBuilder __pyx_vtable_7sklearn_4tree_5_tree_DepthFirstTreeBuilder;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_DepthFirstTreeBuilder(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_TreeBuilder(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder *)o);
|
|
p->__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder*)__pyx_vtabptr_7sklearn_4tree_5_tree_DepthFirstTreeBuilder;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_DepthFirstTreeBuilder[] = {
|
|
{__Pyx_NAMESTR("build"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_3build, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_2build)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.DepthFirstTreeBuilder"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_DepthFirstTreeBuilder), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_TreeBuilder, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Build a decision tree in depth-first fashion."), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_TreeBuilder, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_TreeBuilder, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_DepthFirstTreeBuilder, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_DepthFirstTreeBuilder, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_BestFirstTreeBuilder __pyx_vtable_7sklearn_4tree_5_tree_BestFirstTreeBuilder;
|
|
|
|
static PyObject *__pyx_tp_new_7sklearn_4tree_5_tree_BestFirstTreeBuilder(PyTypeObject *t, PyObject *a, PyObject *k) {
|
|
struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *p;
|
|
PyObject *o = __pyx_tp_new_7sklearn_4tree_5_tree_TreeBuilder(t, a, k);
|
|
if (unlikely(!o)) return 0;
|
|
p = ((struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *)o);
|
|
p->__pyx_base.__pyx_vtab = (struct __pyx_vtabstruct_7sklearn_4tree_5_tree_TreeBuilder*)__pyx_vtabptr_7sklearn_4tree_5_tree_BestFirstTreeBuilder;
|
|
if (unlikely(__pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_1__cinit__(o, a, k) < 0)) {
|
|
Py_DECREF(o); o = 0;
|
|
}
|
|
return o;
|
|
}
|
|
|
|
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_BestFirstTreeBuilder[] = {
|
|
{__Pyx_NAMESTR("build"), (PyCFunction)__pyx_pw_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_3build, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_2build)},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static PyTypeObject __pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder = {
|
|
PyVarObject_HEAD_INIT(0, 0)
|
|
__Pyx_NAMESTR("sklearn.tree._tree.BestFirstTreeBuilder"), /*tp_name*/
|
|
sizeof(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
__pyx_tp_dealloc_7sklearn_4tree_5_tree_TreeBuilder, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
0, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
0, /*tp_compare*/
|
|
#else
|
|
0, /*reserved*/
|
|
#endif
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
0, /*tp_call*/
|
|
0, /*tp_str*/
|
|
0, /*tp_getattro*/
|
|
0, /*tp_setattro*/
|
|
0, /*tp_as_buffer*/
|
|
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_HAVE_VERSION_TAG|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
|
|
__Pyx_DOCSTR("Build a decision tree in best-first fashion.\n\n The best node to expand is given by the node at the frontier that has the\n highest impurity improvement.\n\n NOTE: this TreeBuilder will ignore ``tree.max_depth`` .\n "), /*tp_doc*/
|
|
__pyx_tp_traverse_7sklearn_4tree_5_tree_TreeBuilder, /*tp_traverse*/
|
|
__pyx_tp_clear_7sklearn_4tree_5_tree_TreeBuilder, /*tp_clear*/
|
|
0, /*tp_richcompare*/
|
|
0, /*tp_weaklistoffset*/
|
|
0, /*tp_iter*/
|
|
0, /*tp_iternext*/
|
|
__pyx_methods_7sklearn_4tree_5_tree_BestFirstTreeBuilder, /*tp_methods*/
|
|
0, /*tp_members*/
|
|
0, /*tp_getset*/
|
|
0, /*tp_base*/
|
|
0, /*tp_dict*/
|
|
0, /*tp_descr_get*/
|
|
0, /*tp_descr_set*/
|
|
0, /*tp_dictoffset*/
|
|
0, /*tp_init*/
|
|
0, /*tp_alloc*/
|
|
__pyx_tp_new_7sklearn_4tree_5_tree_BestFirstTreeBuilder, /*tp_new*/
|
|
0, /*tp_free*/
|
|
0, /*tp_is_gc*/
|
|
0, /*tp_bases*/
|
|
0, /*tp_mro*/
|
|
0, /*tp_cache*/
|
|
0, /*tp_subclasses*/
|
|
0, /*tp_weaklist*/
|
|
0, /*tp_del*/
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
0, /*tp_version_tag*/
|
|
#endif
|
|
#if PY_VERSION_HEX >= 0x030400a1
|
|
0, /*tp_finalize*/
|
|
#endif
|
|
};
|
|
|
|
static PyMethodDef __pyx_methods[] = {
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
static struct PyModuleDef __pyx_moduledef = {
|
|
#if PY_VERSION_HEX < 0x03020000
|
|
{ PyObject_HEAD_INIT(NULL) NULL, 0, NULL },
|
|
#else
|
|
PyModuleDef_HEAD_INIT,
|
|
#endif
|
|
__Pyx_NAMESTR("_tree"),
|
|
0, /* m_doc */
|
|
-1, /* m_size */
|
|
__pyx_methods /* m_methods */,
|
|
NULL, /* m_reload */
|
|
NULL, /* m_traverse */
|
|
NULL, /* m_clear */
|
|
NULL /* m_free */
|
|
};
|
|
#endif
|
|
|
|
static __Pyx_StringTabEntry __pyx_string_tab[] = {
|
|
{&__pyx_n_s_C, __pyx_k_C, sizeof(__pyx_k_C), 0, 0, 1, 1},
|
|
{&__pyx_n_s_DOUBLE, __pyx_k_DOUBLE, sizeof(__pyx_k_DOUBLE), 0, 0, 1, 1},
|
|
{&__pyx_n_s_DTYPE, __pyx_k_DTYPE, sizeof(__pyx_k_DTYPE), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_Did_not_recognise_loaded_array_l, __pyx_k_Did_not_recognise_loaded_array_l, sizeof(__pyx_k_Did_not_recognise_loaded_array_l), 0, 0, 1, 0},
|
|
{&__pyx_kp_u_Format_string_allocated_too_shor, __pyx_k_Format_string_allocated_too_shor, sizeof(__pyx_k_Format_string_allocated_too_shor), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_Format_string_allocated_too_shor_2, __pyx_k_Format_string_allocated_too_shor_2, sizeof(__pyx_k_Format_string_allocated_too_shor_2), 0, 1, 0, 0},
|
|
{&__pyx_n_s_MemoryError, __pyx_k_MemoryError, sizeof(__pyx_k_MemoryError), 0, 0, 1, 1},
|
|
{&__pyx_n_s_NODE_DTYPE, __pyx_k_NODE_DTYPE, sizeof(__pyx_k_NODE_DTYPE), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_No_support_for_np_int64_index_ba, __pyx_k_No_support_for_np_int64_index_ba, sizeof(__pyx_k_No_support_for_np_int64_index_ba), 0, 0, 1, 0},
|
|
{&__pyx_kp_u_Non_native_byte_order_not_suppor, __pyx_k_Non_native_byte_order_not_suppor, sizeof(__pyx_k_Non_native_byte_order_not_suppor), 0, 1, 0, 0},
|
|
{&__pyx_n_s_RuntimeError, __pyx_k_RuntimeError, sizeof(__pyx_k_RuntimeError), 0, 0, 1, 1},
|
|
{&__pyx_n_s_TREE_LEAF, __pyx_k_TREE_LEAF, sizeof(__pyx_k_TREE_LEAF), 0, 0, 1, 1},
|
|
{&__pyx_n_s_TREE_UNDEFINED, __pyx_k_TREE_UNDEFINED, sizeof(__pyx_k_TREE_UNDEFINED), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_Users_ajoly_git_scikit_learn_sk, __pyx_k_Users_ajoly_git_scikit_learn_sk, sizeof(__pyx_k_Users_ajoly_git_scikit_learn_sk), 0, 0, 1, 0},
|
|
{&__pyx_n_s_ValueError, __pyx_k_ValueError, sizeof(__pyx_k_ValueError), 0, 0, 1, 1},
|
|
{&__pyx_n_s_X, __pyx_k_X, sizeof(__pyx_k_X), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_X_dtype_should_be_np_float32_got, __pyx_k_X_dtype_should_be_np_float32_got, sizeof(__pyx_k_X_dtype_should_be_np_float32_got), 0, 0, 1, 0},
|
|
{&__pyx_kp_s_X_should_be_in_csc_format, __pyx_k_X_should_be_in_csc_format, sizeof(__pyx_k_X_should_be_in_csc_format), 0, 0, 1, 0},
|
|
{&__pyx_kp_s_X_should_be_in_csr_matrix_format, __pyx_k_X_should_be_in_csr_matrix_format, sizeof(__pyx_k_X_should_be_in_csr_matrix_format), 0, 0, 1, 0},
|
|
{&__pyx_kp_s_X_should_be_in_np_ndarray_format, __pyx_k_X_should_be_in_np_ndarray_format, sizeof(__pyx_k_X_should_be_in_np_ndarray_format), 0, 0, 1, 0},
|
|
{&__pyx_kp_s_You_have_loaded_Tree_version_whi, __pyx_k_You_have_loaded_Tree_version_whi, sizeof(__pyx_k_You_have_loaded_Tree_version_whi), 0, 0, 1, 0},
|
|
{&__pyx_n_s_apply, __pyx_k_apply, sizeof(__pyx_k_apply), 0, 0, 1, 1},
|
|
{&__pyx_n_s_argsort, __pyx_k_argsort, sizeof(__pyx_k_argsort), 0, 0, 1, 1},
|
|
{&__pyx_n_s_asarray, __pyx_k_asarray, sizeof(__pyx_k_asarray), 0, 0, 1, 1},
|
|
{&__pyx_n_s_ascontiguousarray, __pyx_k_ascontiguousarray, sizeof(__pyx_k_ascontiguousarray), 0, 0, 1, 1},
|
|
{&__pyx_n_s_asfortranarray, __pyx_k_asfortranarray, sizeof(__pyx_k_asfortranarray), 0, 0, 1, 1},
|
|
{&__pyx_n_s_axis, __pyx_k_axis, sizeof(__pyx_k_axis), 0, 0, 1, 1},
|
|
{&__pyx_n_s_build, __pyx_k_build, sizeof(__pyx_k_build), 0, 0, 1, 1},
|
|
{&__pyx_n_s_c_contiguous, __pyx_k_c_contiguous, sizeof(__pyx_k_c_contiguous), 0, 0, 1, 1},
|
|
{&__pyx_n_s_clip, __pyx_k_clip, sizeof(__pyx_k_clip), 0, 0, 1, 1},
|
|
{&__pyx_n_s_compute_feature_importances, __pyx_k_compute_feature_importances, sizeof(__pyx_k_compute_feature_importances), 0, 0, 1, 1},
|
|
{&__pyx_n_s_contiguous, __pyx_k_contiguous, sizeof(__pyx_k_contiguous), 0, 0, 1, 1},
|
|
{&__pyx_n_s_copy, __pyx_k_copy, sizeof(__pyx_k_copy), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_could_not_allocate_d_bytes, __pyx_k_could_not_allocate_d_bytes, sizeof(__pyx_k_could_not_allocate_d_bytes), 0, 0, 1, 0},
|
|
{&__pyx_kp_s_could_not_allocate_d_d_bytes, __pyx_k_could_not_allocate_d_d_bytes, sizeof(__pyx_k_could_not_allocate_d_d_bytes), 0, 0, 1, 0},
|
|
{&__pyx_n_s_criterion, __pyx_k_criterion, sizeof(__pyx_k_criterion), 0, 0, 1, 1},
|
|
{&__pyx_n_s_csc_matrix, __pyx_k_csc_matrix, sizeof(__pyx_k_csc_matrix), 0, 0, 1, 1},
|
|
{&__pyx_n_s_csr_matrix, __pyx_k_csr_matrix, sizeof(__pyx_k_csr_matrix), 0, 0, 1, 1},
|
|
{&__pyx_n_s_data, __pyx_k_data, sizeof(__pyx_k_data), 0, 0, 1, 1},
|
|
{&__pyx_n_s_dtype, __pyx_k_dtype, sizeof(__pyx_k_dtype), 0, 0, 1, 1},
|
|
{&__pyx_n_s_feature, __pyx_k_feature, sizeof(__pyx_k_feature), 0, 0, 1, 1},
|
|
{&__pyx_n_s_flags, __pyx_k_flags, sizeof(__pyx_k_flags), 0, 0, 1, 1},
|
|
{&__pyx_n_s_float32, __pyx_k_float32, sizeof(__pyx_k_float32), 0, 0, 1, 1},
|
|
{&__pyx_n_s_float64, __pyx_k_float64, sizeof(__pyx_k_float64), 0, 0, 1, 1},
|
|
{&__pyx_n_s_formats, __pyx_k_formats, sizeof(__pyx_k_formats), 0, 0, 1, 1},
|
|
{&__pyx_n_s_getstate, __pyx_k_getstate, sizeof(__pyx_k_getstate), 0, 0, 1, 1},
|
|
{&__pyx_n_s_import, __pyx_k_import, sizeof(__pyx_k_import), 0, 0, 1, 1},
|
|
{&__pyx_n_s_impurity, __pyx_k_impurity, sizeof(__pyx_k_impurity), 0, 0, 1, 1},
|
|
{&__pyx_n_s_indices, __pyx_k_indices, sizeof(__pyx_k_indices), 0, 0, 1, 1},
|
|
{&__pyx_n_s_indptr, __pyx_k_indptr, sizeof(__pyx_k_indptr), 0, 0, 1, 1},
|
|
{&__pyx_n_s_inf, __pyx_k_inf, sizeof(__pyx_k_inf), 0, 0, 1, 1},
|
|
{&__pyx_n_s_int32, __pyx_k_int32, sizeof(__pyx_k_int32), 0, 0, 1, 1},
|
|
{&__pyx_n_s_intp, __pyx_k_intp, sizeof(__pyx_k_intp), 0, 0, 1, 1},
|
|
{&__pyx_n_s_issparse, __pyx_k_issparse, sizeof(__pyx_k_issparse), 0, 0, 1, 1},
|
|
{&__pyx_n_s_itemsize, __pyx_k_itemsize, sizeof(__pyx_k_itemsize), 0, 0, 1, 1},
|
|
{&__pyx_n_s_left_child, __pyx_k_left_child, sizeof(__pyx_k_left_child), 0, 0, 1, 1},
|
|
{&__pyx_n_s_main, __pyx_k_main, sizeof(__pyx_k_main), 0, 0, 1, 1},
|
|
{&__pyx_n_s_max, __pyx_k_max, sizeof(__pyx_k_max), 0, 0, 1, 1},
|
|
{&__pyx_n_s_max_depth, __pyx_k_max_depth, sizeof(__pyx_k_max_depth), 0, 0, 1, 1},
|
|
{&__pyx_n_s_max_features, __pyx_k_max_features, sizeof(__pyx_k_max_features), 0, 0, 1, 1},
|
|
{&__pyx_n_s_max_leaf_nodes, __pyx_k_max_leaf_nodes, sizeof(__pyx_k_max_leaf_nodes), 0, 0, 1, 1},
|
|
{&__pyx_n_s_min_samples_leaf, __pyx_k_min_samples_leaf, sizeof(__pyx_k_min_samples_leaf), 0, 0, 1, 1},
|
|
{&__pyx_n_s_min_samples_split, __pyx_k_min_samples_split, sizeof(__pyx_k_min_samples_split), 0, 0, 1, 1},
|
|
{&__pyx_n_s_min_weight_leaf, __pyx_k_min_weight_leaf, sizeof(__pyx_k_min_weight_leaf), 0, 0, 1, 1},
|
|
{&__pyx_n_s_mode, __pyx_k_mode, sizeof(__pyx_k_mode), 0, 0, 1, 1},
|
|
{&__pyx_n_s_n_classes, __pyx_k_n_classes, sizeof(__pyx_k_n_classes), 0, 0, 1, 1},
|
|
{&__pyx_n_s_n_features, __pyx_k_n_features, sizeof(__pyx_k_n_features), 0, 0, 1, 1},
|
|
{&__pyx_n_s_n_node_samples, __pyx_k_n_node_samples, sizeof(__pyx_k_n_node_samples), 0, 0, 1, 1},
|
|
{&__pyx_n_s_n_outputs, __pyx_k_n_outputs, sizeof(__pyx_k_n_outputs), 0, 0, 1, 1},
|
|
{&__pyx_n_s_names, __pyx_k_names, sizeof(__pyx_k_names), 0, 0, 1, 1},
|
|
{&__pyx_kp_u_ndarray_is_not_C_contiguous, __pyx_k_ndarray_is_not_C_contiguous, sizeof(__pyx_k_ndarray_is_not_C_contiguous), 0, 1, 0, 0},
|
|
{&__pyx_kp_u_ndarray_is_not_Fortran_contiguou, __pyx_k_ndarray_is_not_Fortran_contiguou, sizeof(__pyx_k_ndarray_is_not_Fortran_contiguou), 0, 1, 0, 0},
|
|
{&__pyx_n_s_ndim, __pyx_k_ndim, sizeof(__pyx_k_ndim), 0, 0, 1, 1},
|
|
{&__pyx_n_s_node_count, __pyx_k_node_count, sizeof(__pyx_k_node_count), 0, 0, 1, 1},
|
|
{&__pyx_n_s_nodes, __pyx_k_nodes, sizeof(__pyx_k_nodes), 0, 0, 1, 1},
|
|
{&__pyx_n_s_normalize, __pyx_k_normalize, sizeof(__pyx_k_normalize), 0, 0, 1, 1},
|
|
{&__pyx_n_s_np, __pyx_k_np, sizeof(__pyx_k_np), 0, 0, 1, 1},
|
|
{&__pyx_n_s_numpy, __pyx_k_numpy, sizeof(__pyx_k_numpy), 0, 0, 1, 1},
|
|
{&__pyx_n_s_offsets, __pyx_k_offsets, sizeof(__pyx_k_offsets), 0, 0, 1, 1},
|
|
{&__pyx_n_s_order, __pyx_k_order, sizeof(__pyx_k_order), 0, 0, 1, 1},
|
|
{&__pyx_n_s_p, __pyx_k_p, sizeof(__pyx_k_p), 0, 0, 1, 1},
|
|
{&__pyx_n_s_predict, __pyx_k_predict, sizeof(__pyx_k_predict), 0, 0, 1, 1},
|
|
{&__pyx_n_s_pyx_getbuffer, __pyx_k_pyx_getbuffer, sizeof(__pyx_k_pyx_getbuffer), 0, 0, 1, 1},
|
|
{&__pyx_n_s_pyx_releasebuffer, __pyx_k_pyx_releasebuffer, sizeof(__pyx_k_pyx_releasebuffer), 0, 0, 1, 1},
|
|
{&__pyx_n_s_pyx_vtable, __pyx_k_pyx_vtable, sizeof(__pyx_k_pyx_vtable), 0, 0, 1, 1},
|
|
{&__pyx_n_s_randint, __pyx_k_randint, sizeof(__pyx_k_randint), 0, 0, 1, 1},
|
|
{&__pyx_n_s_random_state, __pyx_k_random_state, sizeof(__pyx_k_random_state), 0, 0, 1, 1},
|
|
{&__pyx_n_s_range, __pyx_k_range, sizeof(__pyx_k_range), 0, 0, 1, 1},
|
|
{&__pyx_n_s_realloc_test, __pyx_k_realloc_test, sizeof(__pyx_k_realloc_test), 0, 0, 1, 1},
|
|
{&__pyx_n_s_reshape, __pyx_k_reshape, sizeof(__pyx_k_reshape), 0, 0, 1, 1},
|
|
{&__pyx_kp_s_resizing_tree_to_d, __pyx_k_resizing_tree_to_d, sizeof(__pyx_k_resizing_tree_to_d), 0, 0, 1, 0},
|
|
{&__pyx_n_s_right_child, __pyx_k_right_child, sizeof(__pyx_k_right_child), 0, 0, 1, 1},
|
|
{&__pyx_n_s_sample_weight, __pyx_k_sample_weight, sizeof(__pyx_k_sample_weight), 0, 0, 1, 1},
|
|
{&__pyx_n_s_scipy_sparse, __pyx_k_scipy_sparse, sizeof(__pyx_k_scipy_sparse), 0, 0, 1, 1},
|
|
{&__pyx_n_s_shape, __pyx_k_shape, sizeof(__pyx_k_shape), 0, 0, 1, 1},
|
|
{&__pyx_n_s_sklearn_tree__tree, __pyx_k_sklearn_tree__tree, sizeof(__pyx_k_sklearn_tree__tree), 0, 0, 1, 1},
|
|
{&__pyx_n_s_sort_indices, __pyx_k_sort_indices, sizeof(__pyx_k_sort_indices), 0, 0, 1, 1},
|
|
{&__pyx_n_s_splitter, __pyx_k_splitter, sizeof(__pyx_k_splitter), 0, 0, 1, 1},
|
|
{&__pyx_n_s_strides, __pyx_k_strides, sizeof(__pyx_k_strides), 0, 0, 1, 1},
|
|
{&__pyx_n_s_sum, __pyx_k_sum, sizeof(__pyx_k_sum), 0, 0, 1, 1},
|
|
{&__pyx_n_s_take, __pyx_k_take, sizeof(__pyx_k_take), 0, 0, 1, 1},
|
|
{&__pyx_n_s_test, __pyx_k_test, sizeof(__pyx_k_test), 0, 0, 1, 1},
|
|
{&__pyx_n_s_threshold, __pyx_k_threshold, sizeof(__pyx_k_threshold), 0, 0, 1, 1},
|
|
{&__pyx_n_s_tocsc, __pyx_k_tocsc, sizeof(__pyx_k_tocsc), 0, 0, 1, 1},
|
|
{&__pyx_n_s_tree, __pyx_k_tree, sizeof(__pyx_k_tree), 0, 0, 1, 1},
|
|
{&__pyx_kp_u_unknown_dtype_code_in_numpy_pxd, __pyx_k_unknown_dtype_code_in_numpy_pxd, sizeof(__pyx_k_unknown_dtype_code_in_numpy_pxd), 0, 1, 0, 0},
|
|
{&__pyx_n_s_values, __pyx_k_values, sizeof(__pyx_k_values), 0, 0, 1, 1},
|
|
{&__pyx_n_s_weighted_n_node_samples, __pyx_k_weighted_n_node_samples, sizeof(__pyx_k_weighted_n_node_samples), 0, 0, 1, 1},
|
|
{&__pyx_n_s_y, __pyx_k_y, sizeof(__pyx_k_y), 0, 0, 1, 1},
|
|
{&__pyx_n_s_zeros, __pyx_k_zeros, sizeof(__pyx_k_zeros), 0, 0, 1, 1},
|
|
{0, 0, 0, 0, 0, 0, 0}
|
|
};
|
|
static int __Pyx_InitCachedBuiltins(void) {
|
|
__pyx_builtin_range = __Pyx_GetBuiltinName(__pyx_n_s_range); if (!__pyx_builtin_range) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 187; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_builtin_MemoryError = __Pyx_GetBuiltinName(__pyx_n_s_MemoryError); if (!__pyx_builtin_MemoryError) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 205; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_builtin_ValueError = __Pyx_GetBuiltinName(__pyx_n_s_ValueError); if (!__pyx_builtin_ValueError) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1851; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_builtin_RuntimeError = __Pyx_GetBuiltinName(__pyx_n_s_RuntimeError); if (!__pyx_builtin_RuntimeError) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 799; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
return 0;
|
|
__pyx_L1_error:;
|
|
return -1;
|
|
}
|
|
|
|
static int __Pyx_InitCachedConstants(void) {
|
|
__Pyx_RefNannyDeclarations
|
|
__Pyx_RefNannySetupContext("__Pyx_InitCachedConstants", 0);
|
|
|
|
/* "sklearn/tree/_tree.pyx":1851
|
|
*
|
|
* if not isinstance(X, csc_matrix):
|
|
* raise ValueError("X should be in csc format") # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef SIZE_t* samples = self.samples
|
|
*/
|
|
__pyx_tuple_ = PyTuple_Pack(1, __pyx_kp_s_X_should_be_in_csc_format); if (unlikely(!__pyx_tuple_)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1851; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple_);
|
|
__Pyx_GIVEREF(__pyx_tuple_);
|
|
|
|
/* "sklearn/tree/_tree.pyx":2585
|
|
*
|
|
* if X.indices.dtype != np.int32 or X.indptr.dtype != np.int32:
|
|
* raise ValueError("No support for np.int64 index based " # <<<<<<<<<<<<<<
|
|
* "sparse matrices")
|
|
*
|
|
*/
|
|
__pyx_tuple__2 = PyTuple_Pack(1, __pyx_kp_s_No_support_for_np_int64_index_ba); if (unlikely(!__pyx_tuple__2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2585; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__2);
|
|
__Pyx_GIVEREF(__pyx_tuple__2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3099
|
|
*
|
|
* if 'nodes' not in d:
|
|
* raise ValueError('You have loaded Tree version which ' # <<<<<<<<<<<<<<
|
|
* 'cannot be imported')
|
|
*
|
|
*/
|
|
__pyx_tuple__3 = PyTuple_Pack(1, __pyx_kp_s_You_have_loaded_Tree_version_whi); if (unlikely(!__pyx_tuple__3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3099; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__3);
|
|
__Pyx_GIVEREF(__pyx_tuple__3);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3113
|
|
* not value_ndarray.flags.c_contiguous or
|
|
* value_ndarray.dtype != np.float64):
|
|
* raise ValueError('Did not recognise loaded array layout') # <<<<<<<<<<<<<<
|
|
*
|
|
* self.capacity = node_ndarray.shape[0]
|
|
*/
|
|
__pyx_tuple__4 = PyTuple_Pack(1, __pyx_kp_s_Did_not_recognise_loaded_array_l); if (unlikely(!__pyx_tuple__4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__4);
|
|
__Pyx_GIVEREF(__pyx_tuple__4);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":215
|
|
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_C_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not C contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
*/
|
|
__pyx_tuple__6 = PyTuple_Pack(1, __pyx_kp_u_ndarray_is_not_C_contiguous); if (unlikely(!__pyx_tuple__6)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__6);
|
|
__Pyx_GIVEREF(__pyx_tuple__6);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":219
|
|
* if ((flags & pybuf.PyBUF_F_CONTIGUOUS == pybuf.PyBUF_F_CONTIGUOUS)
|
|
* and not PyArray_CHKFLAGS(self, NPY_F_CONTIGUOUS)):
|
|
* raise ValueError(u"ndarray is not Fortran contiguous") # <<<<<<<<<<<<<<
|
|
*
|
|
* info.buf = PyArray_DATA(self)
|
|
*/
|
|
__pyx_tuple__7 = PyTuple_Pack(1, __pyx_kp_u_ndarray_is_not_Fortran_contiguou); if (unlikely(!__pyx_tuple__7)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 219; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__7);
|
|
__Pyx_GIVEREF(__pyx_tuple__7);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":257
|
|
* if ((descr.byteorder == c'>' and little_endian) or
|
|
* (descr.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* if t == NPY_BYTE: f = "b"
|
|
* elif t == NPY_UBYTE: f = "B"
|
|
*/
|
|
__pyx_tuple__8 = PyTuple_Pack(1, __pyx_kp_u_Non_native_byte_order_not_suppor); if (unlikely(!__pyx_tuple__8)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 257; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__8);
|
|
__Pyx_GIVEREF(__pyx_tuple__8);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":799
|
|
*
|
|
* if (end - f) - <int>(new_offset - offset[0]) < 15:
|
|
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd") # <<<<<<<<<<<<<<
|
|
*
|
|
* if ((child.byteorder == c'>' and little_endian) or
|
|
*/
|
|
__pyx_tuple__9 = PyTuple_Pack(1, __pyx_kp_u_Format_string_allocated_too_shor); if (unlikely(!__pyx_tuple__9)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 799; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__9);
|
|
__Pyx_GIVEREF(__pyx_tuple__9);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":803
|
|
* if ((child.byteorder == c'>' and little_endian) or
|
|
* (child.byteorder == c'<' and not little_endian)):
|
|
* raise ValueError(u"Non-native byte order not supported") # <<<<<<<<<<<<<<
|
|
* # One could encode it in the format string and have Cython
|
|
* # complain instead, BUT: < and > in format strings also imply
|
|
*/
|
|
__pyx_tuple__10 = PyTuple_Pack(1, __pyx_kp_u_Non_native_byte_order_not_suppor); if (unlikely(!__pyx_tuple__10)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 803; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__10);
|
|
__Pyx_GIVEREF(__pyx_tuple__10);
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":823
|
|
* t = child.type_num
|
|
* if end - f < 5:
|
|
* raise RuntimeError(u"Format string allocated too short.") # <<<<<<<<<<<<<<
|
|
*
|
|
* # Until ticket #99 is fixed, use integers to avoid warnings
|
|
*/
|
|
__pyx_tuple__11 = PyTuple_Pack(1, __pyx_kp_u_Format_string_allocated_too_shor_2); if (unlikely(!__pyx_tuple__11)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__11);
|
|
__Pyx_GIVEREF(__pyx_tuple__11);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3446
|
|
*
|
|
*
|
|
* def _realloc_test(): # <<<<<<<<<<<<<<
|
|
* # Helper for tests. Tries to allocate <size_t>(-1) / 2 * sizeof(size_t)
|
|
* # bytes, which will always overflow.
|
|
*/
|
|
__pyx_tuple__12 = PyTuple_Pack(1, __pyx_n_s_p); if (unlikely(!__pyx_tuple__12)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3446; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_tuple__12);
|
|
__Pyx_GIVEREF(__pyx_tuple__12);
|
|
__pyx_codeobj__13 = (PyObject*)__Pyx_PyCode_New(0, 0, 1, 0, 0, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__12, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_Users_ajoly_git_scikit_learn_sk, __pyx_n_s_realloc_test, 3446, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__13)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3446; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_RefNannyFinishContext();
|
|
return 0;
|
|
__pyx_L1_error:;
|
|
__Pyx_RefNannyFinishContext();
|
|
return -1;
|
|
}
|
|
|
|
static int __Pyx_InitGlobals(void) {
|
|
if (__Pyx_InitStrings(__pyx_string_tab) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
__pyx_int_0 = PyInt_FromLong(0); if (unlikely(!__pyx_int_0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_int_1 = PyInt_FromLong(1); if (unlikely(!__pyx_int_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_int_neg_1 = PyInt_FromLong(-1); if (unlikely(!__pyx_int_neg_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_int_neg_2 = PyInt_FromLong(-2); if (unlikely(!__pyx_int_neg_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
return 0;
|
|
__pyx_L1_error:;
|
|
return -1;
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
PyMODINIT_FUNC init_tree(void); /*proto*/
|
|
PyMODINIT_FUNC init_tree(void)
|
|
#else
|
|
PyMODINIT_FUNC PyInit__tree(void); /*proto*/
|
|
PyMODINIT_FUNC PyInit__tree(void)
|
|
#endif
|
|
{
|
|
PyObject *__pyx_t_1 = NULL;
|
|
PyObject *__pyx_t_2 = NULL;
|
|
double __pyx_t_3;
|
|
__pyx_t_7sklearn_4tree_5_tree_SIZE_t __pyx_t_4;
|
|
PyObject *__pyx_t_5 = NULL;
|
|
PyObject *__pyx_t_6 = NULL;
|
|
PyObject *__pyx_t_7 = NULL;
|
|
PyObject *__pyx_t_8 = NULL;
|
|
PyObject *__pyx_t_9 = NULL;
|
|
PyObject *__pyx_t_10 = NULL;
|
|
PyObject *__pyx_t_11 = NULL;
|
|
int __pyx_lineno = 0;
|
|
const char *__pyx_filename = NULL;
|
|
int __pyx_clineno = 0;
|
|
__Pyx_RefNannyDeclarations
|
|
#if CYTHON_REFNANNY
|
|
__Pyx_RefNanny = __Pyx_RefNannyImportAPI("refnanny");
|
|
if (!__Pyx_RefNanny) {
|
|
PyErr_Clear();
|
|
__Pyx_RefNanny = __Pyx_RefNannyImportAPI("Cython.Runtime.refnanny");
|
|
if (!__Pyx_RefNanny)
|
|
Py_FatalError("failed to import 'refnanny' module");
|
|
}
|
|
#endif
|
|
__Pyx_RefNannySetupContext("PyMODINIT_FUNC PyInit__tree(void)", 0);
|
|
if ( __Pyx_check_binary_version() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_empty_tuple = PyTuple_New(0); if (unlikely(!__pyx_empty_tuple)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_empty_bytes = PyBytes_FromStringAndSize("", 0); if (unlikely(!__pyx_empty_bytes)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#ifdef __Pyx_CyFunction_USED
|
|
if (__Pyx_CyFunction_init() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
#ifdef __Pyx_FusedFunction_USED
|
|
if (__pyx_FusedFunction_init() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
#ifdef __Pyx_Generator_USED
|
|
if (__pyx_Generator_init() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
/*--- Library function declarations ---*/
|
|
/*--- Threads initialization code ---*/
|
|
#if defined(__PYX_FORCE_INIT_THREADS) && __PYX_FORCE_INIT_THREADS
|
|
#ifdef WITH_THREAD /* Python build with threading support? */
|
|
PyEval_InitThreads();
|
|
#endif
|
|
#endif
|
|
/*--- Module creation code ---*/
|
|
#if PY_MAJOR_VERSION < 3
|
|
__pyx_m = Py_InitModule4(__Pyx_NAMESTR("_tree"), __pyx_methods, 0, 0, PYTHON_API_VERSION); Py_XINCREF(__pyx_m);
|
|
#else
|
|
__pyx_m = PyModule_Create(&__pyx_moduledef);
|
|
#endif
|
|
if (unlikely(!__pyx_m)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_d = PyModule_GetDict(__pyx_m); if (unlikely(!__pyx_d)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
Py_INCREF(__pyx_d);
|
|
__pyx_b = PyImport_AddModule(__Pyx_NAMESTR(__Pyx_BUILTIN_MODULE_NAME)); if (unlikely(!__pyx_b)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#if CYTHON_COMPILING_IN_PYPY
|
|
Py_INCREF(__pyx_b);
|
|
#endif
|
|
if (__Pyx_SetAttrString(__pyx_m, "__builtins__", __pyx_b) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
/*--- Initialize various global constants etc. ---*/
|
|
if (unlikely(__Pyx_InitGlobals() < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#if PY_MAJOR_VERSION < 3 && (__PYX_DEFAULT_STRING_ENCODING_IS_ASCII || __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT)
|
|
if (__Pyx_init_sys_getdefaultencoding_params() < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
#endif
|
|
if (__pyx_module_is_main_sklearn__tree___tree) {
|
|
if (__Pyx_SetAttrString(__pyx_m, "__name__", __pyx_n_s_main) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
|
|
}
|
|
#if PY_MAJOR_VERSION >= 3
|
|
{
|
|
PyObject *modules = PyImport_GetModuleDict(); if (unlikely(!modules)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (!PyDict_GetItemString(modules, "sklearn.tree._tree")) {
|
|
if (unlikely(PyDict_SetItemString(modules, "sklearn.tree._tree", __pyx_m) < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
}
|
|
}
|
|
#endif
|
|
/*--- Builtin init code ---*/
|
|
if (unlikely(__Pyx_InitCachedBuiltins() < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
/*--- Constants init code ---*/
|
|
if (unlikely(__Pyx_InitCachedConstants() < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
/*--- Global init code ---*/
|
|
/*--- Variable export code ---*/
|
|
/*--- Function export code ---*/
|
|
/*--- Type init code ---*/
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Criterion = &__pyx_vtable_7sklearn_4tree_5_tree_Criterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_9Criterion_init;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.reset = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_9Criterion_reset;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.update = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_9Criterion_update;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_9Criterion_node_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.children_impurity = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *))__pyx_f_7sklearn_4tree_5_tree_9Criterion_children_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.node_value = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *))__pyx_f_7sklearn_4tree_5_tree_9Criterion_node_value;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.impurity_improvement = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double))__pyx_f_7sklearn_4tree_5_tree_9Criterion_impurity_improvement;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_Criterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 97; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_Criterion.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_Criterion.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_Criterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 97; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "Criterion", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_Criterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 97; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_Criterion = &__pyx_type_7sklearn_4tree_5_tree_Criterion;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Splitter = &__pyx_vtable_7sklearn_4tree_5_tree_Splitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Splitter.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, PyObject *, PyArrayObject *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *))__pyx_f_7sklearn_4tree_5_tree_8Splitter_init;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Splitter.node_reset = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double *))__pyx_f_7sklearn_4tree_5_tree_8Splitter_node_reset;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Splitter.node_split = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *))__pyx_f_7sklearn_4tree_5_tree_8Splitter_node_split;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Splitter.node_value = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double *))__pyx_f_7sklearn_4tree_5_tree_8Splitter_node_value;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Splitter.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *))__pyx_f_7sklearn_4tree_5_tree_8Splitter_node_impurity;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_Splitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 922; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_Splitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_Splitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_Splitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 922; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "Splitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_Splitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 922; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_Splitter = &__pyx_type_7sklearn_4tree_5_tree_Splitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Tree = &__pyx_vtable_7sklearn_4tree_5_tree_Tree;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._add_node = (__pyx_t_7sklearn_4tree_5_tree_SIZE_t (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, int, int, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double))__pyx_f_7sklearn_4tree_5_tree_4Tree__add_node;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._resize = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_4Tree__resize;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._resize_c = (int (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree__resize_c *__pyx_optional_args))__pyx_f_7sklearn_4tree_5_tree_4Tree__resize_c;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._get_value_ndarray = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *))__pyx_f_7sklearn_4tree_5_tree_4Tree__get_value_ndarray;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._get_node_ndarray = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *))__pyx_f_7sklearn_4tree_5_tree_4Tree__get_node_ndarray;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree.predict = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, int __pyx_skip_dispatch))__pyx_f_7sklearn_4tree_5_tree_4Tree_predict;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree.apply = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, int __pyx_skip_dispatch))__pyx_f_7sklearn_4tree_5_tree_4Tree_apply;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._apply_dense = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *))__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_dense;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree._apply_sparse_csr = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *))__pyx_f_7sklearn_4tree_5_tree_4Tree__apply_sparse_csr;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Tree.compute_feature_importances = (PyObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_4Tree_compute_feature_importances *__pyx_optional_args))__pyx_f_7sklearn_4tree_5_tree_4Tree_compute_feature_importances;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_Tree) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2955; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_Tree.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_Tree.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_Tree) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2955; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "Tree", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_Tree) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2955; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_Tree = &__pyx_type_7sklearn_4tree_5_tree_Tree;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_TreeBuilder = &__pyx_vtable_7sklearn_4tree_5_tree_TreeBuilder;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_TreeBuilder.build = (PyObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, PyArrayObject *, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build *__pyx_optional_args))__pyx_f_7sklearn_4tree_5_tree_11TreeBuilder_build;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_TreeBuilder._check_input = (PyObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, PyObject *, PyArrayObject *, PyArrayObject *))__pyx_f_7sklearn_4tree_5_tree_11TreeBuilder__check_input;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_TreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2566; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_TreeBuilder.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_TreeBuilder.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_TreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2566; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "TreeBuilder", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_TreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2566; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_TreeBuilder = &__pyx_type_7sklearn_4tree_5_tree_TreeBuilder;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_ClassificationCriterion = &__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_Criterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_init;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.reset = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_reset;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.update = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_update;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_node_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.children_impurity = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_children_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.node_value = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_node_value;
|
|
__pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_Criterion;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 150; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_ClassificationCriterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 150; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "ClassificationCriterion", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 150; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion = &__pyx_type_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Entropy = &__pyx_vtable_7sklearn_4tree_5_tree_Entropy;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Entropy.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Entropy.__pyx_base.__pyx_base.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_7Entropy_node_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Entropy.__pyx_base.__pyx_base.children_impurity = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *))__pyx_f_7sklearn_4tree_5_tree_7Entropy_children_impurity;
|
|
__pyx_type_7sklearn_4tree_5_tree_Entropy.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_Entropy) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_Entropy.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_Entropy.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_Entropy) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "Entropy", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_Entropy) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 368; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_Entropy = &__pyx_type_7sklearn_4tree_5_tree_Entropy;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_Gini = &__pyx_vtable_7sklearn_4tree_5_tree_Gini;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Gini.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Gini.__pyx_base.__pyx_base.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_4Gini_node_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_Gini.__pyx_base.__pyx_base.children_impurity = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *))__pyx_f_7sklearn_4tree_5_tree_4Gini_children_impurity;
|
|
__pyx_type_7sklearn_4tree_5_tree_Gini.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_Gini) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_Gini.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_Gini.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_Gini) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "Gini", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_Gini) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 459; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_Gini = &__pyx_type_7sklearn_4tree_5_tree_Gini;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_RegressionCriterion = &__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_Criterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_init;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.reset = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_reset;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.update = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_update;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_node_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.children_impurity = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_children_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.node_value = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_node_value;
|
|
__pyx_type_7sklearn_4tree_5_tree_RegressionCriterion.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_Criterion;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_RegressionCriterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 553; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_RegressionCriterion.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_RegressionCriterion.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_RegressionCriterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 553; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "RegressionCriterion", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_RegressionCriterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 553; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion = &__pyx_type_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_MSE = &__pyx_vtable_7sklearn_4tree_5_tree_MSE;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_MSE.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_MSE.__pyx_base.__pyx_base.node_impurity = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_3MSE_node_impurity;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_MSE.__pyx_base.__pyx_base.children_impurity = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double *, double *))__pyx_f_7sklearn_4tree_5_tree_3MSE_children_impurity;
|
|
__pyx_type_7sklearn_4tree_5_tree_MSE.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_MSE) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_MSE.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_MSE.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_MSE) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "MSE", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_MSE) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_MSE = &__pyx_type_7sklearn_4tree_5_tree_MSE;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_FriedmanMSE = &__pyx_vtable_7sklearn_4tree_5_tree_FriedmanMSE;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_FriedmanMSE.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_MSE;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_FriedmanMSE.__pyx_base.__pyx_base.__pyx_base.impurity_improvement = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, double))__pyx_f_7sklearn_4tree_5_tree_11FriedmanMSE_impurity_improvement;
|
|
__pyx_type_7sklearn_4tree_5_tree_FriedmanMSE.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_MSE;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_FriedmanMSE) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 877; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_FriedmanMSE.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_FriedmanMSE.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_FriedmanMSE) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 877; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "FriedmanMSE", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_FriedmanMSE) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 877; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_FriedmanMSE = &__pyx_type_7sklearn_4tree_5_tree_FriedmanMSE;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BaseDenseSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_Splitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BaseDenseSplitter.__pyx_base.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, PyObject *, PyArrayObject *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *))__pyx_f_7sklearn_4tree_5_tree_17BaseDenseSplitter_init;
|
|
__pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_Splitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1031; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1031; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "BaseDenseSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1031; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_BaseDenseSplitter = &__pyx_type_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_BestSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_BestSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestSplitter.__pyx_base.__pyx_base.node_split = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *))__pyx_f_7sklearn_4tree_5_tree_12BestSplitter_node_split;
|
|
__pyx_type_7sklearn_4tree_5_tree_BestSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_BestSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1062; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_BestSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_BestSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_BestSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1062; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "BestSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_BestSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1062; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_BestSplitter = &__pyx_type_7sklearn_4tree_5_tree_BestSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_RandomSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_RandomSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RandomSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RandomSplitter.__pyx_base.__pyx_base.node_split = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *))__pyx_f_7sklearn_4tree_5_tree_14RandomSplitter_node_split;
|
|
__pyx_type_7sklearn_4tree_5_tree_RandomSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_RandomSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1361; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_RandomSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_RandomSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_RandomSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1361; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "RandomSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_RandomSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1361; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_RandomSplitter = &__pyx_type_7sklearn_4tree_5_tree_RandomSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_PresortBestSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_PresortBestSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_PresortBestSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_PresortBestSplitter.__pyx_base.__pyx_base.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, PyObject *, PyArrayObject *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *))__pyx_f_7sklearn_4tree_5_tree_19PresortBestSplitter_init;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_PresortBestSplitter.__pyx_base.__pyx_base.node_split = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *))__pyx_f_7sklearn_4tree_5_tree_19PresortBestSplitter_node_split;
|
|
__pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_BaseDenseSplitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1563; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_PresortBestSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1563; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "PresortBestSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1563; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_PresortBestSplitter = &__pyx_type_7sklearn_4tree_5_tree_PresortBestSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_BaseSparseSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BaseSparseSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_Splitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BaseSparseSplitter.__pyx_base.init = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, PyObject *, PyArrayObject *, __pyx_t_7sklearn_4tree_5_tree_DOUBLE_t *))__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_init;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BaseSparseSplitter._partition = (__pyx_t_7sklearn_4tree_5_tree_SIZE_t (*)(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *, double, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t))__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter__partition;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BaseSparseSplitter.extract_nnz = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_BaseSparseSplitter *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *, int *))__pyx_f_7sklearn_4tree_5_tree_18BaseSparseSplitter_extract_nnz;
|
|
__pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_Splitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1811; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_BaseSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1811; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "BaseSparseSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1811; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_BaseSparseSplitter = &__pyx_type_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_BestSparseSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_BestSparseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestSparseSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestSparseSplitter.__pyx_base.__pyx_base.node_split = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *))__pyx_f_7sklearn_4tree_5_tree_18BestSparseSplitter_node_split;
|
|
__pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2130; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_BestSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2130; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "BestSparseSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2130; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_BestSparseSplitter = &__pyx_type_7sklearn_4tree_5_tree_BestSparseSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_RandomSparseSplitter = &__pyx_vtable_7sklearn_4tree_5_tree_RandomSparseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RandomSparseSplitter.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_RandomSparseSplitter.__pyx_base.__pyx_base.node_split = (void (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, double, struct __pyx_t_7sklearn_4tree_5_tree_SplitRecord *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t *))__pyx_f_7sklearn_4tree_5_tree_20RandomSparseSplitter_node_split;
|
|
__pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_BaseSparseSplitter;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2348; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_RandomSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2348; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "RandomSparseSplitter", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2348; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_RandomSparseSplitter = &__pyx_type_7sklearn_4tree_5_tree_RandomSparseSplitter;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_DepthFirstTreeBuilder = &__pyx_vtable_7sklearn_4tree_5_tree_DepthFirstTreeBuilder;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_DepthFirstTreeBuilder.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_TreeBuilder;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_DepthFirstTreeBuilder.__pyx_base.build = (PyObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, PyArrayObject *, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build *__pyx_optional_args))__pyx_f_7sklearn_4tree_5_tree_21DepthFirstTreeBuilder_build;
|
|
__pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_TreeBuilder;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2605; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_DepthFirstTreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2605; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "DepthFirstTreeBuilder", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2605; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_DepthFirstTreeBuilder = &__pyx_type_7sklearn_4tree_5_tree_DepthFirstTreeBuilder;
|
|
__pyx_vtabptr_7sklearn_4tree_5_tree_BestFirstTreeBuilder = &__pyx_vtable_7sklearn_4tree_5_tree_BestFirstTreeBuilder;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestFirstTreeBuilder.__pyx_base = *__pyx_vtabptr_7sklearn_4tree_5_tree_TreeBuilder;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestFirstTreeBuilder.__pyx_base.build = (PyObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_TreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, PyObject *, PyArrayObject *, int __pyx_skip_dispatch, struct __pyx_opt_args_7sklearn_4tree_5_tree_11TreeBuilder_build *__pyx_optional_args))__pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder_build;
|
|
__pyx_vtable_7sklearn_4tree_5_tree_BestFirstTreeBuilder._add_split_node = (int (*)(struct __pyx_obj_7sklearn_4tree_5_tree_BestFirstTreeBuilder *, struct __pyx_obj_7sklearn_4tree_5_tree_Splitter *, struct __pyx_obj_7sklearn_4tree_5_tree_Tree *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, double, int, int, struct __pyx_t_7sklearn_4tree_5_tree_Node *, __pyx_t_7sklearn_4tree_5_tree_SIZE_t, struct __pyx_t_7sklearn_4tree_6_utils_PriorityHeapRecord *))__pyx_f_7sklearn_4tree_5_tree_20BestFirstTreeBuilder__add_split_node;
|
|
__pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder.tp_base = __pyx_ptype_7sklearn_4tree_5_tree_TreeBuilder;
|
|
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2751; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder.tp_print = 0;
|
|
if (__Pyx_SetVtable(__pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder.tp_dict, __pyx_vtabptr_7sklearn_4tree_5_tree_BestFirstTreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2751; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
if (__Pyx_SetAttrString(__pyx_m, "BestFirstTreeBuilder", (PyObject *)&__pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 2751; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_5_tree_BestFirstTreeBuilder = &__pyx_type_7sklearn_4tree_5_tree_BestFirstTreeBuilder;
|
|
/*--- Type import code ---*/
|
|
__pyx_ptype_7cpython_4type_type = __Pyx_ImportType(__Pyx_BUILTIN_MODULE_NAME, "type",
|
|
#if CYTHON_COMPILING_IN_PYPY
|
|
sizeof(PyTypeObject),
|
|
#else
|
|
sizeof(PyHeapTypeObject),
|
|
#endif
|
|
0); if (unlikely(!__pyx_ptype_7cpython_4type_type)) {__pyx_filename = __pyx_f[3]; __pyx_lineno = 9; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_dtype = __Pyx_ImportType("numpy", "dtype", sizeof(PyArray_Descr), 0); if (unlikely(!__pyx_ptype_5numpy_dtype)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 155; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_flatiter = __Pyx_ImportType("numpy", "flatiter", sizeof(PyArrayIterObject), 0); if (unlikely(!__pyx_ptype_5numpy_flatiter)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 165; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_broadcast = __Pyx_ImportType("numpy", "broadcast", sizeof(PyArrayMultiIterObject), 0); if (unlikely(!__pyx_ptype_5numpy_broadcast)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 169; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_ndarray = __Pyx_ImportType("numpy", "ndarray", sizeof(PyArrayObject), 0); if (unlikely(!__pyx_ptype_5numpy_ndarray)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 178; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_5numpy_ufunc = __Pyx_ImportType("numpy", "ufunc", sizeof(PyUFuncObject), 0); if (unlikely(!__pyx_ptype_5numpy_ufunc)) {__pyx_filename = __pyx_f[2]; __pyx_lineno = 861; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7cpython_4bool_bool = __Pyx_ImportType(__Pyx_BUILTIN_MODULE_NAME, "bool", sizeof(PyBoolObject), 0); if (unlikely(!__pyx_ptype_7cpython_4bool_bool)) {__pyx_filename = __pyx_f[4]; __pyx_lineno = 8; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7cpython_7complex_complex = __Pyx_ImportType(__Pyx_BUILTIN_MODULE_NAME, "complex", sizeof(PyComplexObject), 0); if (unlikely(!__pyx_ptype_7cpython_7complex_complex)) {__pyx_filename = __pyx_f[5]; __pyx_lineno = 15; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_6_utils_Stack = __Pyx_ImportType("sklearn.tree._utils", "Stack", sizeof(struct __pyx_obj_7sklearn_4tree_6_utils_Stack), 1); if (unlikely(!__pyx_ptype_7sklearn_4tree_6_utils_Stack)) {__pyx_filename = __pyx_f[6]; __pyx_lineno = 29; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_vtabptr_7sklearn_4tree_6_utils_Stack = (struct __pyx_vtabstruct_7sklearn_4tree_6_utils_Stack*)__Pyx_GetVtable(__pyx_ptype_7sklearn_4tree_6_utils_Stack->tp_dict); if (unlikely(!__pyx_vtabptr_7sklearn_4tree_6_utils_Stack)) {__pyx_filename = __pyx_f[6]; __pyx_lineno = 29; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_ptype_7sklearn_4tree_6_utils_PriorityHeap = __Pyx_ImportType("sklearn.tree._utils", "PriorityHeap", sizeof(struct __pyx_obj_7sklearn_4tree_6_utils_PriorityHeap), 1); if (unlikely(!__pyx_ptype_7sklearn_4tree_6_utils_PriorityHeap)) {__pyx_filename = __pyx_f[6]; __pyx_lineno = 58; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__pyx_vtabptr_7sklearn_4tree_6_utils_PriorityHeap = (struct __pyx_vtabstruct_7sklearn_4tree_6_utils_PriorityHeap*)__Pyx_GetVtable(__pyx_ptype_7sklearn_4tree_6_utils_PriorityHeap->tp_dict); if (unlikely(!__pyx_vtabptr_7sklearn_4tree_6_utils_PriorityHeap)) {__pyx_filename = __pyx_f[6]; __pyx_lineno = 58; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
/*--- Variable import code ---*/
|
|
/*--- Function import code ---*/
|
|
/*--- Execution code ---*/
|
|
|
|
/* "sklearn/tree/_tree.pyx":23
|
|
* from cpython cimport Py_INCREF, PyObject
|
|
*
|
|
* import numpy as np # <<<<<<<<<<<<<<
|
|
* cimport numpy as np
|
|
* np.import_array()
|
|
*/
|
|
__pyx_t_1 = __Pyx_Import(__pyx_n_s_numpy, 0, -1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 23; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_np, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 23; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":25
|
|
* import numpy as np
|
|
* cimport numpy as np
|
|
* np.import_array() # <<<<<<<<<<<<<<
|
|
*
|
|
* from scipy.sparse import issparse, csc_matrix, csr_matrix
|
|
*/
|
|
import_array();
|
|
|
|
/* "sklearn/tree/_tree.pyx":27
|
|
* np.import_array()
|
|
*
|
|
* from scipy.sparse import issparse, csc_matrix, csr_matrix # <<<<<<<<<<<<<<
|
|
*
|
|
* from sklearn.tree._utils cimport Stack, StackRecord
|
|
*/
|
|
__pyx_t_1 = PyList_New(3); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(__pyx_n_s_issparse);
|
|
PyList_SET_ITEM(__pyx_t_1, 0, __pyx_n_s_issparse);
|
|
__Pyx_GIVEREF(__pyx_n_s_issparse);
|
|
__Pyx_INCREF(__pyx_n_s_csc_matrix);
|
|
PyList_SET_ITEM(__pyx_t_1, 1, __pyx_n_s_csc_matrix);
|
|
__Pyx_GIVEREF(__pyx_n_s_csc_matrix);
|
|
__Pyx_INCREF(__pyx_n_s_csr_matrix);
|
|
PyList_SET_ITEM(__pyx_t_1, 2, __pyx_n_s_csr_matrix);
|
|
__Pyx_GIVEREF(__pyx_n_s_csr_matrix);
|
|
__pyx_t_2 = __Pyx_Import(__pyx_n_s_scipy_sparse, __pyx_t_1, -1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_ImportFrom(__pyx_t_2, __pyx_n_s_issparse); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_issparse, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_ImportFrom(__pyx_t_2, __pyx_n_s_csc_matrix); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_csc_matrix, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_ImportFrom(__pyx_t_2, __pyx_n_s_csr_matrix); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_csr_matrix, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 27; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":44
|
|
* # =============================================================================
|
|
*
|
|
* from numpy import float32 as DTYPE # <<<<<<<<<<<<<<
|
|
* from numpy import float64 as DOUBLE
|
|
*
|
|
*/
|
|
__pyx_t_2 = PyList_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 44; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_n_s_float32);
|
|
PyList_SET_ITEM(__pyx_t_2, 0, __pyx_n_s_float32);
|
|
__Pyx_GIVEREF(__pyx_n_s_float32);
|
|
__pyx_t_1 = __Pyx_Import(__pyx_n_s_numpy, __pyx_t_2, -1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 44; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_ImportFrom(__pyx_t_1, __pyx_n_s_float32); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 44; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_DTYPE, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 44; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":45
|
|
*
|
|
* from numpy import float32 as DTYPE
|
|
* from numpy import float64 as DOUBLE # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef double INFINITY = np.inf
|
|
*/
|
|
__pyx_t_1 = PyList_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 45; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_INCREF(__pyx_n_s_float64);
|
|
PyList_SET_ITEM(__pyx_t_1, 0, __pyx_n_s_float64);
|
|
__Pyx_GIVEREF(__pyx_n_s_float64);
|
|
__pyx_t_2 = __Pyx_Import(__pyx_n_s_numpy, __pyx_t_1, -1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 45; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_ImportFrom(__pyx_t_2, __pyx_n_s_float64); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 45; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_DOUBLE, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 45; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":47
|
|
* from numpy import float64 as DOUBLE
|
|
*
|
|
* cdef double INFINITY = np.inf # <<<<<<<<<<<<<<
|
|
* TREE_LEAF = -1
|
|
* TREE_UNDEFINED = -2
|
|
*/
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 47; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_inf); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 47; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_3 = __pyx_PyFloat_AsDouble(__pyx_t_1); if (unlikely((__pyx_t_3 == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 47; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_v_7sklearn_4tree_5_tree_INFINITY = __pyx_t_3;
|
|
|
|
/* "sklearn/tree/_tree.pyx":48
|
|
*
|
|
* cdef double INFINITY = np.inf
|
|
* TREE_LEAF = -1 # <<<<<<<<<<<<<<
|
|
* TREE_UNDEFINED = -2
|
|
* cdef SIZE_t _TREE_LEAF = TREE_LEAF
|
|
*/
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_TREE_LEAF, __pyx_int_neg_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 48; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":49
|
|
* cdef double INFINITY = np.inf
|
|
* TREE_LEAF = -1
|
|
* TREE_UNDEFINED = -2 # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t _TREE_LEAF = TREE_LEAF
|
|
* cdef SIZE_t _TREE_UNDEFINED = TREE_UNDEFINED
|
|
*/
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_TREE_UNDEFINED, __pyx_int_neg_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 49; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
|
|
/* "sklearn/tree/_tree.pyx":50
|
|
* TREE_LEAF = -1
|
|
* TREE_UNDEFINED = -2
|
|
* cdef SIZE_t _TREE_LEAF = TREE_LEAF # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t _TREE_UNDEFINED = TREE_UNDEFINED
|
|
* cdef SIZE_t INITIAL_STACK_SIZE = 10
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_TREE_LEAF); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 50; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_4 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_1); if (unlikely((__pyx_t_4 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 50; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_v_7sklearn_4tree_5_tree__TREE_LEAF = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":51
|
|
* TREE_UNDEFINED = -2
|
|
* cdef SIZE_t _TREE_LEAF = TREE_LEAF
|
|
* cdef SIZE_t _TREE_UNDEFINED = TREE_UNDEFINED # <<<<<<<<<<<<<<
|
|
* cdef SIZE_t INITIAL_STACK_SIZE = 10
|
|
*
|
|
*/
|
|
__pyx_t_1 = __Pyx_GetModuleGlobalName(__pyx_n_s_TREE_UNDEFINED); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 51; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__pyx_t_4 = __Pyx_PyInt_As_Py_intptr_t(__pyx_t_1); if (unlikely((__pyx_t_4 == (npy_intp)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 51; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
__pyx_v_7sklearn_4tree_5_tree__TREE_UNDEFINED = __pyx_t_4;
|
|
|
|
/* "sklearn/tree/_tree.pyx":52
|
|
* cdef SIZE_t _TREE_LEAF = TREE_LEAF
|
|
* cdef SIZE_t _TREE_UNDEFINED = TREE_UNDEFINED
|
|
* cdef SIZE_t INITIAL_STACK_SIZE = 10 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef DTYPE_t MIN_IMPURITY_SPLIT = 1e-7
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_INITIAL_STACK_SIZE = 10;
|
|
|
|
/* "sklearn/tree/_tree.pyx":54
|
|
* cdef SIZE_t INITIAL_STACK_SIZE = 10
|
|
*
|
|
* cdef DTYPE_t MIN_IMPURITY_SPLIT = 1e-7 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Mitigate precision differences between 32 bit and 64 bit
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_MIN_IMPURITY_SPLIT = 1e-7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":57
|
|
*
|
|
* # Mitigate precision differences between 32 bit and 64 bit
|
|
* cdef DTYPE_t FEATURE_THRESHOLD = 1e-7 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Constant to switch between algorithm non zero value extract algorithm
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_FEATURE_THRESHOLD = 1e-7;
|
|
|
|
/* "sklearn/tree/_tree.pyx":61
|
|
* # Constant to switch between algorithm non zero value extract algorithm
|
|
* # in SparseSplitter
|
|
* cdef DTYPE_t EXTRACT_NNZ_SWITCH = 0.1 # <<<<<<<<<<<<<<
|
|
*
|
|
* # Some handy constants (BestFirstTreeBuilder)
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_EXTRACT_NNZ_SWITCH = 0.1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":64
|
|
*
|
|
* # Some handy constants (BestFirstTreeBuilder)
|
|
* cdef int IS_FIRST = 1 # <<<<<<<<<<<<<<
|
|
* cdef int IS_NOT_FIRST = 0
|
|
* cdef int IS_LEFT = 1
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_IS_FIRST = 1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":65
|
|
* # Some handy constants (BestFirstTreeBuilder)
|
|
* cdef int IS_FIRST = 1
|
|
* cdef int IS_NOT_FIRST = 0 # <<<<<<<<<<<<<<
|
|
* cdef int IS_LEFT = 1
|
|
* cdef int IS_NOT_LEFT = 0
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_IS_NOT_FIRST = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":66
|
|
* cdef int IS_FIRST = 1
|
|
* cdef int IS_NOT_FIRST = 0
|
|
* cdef int IS_LEFT = 1 # <<<<<<<<<<<<<<
|
|
* cdef int IS_NOT_LEFT = 0
|
|
*
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_IS_LEFT = 1;
|
|
|
|
/* "sklearn/tree/_tree.pyx":67
|
|
* cdef int IS_NOT_FIRST = 0
|
|
* cdef int IS_LEFT = 1
|
|
* cdef int IS_NOT_LEFT = 0 # <<<<<<<<<<<<<<
|
|
*
|
|
* cdef enum:
|
|
*/
|
|
__pyx_v_7sklearn_4tree_5_tree_IS_NOT_LEFT = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":76
|
|
*
|
|
* # Repeat struct definition for numpy
|
|
* NODE_DTYPE = np.dtype({ # <<<<<<<<<<<<<<
|
|
* 'names': ['left_child', 'right_child', 'feature', 'threshold', 'impurity',
|
|
* 'n_node_samples', 'weighted_n_node_samples'],
|
|
*/
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":77
|
|
* # Repeat struct definition for numpy
|
|
* NODE_DTYPE = np.dtype({
|
|
* 'names': ['left_child', 'right_child', 'feature', 'threshold', 'impurity', # <<<<<<<<<<<<<<
|
|
* 'n_node_samples', 'weighted_n_node_samples'],
|
|
* 'formats': [np.intp, np.intp, np.intp, np.float64, np.float64, np.intp,
|
|
*/
|
|
__pyx_t_2 = PyList_New(7); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 77; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__Pyx_INCREF(__pyx_n_s_left_child);
|
|
PyList_SET_ITEM(__pyx_t_2, 0, __pyx_n_s_left_child);
|
|
__Pyx_GIVEREF(__pyx_n_s_left_child);
|
|
__Pyx_INCREF(__pyx_n_s_right_child);
|
|
PyList_SET_ITEM(__pyx_t_2, 1, __pyx_n_s_right_child);
|
|
__Pyx_GIVEREF(__pyx_n_s_right_child);
|
|
__Pyx_INCREF(__pyx_n_s_feature);
|
|
PyList_SET_ITEM(__pyx_t_2, 2, __pyx_n_s_feature);
|
|
__Pyx_GIVEREF(__pyx_n_s_feature);
|
|
__Pyx_INCREF(__pyx_n_s_threshold);
|
|
PyList_SET_ITEM(__pyx_t_2, 3, __pyx_n_s_threshold);
|
|
__Pyx_GIVEREF(__pyx_n_s_threshold);
|
|
__Pyx_INCREF(__pyx_n_s_impurity);
|
|
PyList_SET_ITEM(__pyx_t_2, 4, __pyx_n_s_impurity);
|
|
__Pyx_GIVEREF(__pyx_n_s_impurity);
|
|
__Pyx_INCREF(__pyx_n_s_n_node_samples);
|
|
PyList_SET_ITEM(__pyx_t_2, 5, __pyx_n_s_n_node_samples);
|
|
__Pyx_GIVEREF(__pyx_n_s_n_node_samples);
|
|
__Pyx_INCREF(__pyx_n_s_weighted_n_node_samples);
|
|
PyList_SET_ITEM(__pyx_t_2, 6, __pyx_n_s_weighted_n_node_samples);
|
|
__Pyx_GIVEREF(__pyx_n_s_weighted_n_node_samples);
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_names, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":79
|
|
* 'names': ['left_child', 'right_child', 'feature', 'threshold', 'impurity',
|
|
* 'n_node_samples', 'weighted_n_node_samples'],
|
|
* 'formats': [np.intp, np.intp, np.intp, np.float64, np.float64, np.intp, # <<<<<<<<<<<<<<
|
|
* np.float64],
|
|
* 'offsets': [
|
|
*/
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_5 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_intp); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_6 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_intp); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_7 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_intp); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_8 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_float64); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_9 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_float64); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_intp); if (unlikely(!__pyx_t_10)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_10);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":80
|
|
* 'n_node_samples', 'weighted_n_node_samples'],
|
|
* 'formats': [np.intp, np.intp, np.intp, np.float64, np.float64, np.intp,
|
|
* np.float64], # <<<<<<<<<<<<<<
|
|
* 'offsets': [
|
|
* <Py_ssize_t> &(<Node*> NULL).left_child,
|
|
*/
|
|
__pyx_t_2 = __Pyx_GetModuleGlobalName(__pyx_n_s_np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 80; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
__pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_2, __pyx_n_s_float64); if (unlikely(!__pyx_t_11)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 80; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_11);
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":79
|
|
* 'names': ['left_child', 'right_child', 'feature', 'threshold', 'impurity',
|
|
* 'n_node_samples', 'weighted_n_node_samples'],
|
|
* 'formats': [np.intp, np.intp, np.intp, np.float64, np.float64, np.intp, # <<<<<<<<<<<<<<
|
|
* np.float64],
|
|
* 'offsets': [
|
|
*/
|
|
__pyx_t_2 = PyList_New(7); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 79; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
PyList_SET_ITEM(__pyx_t_2, 0, __pyx_t_5);
|
|
__Pyx_GIVEREF(__pyx_t_5);
|
|
PyList_SET_ITEM(__pyx_t_2, 1, __pyx_t_6);
|
|
__Pyx_GIVEREF(__pyx_t_6);
|
|
PyList_SET_ITEM(__pyx_t_2, 2, __pyx_t_7);
|
|
__Pyx_GIVEREF(__pyx_t_7);
|
|
PyList_SET_ITEM(__pyx_t_2, 3, __pyx_t_8);
|
|
__Pyx_GIVEREF(__pyx_t_8);
|
|
PyList_SET_ITEM(__pyx_t_2, 4, __pyx_t_9);
|
|
__Pyx_GIVEREF(__pyx_t_9);
|
|
PyList_SET_ITEM(__pyx_t_2, 5, __pyx_t_10);
|
|
__Pyx_GIVEREF(__pyx_t_10);
|
|
PyList_SET_ITEM(__pyx_t_2, 6, __pyx_t_11);
|
|
__Pyx_GIVEREF(__pyx_t_11);
|
|
__pyx_t_5 = 0;
|
|
__pyx_t_6 = 0;
|
|
__pyx_t_7 = 0;
|
|
__pyx_t_8 = 0;
|
|
__pyx_t_9 = 0;
|
|
__pyx_t_10 = 0;
|
|
__pyx_t_11 = 0;
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_formats, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":82
|
|
* np.float64],
|
|
* 'offsets': [
|
|
* <Py_ssize_t> &(<Node*> NULL).left_child, # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).right_child,
|
|
* <Py_ssize_t> &(<Node*> NULL).feature,
|
|
*/
|
|
__pyx_t_2 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->left_child))); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 82; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_2);
|
|
|
|
/* "sklearn/tree/_tree.pyx":83
|
|
* 'offsets': [
|
|
* <Py_ssize_t> &(<Node*> NULL).left_child,
|
|
* <Py_ssize_t> &(<Node*> NULL).right_child, # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).feature,
|
|
* <Py_ssize_t> &(<Node*> NULL).threshold,
|
|
*/
|
|
__pyx_t_11 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->right_child))); if (unlikely(!__pyx_t_11)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 83; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_11);
|
|
|
|
/* "sklearn/tree/_tree.pyx":84
|
|
* <Py_ssize_t> &(<Node*> NULL).left_child,
|
|
* <Py_ssize_t> &(<Node*> NULL).right_child,
|
|
* <Py_ssize_t> &(<Node*> NULL).feature, # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).threshold,
|
|
* <Py_ssize_t> &(<Node*> NULL).impurity,
|
|
*/
|
|
__pyx_t_10 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->feature))); if (unlikely(!__pyx_t_10)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 84; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_10);
|
|
|
|
/* "sklearn/tree/_tree.pyx":85
|
|
* <Py_ssize_t> &(<Node*> NULL).right_child,
|
|
* <Py_ssize_t> &(<Node*> NULL).feature,
|
|
* <Py_ssize_t> &(<Node*> NULL).threshold, # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).impurity,
|
|
* <Py_ssize_t> &(<Node*> NULL).n_node_samples,
|
|
*/
|
|
__pyx_t_9 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->threshold))); if (unlikely(!__pyx_t_9)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 85; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_9);
|
|
|
|
/* "sklearn/tree/_tree.pyx":86
|
|
* <Py_ssize_t> &(<Node*> NULL).feature,
|
|
* <Py_ssize_t> &(<Node*> NULL).threshold,
|
|
* <Py_ssize_t> &(<Node*> NULL).impurity, # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).n_node_samples,
|
|
* <Py_ssize_t> &(<Node*> NULL).weighted_n_node_samples
|
|
*/
|
|
__pyx_t_8 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->impurity))); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 86; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_8);
|
|
|
|
/* "sklearn/tree/_tree.pyx":87
|
|
* <Py_ssize_t> &(<Node*> NULL).threshold,
|
|
* <Py_ssize_t> &(<Node*> NULL).impurity,
|
|
* <Py_ssize_t> &(<Node*> NULL).n_node_samples, # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).weighted_n_node_samples
|
|
* ]
|
|
*/
|
|
__pyx_t_7 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->n_node_samples))); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 87; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_7);
|
|
|
|
/* "sklearn/tree/_tree.pyx":88
|
|
* <Py_ssize_t> &(<Node*> NULL).impurity,
|
|
* <Py_ssize_t> &(<Node*> NULL).n_node_samples,
|
|
* <Py_ssize_t> &(<Node*> NULL).weighted_n_node_samples # <<<<<<<<<<<<<<
|
|
* ]
|
|
* })
|
|
*/
|
|
__pyx_t_6 = PyInt_FromSsize_t(((Py_ssize_t)(&((struct __pyx_t_7sklearn_4tree_5_tree_Node *)NULL)->weighted_n_node_samples))); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 88; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_6);
|
|
|
|
/* "sklearn/tree/_tree.pyx":81
|
|
* 'formats': [np.intp, np.intp, np.intp, np.float64, np.float64, np.intp,
|
|
* np.float64],
|
|
* 'offsets': [ # <<<<<<<<<<<<<<
|
|
* <Py_ssize_t> &(<Node*> NULL).left_child,
|
|
* <Py_ssize_t> &(<Node*> NULL).right_child,
|
|
*/
|
|
__pyx_t_5 = PyList_New(7); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 81; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyList_SET_ITEM(__pyx_t_5, 0, __pyx_t_2);
|
|
__Pyx_GIVEREF(__pyx_t_2);
|
|
PyList_SET_ITEM(__pyx_t_5, 1, __pyx_t_11);
|
|
__Pyx_GIVEREF(__pyx_t_11);
|
|
PyList_SET_ITEM(__pyx_t_5, 2, __pyx_t_10);
|
|
__Pyx_GIVEREF(__pyx_t_10);
|
|
PyList_SET_ITEM(__pyx_t_5, 3, __pyx_t_9);
|
|
__Pyx_GIVEREF(__pyx_t_9);
|
|
PyList_SET_ITEM(__pyx_t_5, 4, __pyx_t_8);
|
|
__Pyx_GIVEREF(__pyx_t_8);
|
|
PyList_SET_ITEM(__pyx_t_5, 5, __pyx_t_7);
|
|
__Pyx_GIVEREF(__pyx_t_7);
|
|
PyList_SET_ITEM(__pyx_t_5, 6, __pyx_t_6);
|
|
__Pyx_GIVEREF(__pyx_t_6);
|
|
__pyx_t_2 = 0;
|
|
__pyx_t_11 = 0;
|
|
__pyx_t_10 = 0;
|
|
__pyx_t_9 = 0;
|
|
__pyx_t_8 = 0;
|
|
__pyx_t_7 = 0;
|
|
__pyx_t_6 = 0;
|
|
if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_offsets, __pyx_t_5) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":76
|
|
*
|
|
* # Repeat struct definition for numpy
|
|
* NODE_DTYPE = np.dtype({ # <<<<<<<<<<<<<<
|
|
* 'names': ['left_child', 'right_child', 'feature', 'threshold', 'impurity',
|
|
* 'n_node_samples', 'weighted_n_node_samples'],
|
|
*/
|
|
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_5);
|
|
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_1);
|
|
__Pyx_GIVEREF(__pyx_t_1);
|
|
__pyx_t_1 = 0;
|
|
__pyx_t_1 = __Pyx_PyObject_Call(((PyObject *)((PyObject*)__pyx_ptype_5numpy_dtype)), __pyx_t_5, NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_NODE_DTYPE, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 76; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":3131
|
|
* # XXX using (size_t)(-1) is ugly, but SIZE_MAX is not available in C89
|
|
* # (i.e., older MSVC).
|
|
* cdef int _resize_c(self, SIZE_t capacity=<SIZE_t>(-1)) nogil: # <<<<<<<<<<<<<<
|
|
* """Guts of _resize. Returns 0 for success, -1 for error."""
|
|
* if capacity == self.capacity and self.nodes != NULL:
|
|
*/
|
|
__pyx_k__5 = ((__pyx_t_7sklearn_4tree_5_tree_SIZE_t)-1);
|
|
|
|
/* "sklearn/tree/_tree.pyx":3446
|
|
*
|
|
*
|
|
* def _realloc_test(): # <<<<<<<<<<<<<<
|
|
* # Helper for tests. Tries to allocate <size_t>(-1) / 2 * sizeof(size_t)
|
|
* # bytes, which will always overflow.
|
|
*/
|
|
__pyx_t_1 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_1_realloc_test, NULL, __pyx_n_s_sklearn_tree__tree); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3446; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_realloc_test, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 3446; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "sklearn/tree/_tree.pyx":1
|
|
* # cython: cdivision=True # <<<<<<<<<<<<<<
|
|
* # cython: boundscheck=False
|
|
* # cython: wraparound=False
|
|
*/
|
|
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_GOTREF(__pyx_t_1);
|
|
if (PyDict_SetItem(__pyx_d, __pyx_n_s_test, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
|
|
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
|
|
|
|
/* "/Users/ajoly/.virtualenvs/research/lib/python2.7/site-packages/Cython/Includes/numpy/__init__.pxd":976
|
|
* arr.base = baseptr
|
|
*
|
|
* cdef inline object get_array_base(ndarray arr): # <<<<<<<<<<<<<<
|
|
* if arr.base is NULL:
|
|
* return None
|
|
*/
|
|
goto __pyx_L0;
|
|
__pyx_L1_error:;
|
|
__Pyx_XDECREF(__pyx_t_1);
|
|
__Pyx_XDECREF(__pyx_t_2);
|
|
__Pyx_XDECREF(__pyx_t_5);
|
|
__Pyx_XDECREF(__pyx_t_6);
|
|
__Pyx_XDECREF(__pyx_t_7);
|
|
__Pyx_XDECREF(__pyx_t_8);
|
|
__Pyx_XDECREF(__pyx_t_9);
|
|
__Pyx_XDECREF(__pyx_t_10);
|
|
__Pyx_XDECREF(__pyx_t_11);
|
|
if (__pyx_m) {
|
|
__Pyx_AddTraceback("init sklearn.tree._tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
|
|
Py_DECREF(__pyx_m); __pyx_m = 0;
|
|
} else if (!PyErr_Occurred()) {
|
|
PyErr_SetString(PyExc_ImportError, "init sklearn.tree._tree");
|
|
}
|
|
__pyx_L0:;
|
|
__Pyx_RefNannyFinishContext();
|
|
#if PY_MAJOR_VERSION < 3
|
|
return;
|
|
#else
|
|
return __pyx_m;
|
|
#endif
|
|
}
|
|
|
|
/* Runtime support code */
|
|
#if CYTHON_REFNANNY
|
|
static __Pyx_RefNannyAPIStruct *__Pyx_RefNannyImportAPI(const char *modname) {
|
|
PyObject *m = NULL, *p = NULL;
|
|
void *r = NULL;
|
|
m = PyImport_ImportModule((char *)modname);
|
|
if (!m) goto end;
|
|
p = PyObject_GetAttrString(m, (char *)"RefNannyAPI");
|
|
if (!p) goto end;
|
|
r = PyLong_AsVoidPtr(p);
|
|
end:
|
|
Py_XDECREF(p);
|
|
Py_XDECREF(m);
|
|
return (__Pyx_RefNannyAPIStruct *)r;
|
|
}
|
|
#endif /* CYTHON_REFNANNY */
|
|
|
|
static PyObject *__Pyx_GetBuiltinName(PyObject *name) {
|
|
PyObject* result = __Pyx_PyObject_GetAttrStr(__pyx_b, name);
|
|
if (unlikely(!result)) {
|
|
PyErr_Format(PyExc_NameError,
|
|
#if PY_MAJOR_VERSION >= 3
|
|
"name '%U' is not defined", name);
|
|
#else
|
|
"name '%.200s' is not defined", PyString_AS_STRING(name));
|
|
#endif
|
|
}
|
|
return result;
|
|
}
|
|
|
|
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)
|
|
{
|
|
Py_ssize_t num_expected;
|
|
const char *more_or_less;
|
|
if (num_found < num_min) {
|
|
num_expected = num_min;
|
|
more_or_less = "at least";
|
|
} else {
|
|
num_expected = num_max;
|
|
more_or_less = "at most";
|
|
}
|
|
if (exact) {
|
|
more_or_less = "exactly";
|
|
}
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%.200s() takes %.8s %" CYTHON_FORMAT_SSIZE_T "d positional argument%.1s (%" CYTHON_FORMAT_SSIZE_T "d given)",
|
|
func_name, more_or_less, num_expected,
|
|
(num_expected == 1) ? "" : "s", num_found);
|
|
}
|
|
|
|
static void __Pyx_RaiseDoubleKeywordsError(
|
|
const char* func_name,
|
|
PyObject* kw_name)
|
|
{
|
|
PyErr_Format(PyExc_TypeError,
|
|
#if PY_MAJOR_VERSION >= 3
|
|
"%s() got multiple values for keyword argument '%U'", func_name, kw_name);
|
|
#else
|
|
"%s() got multiple values for keyword argument '%s'", func_name,
|
|
PyString_AsString(kw_name));
|
|
#endif
|
|
}
|
|
|
|
static int __Pyx_ParseOptionalKeywords(
|
|
PyObject *kwds,
|
|
PyObject **argnames[],
|
|
PyObject *kwds2,
|
|
PyObject *values[],
|
|
Py_ssize_t num_pos_args,
|
|
const char* function_name)
|
|
{
|
|
PyObject *key = 0, *value = 0;
|
|
Py_ssize_t pos = 0;
|
|
PyObject*** name;
|
|
PyObject*** first_kw_arg = argnames + num_pos_args;
|
|
while (PyDict_Next(kwds, &pos, &key, &value)) {
|
|
name = first_kw_arg;
|
|
while (*name && (**name != key)) name++;
|
|
if (*name) {
|
|
values[name-argnames] = value;
|
|
continue;
|
|
}
|
|
name = first_kw_arg;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyString_CheckExact(key)) || likely(PyString_Check(key))) {
|
|
while (*name) {
|
|
if ((CYTHON_COMPILING_IN_PYPY || PyString_GET_SIZE(**name) == PyString_GET_SIZE(key))
|
|
&& _PyString_Eq(**name, key)) {
|
|
values[name-argnames] = value;
|
|
break;
|
|
}
|
|
name++;
|
|
}
|
|
if (*name) continue;
|
|
else {
|
|
PyObject*** argname = argnames;
|
|
while (argname != first_kw_arg) {
|
|
if ((**argname == key) || (
|
|
(CYTHON_COMPILING_IN_PYPY || PyString_GET_SIZE(**argname) == PyString_GET_SIZE(key))
|
|
&& _PyString_Eq(**argname, key))) {
|
|
goto arg_passed_twice;
|
|
}
|
|
argname++;
|
|
}
|
|
}
|
|
} else
|
|
#endif
|
|
if (likely(PyUnicode_Check(key))) {
|
|
while (*name) {
|
|
int cmp = (**name == key) ? 0 :
|
|
#if !CYTHON_COMPILING_IN_PYPY && PY_MAJOR_VERSION >= 3
|
|
(PyUnicode_GET_SIZE(**name) != PyUnicode_GET_SIZE(key)) ? 1 :
|
|
#endif
|
|
PyUnicode_Compare(**name, key);
|
|
if (cmp < 0 && unlikely(PyErr_Occurred())) goto bad;
|
|
if (cmp == 0) {
|
|
values[name-argnames] = value;
|
|
break;
|
|
}
|
|
name++;
|
|
}
|
|
if (*name) continue;
|
|
else {
|
|
PyObject*** argname = argnames;
|
|
while (argname != first_kw_arg) {
|
|
int cmp = (**argname == key) ? 0 :
|
|
#if !CYTHON_COMPILING_IN_PYPY && PY_MAJOR_VERSION >= 3
|
|
(PyUnicode_GET_SIZE(**argname) != PyUnicode_GET_SIZE(key)) ? 1 :
|
|
#endif
|
|
PyUnicode_Compare(**argname, key);
|
|
if (cmp < 0 && unlikely(PyErr_Occurred())) goto bad;
|
|
if (cmp == 0) goto arg_passed_twice;
|
|
argname++;
|
|
}
|
|
}
|
|
} else
|
|
goto invalid_keyword_type;
|
|
if (kwds2) {
|
|
if (unlikely(PyDict_SetItem(kwds2, key, value))) goto bad;
|
|
} else {
|
|
goto invalid_keyword;
|
|
}
|
|
}
|
|
return 0;
|
|
arg_passed_twice:
|
|
__Pyx_RaiseDoubleKeywordsError(function_name, key);
|
|
goto bad;
|
|
invalid_keyword_type:
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%.200s() keywords must be strings", function_name);
|
|
goto bad;
|
|
invalid_keyword:
|
|
PyErr_Format(PyExc_TypeError,
|
|
#if PY_MAJOR_VERSION < 3
|
|
"%.200s() got an unexpected keyword argument '%.200s'",
|
|
function_name, PyString_AsString(key));
|
|
#else
|
|
"%s() got an unexpected keyword argument '%U'",
|
|
function_name, key);
|
|
#endif
|
|
bad:
|
|
return -1;
|
|
}
|
|
|
|
static void __Pyx_RaiseArgumentTypeInvalid(const char* name, PyObject *obj, PyTypeObject *type) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"Argument '%.200s' has incorrect type (expected %.200s, got %.200s)",
|
|
name, type->tp_name, Py_TYPE(obj)->tp_name);
|
|
}
|
|
static CYTHON_INLINE int __Pyx_ArgTypeTest(PyObject *obj, PyTypeObject *type, int none_allowed,
|
|
const char *name, int exact)
|
|
{
|
|
if (unlikely(!type)) {
|
|
PyErr_SetString(PyExc_SystemError, "Missing type object");
|
|
return 0;
|
|
}
|
|
if (none_allowed && obj == Py_None) return 1;
|
|
else if (exact) {
|
|
if (likely(Py_TYPE(obj) == type)) return 1;
|
|
#if PY_MAJOR_VERSION == 2
|
|
else if ((type == &PyBaseString_Type) && likely(__Pyx_PyBaseString_CheckExact(obj))) return 1;
|
|
#endif
|
|
}
|
|
else {
|
|
if (likely(PyObject_TypeCheck(obj, type))) return 1;
|
|
}
|
|
__Pyx_RaiseArgumentTypeInvalid(name, obj, type);
|
|
return 0;
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_IsLittleEndian(void) {
|
|
unsigned int n = 1;
|
|
return *(unsigned char*)(&n) != 0;
|
|
}
|
|
static void __Pyx_BufFmt_Init(__Pyx_BufFmt_Context* ctx,
|
|
__Pyx_BufFmt_StackElem* stack,
|
|
__Pyx_TypeInfo* type) {
|
|
stack[0].field = &ctx->root;
|
|
stack[0].parent_offset = 0;
|
|
ctx->root.type = type;
|
|
ctx->root.name = "buffer dtype";
|
|
ctx->root.offset = 0;
|
|
ctx->head = stack;
|
|
ctx->head->field = &ctx->root;
|
|
ctx->fmt_offset = 0;
|
|
ctx->head->parent_offset = 0;
|
|
ctx->new_packmode = '@';
|
|
ctx->enc_packmode = '@';
|
|
ctx->new_count = 1;
|
|
ctx->enc_count = 0;
|
|
ctx->enc_type = 0;
|
|
ctx->is_complex = 0;
|
|
ctx->is_valid_array = 0;
|
|
ctx->struct_alignment = 0;
|
|
while (type->typegroup == 'S') {
|
|
++ctx->head;
|
|
ctx->head->field = type->fields;
|
|
ctx->head->parent_offset = 0;
|
|
type = type->fields->type;
|
|
}
|
|
}
|
|
static int __Pyx_BufFmt_ParseNumber(const char** ts) {
|
|
int count;
|
|
const char* t = *ts;
|
|
if (*t < '0' || *t > '9') {
|
|
return -1;
|
|
} else {
|
|
count = *t++ - '0';
|
|
while (*t >= '0' && *t < '9') {
|
|
count *= 10;
|
|
count += *t++ - '0';
|
|
}
|
|
}
|
|
*ts = t;
|
|
return count;
|
|
}
|
|
static int __Pyx_BufFmt_ExpectNumber(const char **ts) {
|
|
int number = __Pyx_BufFmt_ParseNumber(ts);
|
|
if (number == -1) /* First char was not a digit */
|
|
PyErr_Format(PyExc_ValueError,\
|
|
"Does not understand character buffer dtype format string ('%c')", **ts);
|
|
return number;
|
|
}
|
|
static void __Pyx_BufFmt_RaiseUnexpectedChar(char ch) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Unexpected format string character: '%c'", ch);
|
|
}
|
|
static const char* __Pyx_BufFmt_DescribeTypeChar(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case 'c': return "'char'";
|
|
case 'b': return "'signed char'";
|
|
case 'B': return "'unsigned char'";
|
|
case 'h': return "'short'";
|
|
case 'H': return "'unsigned short'";
|
|
case 'i': return "'int'";
|
|
case 'I': return "'unsigned int'";
|
|
case 'l': return "'long'";
|
|
case 'L': return "'unsigned long'";
|
|
case 'q': return "'long long'";
|
|
case 'Q': return "'unsigned long long'";
|
|
case 'f': return (is_complex ? "'complex float'" : "'float'");
|
|
case 'd': return (is_complex ? "'complex double'" : "'double'");
|
|
case 'g': return (is_complex ? "'complex long double'" : "'long double'");
|
|
case 'T': return "a struct";
|
|
case 'O': return "Python object";
|
|
case 'P': return "a pointer";
|
|
case 's': case 'p': return "a string";
|
|
case 0: return "end";
|
|
default: return "unparseable format string";
|
|
}
|
|
}
|
|
static size_t __Pyx_BufFmt_TypeCharToStandardSize(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case '?': case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return 2;
|
|
case 'i': case 'I': case 'l': case 'L': return 4;
|
|
case 'q': case 'Q': return 8;
|
|
case 'f': return (is_complex ? 8 : 4);
|
|
case 'd': return (is_complex ? 16 : 8);
|
|
case 'g': {
|
|
PyErr_SetString(PyExc_ValueError, "Python does not define a standard format string size for long double ('g')..");
|
|
return 0;
|
|
}
|
|
case 'O': case 'P': return sizeof(void*);
|
|
default:
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
static size_t __Pyx_BufFmt_TypeCharToNativeSize(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return sizeof(short);
|
|
case 'i': case 'I': return sizeof(int);
|
|
case 'l': case 'L': return sizeof(long);
|
|
#ifdef HAVE_LONG_LONG
|
|
case 'q': case 'Q': return sizeof(PY_LONG_LONG);
|
|
#endif
|
|
case 'f': return sizeof(float) * (is_complex ? 2 : 1);
|
|
case 'd': return sizeof(double) * (is_complex ? 2 : 1);
|
|
case 'g': return sizeof(long double) * (is_complex ? 2 : 1);
|
|
case 'O': case 'P': return sizeof(void*);
|
|
default: {
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
typedef struct { char c; short x; } __Pyx_st_short;
|
|
typedef struct { char c; int x; } __Pyx_st_int;
|
|
typedef struct { char c; long x; } __Pyx_st_long;
|
|
typedef struct { char c; float x; } __Pyx_st_float;
|
|
typedef struct { char c; double x; } __Pyx_st_double;
|
|
typedef struct { char c; long double x; } __Pyx_st_longdouble;
|
|
typedef struct { char c; void *x; } __Pyx_st_void_p;
|
|
#ifdef HAVE_LONG_LONG
|
|
typedef struct { char c; PY_LONG_LONG x; } __Pyx_st_longlong;
|
|
#endif
|
|
static size_t __Pyx_BufFmt_TypeCharToAlignment(char ch, CYTHON_UNUSED int is_complex) {
|
|
switch (ch) {
|
|
case '?': case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return sizeof(__Pyx_st_short) - sizeof(short);
|
|
case 'i': case 'I': return sizeof(__Pyx_st_int) - sizeof(int);
|
|
case 'l': case 'L': return sizeof(__Pyx_st_long) - sizeof(long);
|
|
#ifdef HAVE_LONG_LONG
|
|
case 'q': case 'Q': return sizeof(__Pyx_st_longlong) - sizeof(PY_LONG_LONG);
|
|
#endif
|
|
case 'f': return sizeof(__Pyx_st_float) - sizeof(float);
|
|
case 'd': return sizeof(__Pyx_st_double) - sizeof(double);
|
|
case 'g': return sizeof(__Pyx_st_longdouble) - sizeof(long double);
|
|
case 'P': case 'O': return sizeof(__Pyx_st_void_p) - sizeof(void*);
|
|
default:
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
/* These are for computing the padding at the end of the struct to align
|
|
on the first member of the struct. This will probably the same as above,
|
|
but we don't have any guarantees.
|
|
*/
|
|
typedef struct { short x; char c; } __Pyx_pad_short;
|
|
typedef struct { int x; char c; } __Pyx_pad_int;
|
|
typedef struct { long x; char c; } __Pyx_pad_long;
|
|
typedef struct { float x; char c; } __Pyx_pad_float;
|
|
typedef struct { double x; char c; } __Pyx_pad_double;
|
|
typedef struct { long double x; char c; } __Pyx_pad_longdouble;
|
|
typedef struct { void *x; char c; } __Pyx_pad_void_p;
|
|
#ifdef HAVE_LONG_LONG
|
|
typedef struct { PY_LONG_LONG x; char c; } __Pyx_pad_longlong;
|
|
#endif
|
|
static size_t __Pyx_BufFmt_TypeCharToPadding(char ch, CYTHON_UNUSED int is_complex) {
|
|
switch (ch) {
|
|
case '?': case 'c': case 'b': case 'B': case 's': case 'p': return 1;
|
|
case 'h': case 'H': return sizeof(__Pyx_pad_short) - sizeof(short);
|
|
case 'i': case 'I': return sizeof(__Pyx_pad_int) - sizeof(int);
|
|
case 'l': case 'L': return sizeof(__Pyx_pad_long) - sizeof(long);
|
|
#ifdef HAVE_LONG_LONG
|
|
case 'q': case 'Q': return sizeof(__Pyx_pad_longlong) - sizeof(PY_LONG_LONG);
|
|
#endif
|
|
case 'f': return sizeof(__Pyx_pad_float) - sizeof(float);
|
|
case 'd': return sizeof(__Pyx_pad_double) - sizeof(double);
|
|
case 'g': return sizeof(__Pyx_pad_longdouble) - sizeof(long double);
|
|
case 'P': case 'O': return sizeof(__Pyx_pad_void_p) - sizeof(void*);
|
|
default:
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
static char __Pyx_BufFmt_TypeCharToGroup(char ch, int is_complex) {
|
|
switch (ch) {
|
|
case 'c':
|
|
return 'H';
|
|
case 'b': case 'h': case 'i':
|
|
case 'l': case 'q': case 's': case 'p':
|
|
return 'I';
|
|
case 'B': case 'H': case 'I': case 'L': case 'Q':
|
|
return 'U';
|
|
case 'f': case 'd': case 'g':
|
|
return (is_complex ? 'C' : 'R');
|
|
case 'O':
|
|
return 'O';
|
|
case 'P':
|
|
return 'P';
|
|
default: {
|
|
__Pyx_BufFmt_RaiseUnexpectedChar(ch);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
static void __Pyx_BufFmt_RaiseExpected(__Pyx_BufFmt_Context* ctx) {
|
|
if (ctx->head == NULL || ctx->head->field == &ctx->root) {
|
|
const char* expected;
|
|
const char* quote;
|
|
if (ctx->head == NULL) {
|
|
expected = "end";
|
|
quote = "";
|
|
} else {
|
|
expected = ctx->head->field->type->name;
|
|
quote = "'";
|
|
}
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer dtype mismatch, expected %s%s%s but got %s",
|
|
quote, expected, quote,
|
|
__Pyx_BufFmt_DescribeTypeChar(ctx->enc_type, ctx->is_complex));
|
|
} else {
|
|
__Pyx_StructField* field = ctx->head->field;
|
|
__Pyx_StructField* parent = (ctx->head - 1)->field;
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer dtype mismatch, expected '%s' but got %s in '%s.%s'",
|
|
field->type->name, __Pyx_BufFmt_DescribeTypeChar(ctx->enc_type, ctx->is_complex),
|
|
parent->type->name, field->name);
|
|
}
|
|
}
|
|
static int __Pyx_BufFmt_ProcessTypeChunk(__Pyx_BufFmt_Context* ctx) {
|
|
char group;
|
|
size_t size, offset, arraysize = 1;
|
|
if (ctx->enc_type == 0) return 0;
|
|
if (ctx->head->field->type->arraysize[0]) {
|
|
int i, ndim = 0;
|
|
if (ctx->enc_type == 's' || ctx->enc_type == 'p') {
|
|
ctx->is_valid_array = ctx->head->field->type->ndim == 1;
|
|
ndim = 1;
|
|
if (ctx->enc_count != ctx->head->field->type->arraysize[0]) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Expected a dimension of size %zu, got %zu",
|
|
ctx->head->field->type->arraysize[0], ctx->enc_count);
|
|
return -1;
|
|
}
|
|
}
|
|
if (!ctx->is_valid_array) {
|
|
PyErr_Format(PyExc_ValueError, "Expected %d dimensions, got %d",
|
|
ctx->head->field->type->ndim, ndim);
|
|
return -1;
|
|
}
|
|
for (i = 0; i < ctx->head->field->type->ndim; i++) {
|
|
arraysize *= ctx->head->field->type->arraysize[i];
|
|
}
|
|
ctx->is_valid_array = 0;
|
|
ctx->enc_count = 1;
|
|
}
|
|
group = __Pyx_BufFmt_TypeCharToGroup(ctx->enc_type, ctx->is_complex);
|
|
do {
|
|
__Pyx_StructField* field = ctx->head->field;
|
|
__Pyx_TypeInfo* type = field->type;
|
|
if (ctx->enc_packmode == '@' || ctx->enc_packmode == '^') {
|
|
size = __Pyx_BufFmt_TypeCharToNativeSize(ctx->enc_type, ctx->is_complex);
|
|
} else {
|
|
size = __Pyx_BufFmt_TypeCharToStandardSize(ctx->enc_type, ctx->is_complex);
|
|
}
|
|
if (ctx->enc_packmode == '@') {
|
|
size_t align_at = __Pyx_BufFmt_TypeCharToAlignment(ctx->enc_type, ctx->is_complex);
|
|
size_t align_mod_offset;
|
|
if (align_at == 0) return -1;
|
|
align_mod_offset = ctx->fmt_offset % align_at;
|
|
if (align_mod_offset > 0) ctx->fmt_offset += align_at - align_mod_offset;
|
|
if (ctx->struct_alignment == 0)
|
|
ctx->struct_alignment = __Pyx_BufFmt_TypeCharToPadding(ctx->enc_type,
|
|
ctx->is_complex);
|
|
}
|
|
if (type->size != size || type->typegroup != group) {
|
|
if (type->typegroup == 'C' && type->fields != NULL) {
|
|
size_t parent_offset = ctx->head->parent_offset + field->offset;
|
|
++ctx->head;
|
|
ctx->head->field = type->fields;
|
|
ctx->head->parent_offset = parent_offset;
|
|
continue;
|
|
}
|
|
if ((type->typegroup == 'H' || group == 'H') && type->size == size) {
|
|
} else {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return -1;
|
|
}
|
|
}
|
|
offset = ctx->head->parent_offset + field->offset;
|
|
if (ctx->fmt_offset != offset) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer dtype mismatch; next field is at offset %" CYTHON_FORMAT_SSIZE_T "d but %" CYTHON_FORMAT_SSIZE_T "d expected",
|
|
(Py_ssize_t)ctx->fmt_offset, (Py_ssize_t)offset);
|
|
return -1;
|
|
}
|
|
ctx->fmt_offset += size;
|
|
if (arraysize)
|
|
ctx->fmt_offset += (arraysize - 1) * size;
|
|
--ctx->enc_count; /* Consume from buffer string */
|
|
while (1) {
|
|
if (field == &ctx->root) {
|
|
ctx->head = NULL;
|
|
if (ctx->enc_count != 0) {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return -1;
|
|
}
|
|
break; /* breaks both loops as ctx->enc_count == 0 */
|
|
}
|
|
ctx->head->field = ++field;
|
|
if (field->type == NULL) {
|
|
--ctx->head;
|
|
field = ctx->head->field;
|
|
continue;
|
|
} else if (field->type->typegroup == 'S') {
|
|
size_t parent_offset = ctx->head->parent_offset + field->offset;
|
|
if (field->type->fields->type == NULL) continue; /* empty struct */
|
|
field = field->type->fields;
|
|
++ctx->head;
|
|
ctx->head->field = field;
|
|
ctx->head->parent_offset = parent_offset;
|
|
break;
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
} while (ctx->enc_count);
|
|
ctx->enc_type = 0;
|
|
ctx->is_complex = 0;
|
|
return 0;
|
|
}
|
|
static CYTHON_INLINE PyObject *
|
|
__pyx_buffmt_parse_array(__Pyx_BufFmt_Context* ctx, const char** tsp)
|
|
{
|
|
const char *ts = *tsp;
|
|
int i = 0, number;
|
|
int ndim = ctx->head->field->type->ndim;
|
|
;
|
|
++ts;
|
|
if (ctx->new_count != 1) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Cannot handle repeated arrays in format string");
|
|
return NULL;
|
|
}
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
while (*ts && *ts != ')') {
|
|
switch (*ts) {
|
|
case ' ': case '\f': case '\r': case '\n': case '\t': case '\v': continue;
|
|
default: break; /* not a 'break' in the loop */
|
|
}
|
|
number = __Pyx_BufFmt_ExpectNumber(&ts);
|
|
if (number == -1) return NULL;
|
|
if (i < ndim && (size_t) number != ctx->head->field->type->arraysize[i])
|
|
return PyErr_Format(PyExc_ValueError,
|
|
"Expected a dimension of size %zu, got %d",
|
|
ctx->head->field->type->arraysize[i], number);
|
|
if (*ts != ',' && *ts != ')')
|
|
return PyErr_Format(PyExc_ValueError,
|
|
"Expected a comma in format string, got '%c'", *ts);
|
|
if (*ts == ',') ts++;
|
|
i++;
|
|
}
|
|
if (i != ndim)
|
|
return PyErr_Format(PyExc_ValueError, "Expected %d dimension(s), got %d",
|
|
ctx->head->field->type->ndim, i);
|
|
if (!*ts) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Unexpected end of format string, expected ')'");
|
|
return NULL;
|
|
}
|
|
ctx->is_valid_array = 1;
|
|
ctx->new_count = 1;
|
|
*tsp = ++ts;
|
|
return Py_None;
|
|
}
|
|
static const char* __Pyx_BufFmt_CheckString(__Pyx_BufFmt_Context* ctx, const char* ts) {
|
|
int got_Z = 0;
|
|
while (1) {
|
|
switch(*ts) {
|
|
case 0:
|
|
if (ctx->enc_type != 0 && ctx->head == NULL) {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return NULL;
|
|
}
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
if (ctx->head != NULL) {
|
|
__Pyx_BufFmt_RaiseExpected(ctx);
|
|
return NULL;
|
|
}
|
|
return ts;
|
|
case ' ':
|
|
case '\r':
|
|
case '\n':
|
|
++ts;
|
|
break;
|
|
case '<':
|
|
if (!__Pyx_IsLittleEndian()) {
|
|
PyErr_SetString(PyExc_ValueError, "Little-endian buffer not supported on big-endian compiler");
|
|
return NULL;
|
|
}
|
|
ctx->new_packmode = '=';
|
|
++ts;
|
|
break;
|
|
case '>':
|
|
case '!':
|
|
if (__Pyx_IsLittleEndian()) {
|
|
PyErr_SetString(PyExc_ValueError, "Big-endian buffer not supported on little-endian compiler");
|
|
return NULL;
|
|
}
|
|
ctx->new_packmode = '=';
|
|
++ts;
|
|
break;
|
|
case '=':
|
|
case '@':
|
|
case '^':
|
|
ctx->new_packmode = *ts++;
|
|
break;
|
|
case 'T': /* substruct */
|
|
{
|
|
const char* ts_after_sub;
|
|
size_t i, struct_count = ctx->new_count;
|
|
size_t struct_alignment = ctx->struct_alignment;
|
|
ctx->new_count = 1;
|
|
++ts;
|
|
if (*ts != '{') {
|
|
PyErr_SetString(PyExc_ValueError, "Buffer acquisition: Expected '{' after 'T'");
|
|
return NULL;
|
|
}
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->enc_type = 0; /* Erase processed last struct element */
|
|
ctx->enc_count = 0;
|
|
ctx->struct_alignment = 0;
|
|
++ts;
|
|
ts_after_sub = ts;
|
|
for (i = 0; i != struct_count; ++i) {
|
|
ts_after_sub = __Pyx_BufFmt_CheckString(ctx, ts);
|
|
if (!ts_after_sub) return NULL;
|
|
}
|
|
ts = ts_after_sub;
|
|
if (struct_alignment) ctx->struct_alignment = struct_alignment;
|
|
}
|
|
break;
|
|
case '}': /* end of substruct; either repeat or move on */
|
|
{
|
|
size_t alignment = ctx->struct_alignment;
|
|
++ts;
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->enc_type = 0; /* Erase processed last struct element */
|
|
if (alignment && ctx->fmt_offset % alignment) {
|
|
ctx->fmt_offset += alignment - (ctx->fmt_offset % alignment);
|
|
}
|
|
}
|
|
return ts;
|
|
case 'x':
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->fmt_offset += ctx->new_count;
|
|
ctx->new_count = 1;
|
|
ctx->enc_count = 0;
|
|
ctx->enc_type = 0;
|
|
ctx->enc_packmode = ctx->new_packmode;
|
|
++ts;
|
|
break;
|
|
case 'Z':
|
|
got_Z = 1;
|
|
++ts;
|
|
if (*ts != 'f' && *ts != 'd' && *ts != 'g') {
|
|
__Pyx_BufFmt_RaiseUnexpectedChar('Z');
|
|
return NULL;
|
|
}
|
|
case 'c': case 'b': case 'B': case 'h': case 'H': case 'i': case 'I':
|
|
case 'l': case 'L': case 'q': case 'Q':
|
|
case 'f': case 'd': case 'g':
|
|
case 'O': case 'p':
|
|
if (ctx->enc_type == *ts && got_Z == ctx->is_complex &&
|
|
ctx->enc_packmode == ctx->new_packmode) {
|
|
ctx->enc_count += ctx->new_count;
|
|
ctx->new_count = 1;
|
|
got_Z = 0;
|
|
++ts;
|
|
break;
|
|
}
|
|
case 's':
|
|
if (__Pyx_BufFmt_ProcessTypeChunk(ctx) == -1) return NULL;
|
|
ctx->enc_count = ctx->new_count;
|
|
ctx->enc_packmode = ctx->new_packmode;
|
|
ctx->enc_type = *ts;
|
|
ctx->is_complex = got_Z;
|
|
++ts;
|
|
ctx->new_count = 1;
|
|
got_Z = 0;
|
|
break;
|
|
case ':':
|
|
++ts;
|
|
while(*ts != ':') ++ts;
|
|
++ts;
|
|
break;
|
|
case '(':
|
|
if (!__pyx_buffmt_parse_array(ctx, &ts)) return NULL;
|
|
break;
|
|
default:
|
|
{
|
|
int number = __Pyx_BufFmt_ExpectNumber(&ts);
|
|
if (number == -1) return NULL;
|
|
ctx->new_count = (size_t)number;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
static CYTHON_INLINE void __Pyx_ZeroBuffer(Py_buffer* buf) {
|
|
buf->buf = NULL;
|
|
buf->obj = NULL;
|
|
buf->strides = __Pyx_zeros;
|
|
buf->shape = __Pyx_zeros;
|
|
buf->suboffsets = __Pyx_minusones;
|
|
}
|
|
static CYTHON_INLINE int __Pyx_GetBufferAndValidate(
|
|
Py_buffer* buf, PyObject* obj, __Pyx_TypeInfo* dtype, int flags,
|
|
int nd, int cast, __Pyx_BufFmt_StackElem* stack)
|
|
{
|
|
if (obj == Py_None || obj == NULL) {
|
|
__Pyx_ZeroBuffer(buf);
|
|
return 0;
|
|
}
|
|
buf->buf = NULL;
|
|
if (__Pyx_GetBuffer(obj, buf, flags) == -1) goto fail;
|
|
if (buf->ndim != nd) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Buffer has wrong number of dimensions (expected %d, got %d)",
|
|
nd, buf->ndim);
|
|
goto fail;
|
|
}
|
|
if (!cast) {
|
|
__Pyx_BufFmt_Context ctx;
|
|
__Pyx_BufFmt_Init(&ctx, stack, dtype);
|
|
if (!__Pyx_BufFmt_CheckString(&ctx, buf->format)) goto fail;
|
|
}
|
|
if ((unsigned)buf->itemsize != dtype->size) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"Item size of buffer (%" CYTHON_FORMAT_SSIZE_T "d byte%s) does not match size of '%s' (%" CYTHON_FORMAT_SSIZE_T "d byte%s)",
|
|
buf->itemsize, (buf->itemsize > 1) ? "s" : "",
|
|
dtype->name, (Py_ssize_t)dtype->size, (dtype->size > 1) ? "s" : "");
|
|
goto fail;
|
|
}
|
|
if (buf->suboffsets == NULL) buf->suboffsets = __Pyx_minusones;
|
|
return 0;
|
|
fail:;
|
|
__Pyx_ZeroBuffer(buf);
|
|
return -1;
|
|
}
|
|
static CYTHON_INLINE void __Pyx_SafeReleaseBuffer(Py_buffer* info) {
|
|
if (info->buf == NULL) return;
|
|
if (info->suboffsets == __Pyx_minusones) info->suboffsets = NULL;
|
|
__Pyx_ReleaseBuffer(info);
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_ErrRestore(PyObject *type, PyObject *value, PyObject *tb) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
PyObject *tmp_type, *tmp_value, *tmp_tb;
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
tmp_type = tstate->curexc_type;
|
|
tmp_value = tstate->curexc_value;
|
|
tmp_tb = tstate->curexc_traceback;
|
|
tstate->curexc_type = type;
|
|
tstate->curexc_value = value;
|
|
tstate->curexc_traceback = tb;
|
|
Py_XDECREF(tmp_type);
|
|
Py_XDECREF(tmp_value);
|
|
Py_XDECREF(tmp_tb);
|
|
#else
|
|
PyErr_Restore(type, value, tb);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE void __Pyx_ErrFetch(PyObject **type, PyObject **value, PyObject **tb) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
*type = tstate->curexc_type;
|
|
*value = tstate->curexc_value;
|
|
*tb = tstate->curexc_traceback;
|
|
tstate->curexc_type = 0;
|
|
tstate->curexc_value = 0;
|
|
tstate->curexc_traceback = 0;
|
|
#else
|
|
PyErr_Fetch(type, value, tb);
|
|
#endif
|
|
}
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
static CYTHON_INLINE PyObject* __Pyx_PyObject_Call(PyObject *func, PyObject *arg, PyObject *kw) {
|
|
PyObject *result;
|
|
ternaryfunc call = func->ob_type->tp_call;
|
|
if (unlikely(!call))
|
|
return PyObject_Call(func, arg, kw);
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
if (unlikely(Py_EnterRecursiveCall((char*)" while calling a Python object")))
|
|
return NULL;
|
|
#endif
|
|
result = (*call)(func, arg, kw);
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
Py_LeaveRecursiveCall();
|
|
#endif
|
|
if (unlikely(!result) && unlikely(!PyErr_Occurred())) {
|
|
PyErr_SetString(
|
|
PyExc_SystemError,
|
|
"NULL result without error in PyObject_Call");
|
|
}
|
|
return result;
|
|
}
|
|
#endif
|
|
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Generic(PyObject *o, PyObject* j) {
|
|
PyObject *r;
|
|
if (!j) return NULL;
|
|
r = PyObject_GetItem(o, j);
|
|
Py_DECREF(j);
|
|
return r;
|
|
}
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_List_Fast(PyObject *o, Py_ssize_t i,
|
|
int wraparound, int boundscheck) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
if (wraparound & unlikely(i < 0)) i += PyList_GET_SIZE(o);
|
|
if ((!boundscheck) || likely((0 <= i) & (i < PyList_GET_SIZE(o)))) {
|
|
PyObject *r = PyList_GET_ITEM(o, i);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
return __Pyx_GetItemInt_Generic(o, PyInt_FromSsize_t(i));
|
|
#else
|
|
return PySequence_GetItem(o, i);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Tuple_Fast(PyObject *o, Py_ssize_t i,
|
|
int wraparound, int boundscheck) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
if (wraparound & unlikely(i < 0)) i += PyTuple_GET_SIZE(o);
|
|
if ((!boundscheck) || likely((0 <= i) & (i < PyTuple_GET_SIZE(o)))) {
|
|
PyObject *r = PyTuple_GET_ITEM(o, i);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
return __Pyx_GetItemInt_Generic(o, PyInt_FromSsize_t(i));
|
|
#else
|
|
return PySequence_GetItem(o, i);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE PyObject *__Pyx_GetItemInt_Fast(PyObject *o, Py_ssize_t i,
|
|
int is_list, int wraparound, int boundscheck) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
if (is_list || PyList_CheckExact(o)) {
|
|
Py_ssize_t n = ((!wraparound) | likely(i >= 0)) ? i : i + PyList_GET_SIZE(o);
|
|
if ((!boundscheck) || (likely((n >= 0) & (n < PyList_GET_SIZE(o))))) {
|
|
PyObject *r = PyList_GET_ITEM(o, n);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
}
|
|
else if (PyTuple_CheckExact(o)) {
|
|
Py_ssize_t n = ((!wraparound) | likely(i >= 0)) ? i : i + PyTuple_GET_SIZE(o);
|
|
if ((!boundscheck) || likely((n >= 0) & (n < PyTuple_GET_SIZE(o)))) {
|
|
PyObject *r = PyTuple_GET_ITEM(o, n);
|
|
Py_INCREF(r);
|
|
return r;
|
|
}
|
|
} else {
|
|
PySequenceMethods *m = Py_TYPE(o)->tp_as_sequence;
|
|
if (likely(m && m->sq_item)) {
|
|
if (wraparound && unlikely(i < 0) && likely(m->sq_length)) {
|
|
Py_ssize_t l = m->sq_length(o);
|
|
if (likely(l >= 0)) {
|
|
i += l;
|
|
} else {
|
|
if (PyErr_ExceptionMatches(PyExc_OverflowError))
|
|
PyErr_Clear();
|
|
else
|
|
return NULL;
|
|
}
|
|
}
|
|
return m->sq_item(o, i);
|
|
}
|
|
}
|
|
#else
|
|
if (is_list || PySequence_Check(o)) {
|
|
return PySequence_GetItem(o, i);
|
|
}
|
|
#endif
|
|
return __Pyx_GetItemInt_Generic(o, PyInt_FromSsize_t(i));
|
|
}
|
|
|
|
static void __Pyx_WriteUnraisable(const char *name, CYTHON_UNUSED int clineno,
|
|
CYTHON_UNUSED int lineno, CYTHON_UNUSED const char *filename,
|
|
int full_traceback) {
|
|
PyObject *old_exc, *old_val, *old_tb;
|
|
PyObject *ctx;
|
|
__Pyx_ErrFetch(&old_exc, &old_val, &old_tb);
|
|
if (full_traceback) {
|
|
Py_XINCREF(old_exc);
|
|
Py_XINCREF(old_val);
|
|
Py_XINCREF(old_tb);
|
|
__Pyx_ErrRestore(old_exc, old_val, old_tb);
|
|
PyErr_PrintEx(1);
|
|
}
|
|
#if PY_MAJOR_VERSION < 3
|
|
ctx = PyString_FromString(name);
|
|
#else
|
|
ctx = PyUnicode_FromString(name);
|
|
#endif
|
|
__Pyx_ErrRestore(old_exc, old_val, old_tb);
|
|
if (!ctx) {
|
|
PyErr_WriteUnraisable(Py_None);
|
|
} else {
|
|
PyErr_WriteUnraisable(ctx);
|
|
Py_DECREF(ctx);
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type) {
|
|
if (unlikely(!type)) {
|
|
PyErr_SetString(PyExc_SystemError, "Missing type object");
|
|
return 0;
|
|
}
|
|
if (likely(PyObject_TypeCheck(obj, type)))
|
|
return 1;
|
|
PyErr_Format(PyExc_TypeError, "Cannot convert %.200s to %.200s",
|
|
Py_TYPE(obj)->tp_name, type->tp_name);
|
|
return 0;
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject *__Pyx_GetModuleGlobalName(PyObject *name) {
|
|
PyObject *result;
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
result = PyDict_GetItem(__pyx_d, name);
|
|
if (result) {
|
|
Py_INCREF(result);
|
|
} else {
|
|
#else
|
|
result = PyObject_GetItem(__pyx_d, name);
|
|
if (!result) {
|
|
PyErr_Clear();
|
|
#endif
|
|
result = __Pyx_GetBuiltinName(name);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb,
|
|
CYTHON_UNUSED PyObject *cause) {
|
|
Py_XINCREF(type);
|
|
if (!value || value == Py_None)
|
|
value = NULL;
|
|
else
|
|
Py_INCREF(value);
|
|
if (!tb || tb == Py_None)
|
|
tb = NULL;
|
|
else {
|
|
Py_INCREF(tb);
|
|
if (!PyTraceBack_Check(tb)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: arg 3 must be a traceback or None");
|
|
goto raise_error;
|
|
}
|
|
}
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (PyClass_Check(type)) {
|
|
#else
|
|
if (PyType_Check(type)) {
|
|
#endif
|
|
#if CYTHON_COMPILING_IN_PYPY
|
|
if (!value) {
|
|
Py_INCREF(Py_None);
|
|
value = Py_None;
|
|
}
|
|
#endif
|
|
PyErr_NormalizeException(&type, &value, &tb);
|
|
} else {
|
|
if (value) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"instance exception may not have a separate value");
|
|
goto raise_error;
|
|
}
|
|
value = type;
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (PyInstance_Check(type)) {
|
|
type = (PyObject*) ((PyInstanceObject*)type)->in_class;
|
|
Py_INCREF(type);
|
|
} else {
|
|
type = 0;
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: exception must be an old-style class or instance");
|
|
goto raise_error;
|
|
}
|
|
#else
|
|
type = (PyObject*) Py_TYPE(type);
|
|
Py_INCREF(type);
|
|
if (!PyType_IsSubtype((PyTypeObject *)type, (PyTypeObject *)PyExc_BaseException)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: exception class must be a subclass of BaseException");
|
|
goto raise_error;
|
|
}
|
|
#endif
|
|
}
|
|
__Pyx_ErrRestore(type, value, tb);
|
|
return;
|
|
raise_error:
|
|
Py_XDECREF(value);
|
|
Py_XDECREF(type);
|
|
Py_XDECREF(tb);
|
|
return;
|
|
}
|
|
#else /* Python 3+ */
|
|
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause) {
|
|
PyObject* owned_instance = NULL;
|
|
if (tb == Py_None) {
|
|
tb = 0;
|
|
} else if (tb && !PyTraceBack_Check(tb)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: arg 3 must be a traceback or None");
|
|
goto bad;
|
|
}
|
|
if (value == Py_None)
|
|
value = 0;
|
|
if (PyExceptionInstance_Check(type)) {
|
|
if (value) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"instance exception may not have a separate value");
|
|
goto bad;
|
|
}
|
|
value = type;
|
|
type = (PyObject*) Py_TYPE(value);
|
|
} else if (PyExceptionClass_Check(type)) {
|
|
PyObject *instance_class = NULL;
|
|
if (value && PyExceptionInstance_Check(value)) {
|
|
instance_class = (PyObject*) Py_TYPE(value);
|
|
if (instance_class != type) {
|
|
if (PyObject_IsSubclass(instance_class, type)) {
|
|
type = instance_class;
|
|
} else {
|
|
instance_class = NULL;
|
|
}
|
|
}
|
|
}
|
|
if (!instance_class) {
|
|
PyObject *args;
|
|
if (!value)
|
|
args = PyTuple_New(0);
|
|
else if (PyTuple_Check(value)) {
|
|
Py_INCREF(value);
|
|
args = value;
|
|
} else
|
|
args = PyTuple_Pack(1, value);
|
|
if (!args)
|
|
goto bad;
|
|
owned_instance = PyObject_Call(type, args, NULL);
|
|
Py_DECREF(args);
|
|
if (!owned_instance)
|
|
goto bad;
|
|
value = owned_instance;
|
|
if (!PyExceptionInstance_Check(value)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"calling %R should have returned an instance of "
|
|
"BaseException, not %R",
|
|
type, Py_TYPE(value));
|
|
goto bad;
|
|
}
|
|
}
|
|
} else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"raise: exception class must be a subclass of BaseException");
|
|
goto bad;
|
|
}
|
|
#if PY_VERSION_HEX >= 0x03030000
|
|
if (cause) {
|
|
#else
|
|
if (cause && cause != Py_None) {
|
|
#endif
|
|
PyObject *fixed_cause;
|
|
if (cause == Py_None) {
|
|
fixed_cause = NULL;
|
|
} else if (PyExceptionClass_Check(cause)) {
|
|
fixed_cause = PyObject_CallObject(cause, NULL);
|
|
if (fixed_cause == NULL)
|
|
goto bad;
|
|
} else if (PyExceptionInstance_Check(cause)) {
|
|
fixed_cause = cause;
|
|
Py_INCREF(fixed_cause);
|
|
} else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"exception causes must derive from "
|
|
"BaseException");
|
|
goto bad;
|
|
}
|
|
PyException_SetCause(value, fixed_cause);
|
|
}
|
|
PyErr_SetObject(type, value);
|
|
if (tb) {
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
PyObject* tmp_tb = tstate->curexc_traceback;
|
|
if (tb != tmp_tb) {
|
|
Py_INCREF(tb);
|
|
tstate->curexc_traceback = tb;
|
|
Py_XDECREF(tmp_tb);
|
|
}
|
|
}
|
|
bad:
|
|
Py_XDECREF(owned_instance);
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseTooManyValuesError(Py_ssize_t expected) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"too many values to unpack (expected %" CYTHON_FORMAT_SSIZE_T "d)", expected);
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNeedMoreValuesError(Py_ssize_t index) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"need more than %" CYTHON_FORMAT_SSIZE_T "d value%.1s to unpack",
|
|
index, (index == 1) ? "" : "s");
|
|
}
|
|
|
|
static CYTHON_INLINE int __Pyx_IterFinish(void) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
PyThreadState *tstate = PyThreadState_GET();
|
|
PyObject* exc_type = tstate->curexc_type;
|
|
if (unlikely(exc_type)) {
|
|
if (likely(exc_type == PyExc_StopIteration) || PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration)) {
|
|
PyObject *exc_value, *exc_tb;
|
|
exc_value = tstate->curexc_value;
|
|
exc_tb = tstate->curexc_traceback;
|
|
tstate->curexc_type = 0;
|
|
tstate->curexc_value = 0;
|
|
tstate->curexc_traceback = 0;
|
|
Py_DECREF(exc_type);
|
|
Py_XDECREF(exc_value);
|
|
Py_XDECREF(exc_tb);
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
#else
|
|
if (unlikely(PyErr_Occurred())) {
|
|
if (likely(PyErr_ExceptionMatches(PyExc_StopIteration))) {
|
|
PyErr_Clear();
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
return 0;
|
|
#endif
|
|
}
|
|
|
|
static int __Pyx_IternextUnpackEndCheck(PyObject *retval, Py_ssize_t expected) {
|
|
if (unlikely(retval)) {
|
|
Py_DECREF(retval);
|
|
__Pyx_RaiseTooManyValuesError(expected);
|
|
return -1;
|
|
} else {
|
|
return __Pyx_IterFinish();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyObject_GetSlice(
|
|
PyObject* obj, Py_ssize_t cstart, Py_ssize_t cstop,
|
|
PyObject** _py_start, PyObject** _py_stop, PyObject** _py_slice,
|
|
int has_cstart, int has_cstop, CYTHON_UNUSED int wraparound) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
PyMappingMethods* mp;
|
|
#if PY_MAJOR_VERSION < 3
|
|
PySequenceMethods* ms = Py_TYPE(obj)->tp_as_sequence;
|
|
if (likely(ms && ms->sq_slice)) {
|
|
if (!has_cstart) {
|
|
if (_py_start && (*_py_start != Py_None)) {
|
|
cstart = __Pyx_PyIndex_AsSsize_t(*_py_start);
|
|
if ((cstart == (Py_ssize_t)-1) && PyErr_Occurred()) goto bad;
|
|
} else
|
|
cstart = 0;
|
|
}
|
|
if (!has_cstop) {
|
|
if (_py_stop && (*_py_stop != Py_None)) {
|
|
cstop = __Pyx_PyIndex_AsSsize_t(*_py_stop);
|
|
if ((cstop == (Py_ssize_t)-1) && PyErr_Occurred()) goto bad;
|
|
} else
|
|
cstop = PY_SSIZE_T_MAX;
|
|
}
|
|
if (wraparound && unlikely((cstart < 0) | (cstop < 0)) && likely(ms->sq_length)) {
|
|
Py_ssize_t l = ms->sq_length(obj);
|
|
if (likely(l >= 0)) {
|
|
if (cstop < 0) {
|
|
cstop += l;
|
|
if (cstop < 0) cstop = 0;
|
|
}
|
|
if (cstart < 0) {
|
|
cstart += l;
|
|
if (cstart < 0) cstart = 0;
|
|
}
|
|
} else {
|
|
if (PyErr_ExceptionMatches(PyExc_OverflowError))
|
|
PyErr_Clear();
|
|
else
|
|
goto bad;
|
|
}
|
|
}
|
|
return ms->sq_slice(obj, cstart, cstop);
|
|
}
|
|
#endif
|
|
mp = Py_TYPE(obj)->tp_as_mapping;
|
|
if (likely(mp && mp->mp_subscript))
|
|
#endif
|
|
{
|
|
PyObject* result;
|
|
PyObject *py_slice, *py_start, *py_stop;
|
|
if (_py_slice) {
|
|
py_slice = *_py_slice;
|
|
} else {
|
|
PyObject* owned_start = NULL;
|
|
PyObject* owned_stop = NULL;
|
|
if (_py_start) {
|
|
py_start = *_py_start;
|
|
} else {
|
|
if (has_cstart) {
|
|
owned_start = py_start = PyInt_FromSsize_t(cstart);
|
|
if (unlikely(!py_start)) goto bad;
|
|
} else
|
|
py_start = Py_None;
|
|
}
|
|
if (_py_stop) {
|
|
py_stop = *_py_stop;
|
|
} else {
|
|
if (has_cstop) {
|
|
owned_stop = py_stop = PyInt_FromSsize_t(cstop);
|
|
if (unlikely(!py_stop)) {
|
|
Py_XDECREF(owned_start);
|
|
goto bad;
|
|
}
|
|
} else
|
|
py_stop = Py_None;
|
|
}
|
|
py_slice = PySlice_New(py_start, py_stop, Py_None);
|
|
Py_XDECREF(owned_start);
|
|
Py_XDECREF(owned_stop);
|
|
if (unlikely(!py_slice)) goto bad;
|
|
}
|
|
#if CYTHON_COMPILING_IN_CPYTHON
|
|
result = mp->mp_subscript(obj, py_slice);
|
|
#else
|
|
result = PyObject_GetItem(obj, py_slice);
|
|
#endif
|
|
if (!_py_slice) {
|
|
Py_DECREF(py_slice);
|
|
}
|
|
return result;
|
|
}
|
|
PyErr_Format(PyExc_TypeError,
|
|
"'%.200s' object is unsliceable", Py_TYPE(obj)->tp_name);
|
|
bad:
|
|
return NULL;
|
|
}
|
|
|
|
static void __Pyx_RaiseBufferFallbackError(void) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"Buffer acquisition failed on assignment; and then reacquiring the old buffer failed too!");
|
|
}
|
|
|
|
static CYTHON_INLINE void __Pyx_RaiseNoneNotIterableError(void) {
|
|
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable");
|
|
}
|
|
|
|
static int __Pyx_SetVtable(PyObject *dict, void *vtable) {
|
|
#if PY_VERSION_HEX >= 0x02070000 && !(PY_MAJOR_VERSION==3&&PY_MINOR_VERSION==0)
|
|
PyObject *ob = PyCapsule_New(vtable, 0, 0);
|
|
#else
|
|
PyObject *ob = PyCObject_FromVoidPtr(vtable, 0);
|
|
#endif
|
|
if (!ob)
|
|
goto bad;
|
|
if (PyDict_SetItem(dict, __pyx_n_s_pyx_vtable, ob) < 0)
|
|
goto bad;
|
|
Py_DECREF(ob);
|
|
return 0;
|
|
bad:
|
|
Py_XDECREF(ob);
|
|
return -1;
|
|
}
|
|
|
|
static void* __Pyx_GetVtable(PyObject *dict) {
|
|
void* ptr;
|
|
PyObject *ob = PyObject_GetItem(dict, __pyx_n_s_pyx_vtable);
|
|
if (!ob)
|
|
goto bad;
|
|
#if PY_VERSION_HEX >= 0x02070000 && !(PY_MAJOR_VERSION==3&&PY_MINOR_VERSION==0)
|
|
ptr = PyCapsule_GetPointer(ob, 0);
|
|
#else
|
|
ptr = PyCObject_AsVoidPtr(ob);
|
|
#endif
|
|
if (!ptr && !PyErr_Occurred())
|
|
PyErr_SetString(PyExc_RuntimeError, "invalid vtable found for imported type");
|
|
Py_DECREF(ob);
|
|
return ptr;
|
|
bad:
|
|
Py_XDECREF(ob);
|
|
return NULL;
|
|
}
|
|
|
|
static PyObject* __Pyx_ImportFrom(PyObject* module, PyObject* name) {
|
|
PyObject* value = __Pyx_PyObject_GetAttrStr(module, name);
|
|
if (unlikely(!value) && PyErr_ExceptionMatches(PyExc_AttributeError)) {
|
|
PyErr_Format(PyExc_ImportError,
|
|
#if PY_MAJOR_VERSION < 3
|
|
"cannot import name %.230s", PyString_AS_STRING(name));
|
|
#else
|
|
"cannot import name %S", name);
|
|
#endif
|
|
}
|
|
return value;
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_Py_intptr_t(Py_intptr_t value) {
|
|
const Py_intptr_t neg_one = (Py_intptr_t) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (is_unsigned) {
|
|
if (sizeof(Py_intptr_t) < sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(Py_intptr_t) <= sizeof(unsigned long)) {
|
|
return PyLong_FromUnsignedLong((unsigned long) value);
|
|
} else if (sizeof(Py_intptr_t) <= sizeof(unsigned long long)) {
|
|
return PyLong_FromUnsignedLongLong((unsigned long long) value);
|
|
}
|
|
} else {
|
|
if (sizeof(Py_intptr_t) <= sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(Py_intptr_t) <= sizeof(long long)) {
|
|
return PyLong_FromLongLong((long long) value);
|
|
}
|
|
}
|
|
{
|
|
int one = 1; int little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&value;
|
|
return _PyLong_FromByteArray(bytes, sizeof(Py_intptr_t),
|
|
little, !is_unsigned);
|
|
}
|
|
}
|
|
|
|
#define __PYX_VERIFY_RETURN_INT(target_type, func_type, func) \
|
|
{ \
|
|
func_type value = func(x); \
|
|
if (sizeof(target_type) < sizeof(func_type)) { \
|
|
if (unlikely(value != (func_type) (target_type) value)) { \
|
|
func_type zero = 0; \
|
|
PyErr_SetString(PyExc_OverflowError, \
|
|
(is_unsigned && unlikely(value < zero)) ? \
|
|
"can't convert negative value to " #target_type : \
|
|
"value too large to convert to " #target_type); \
|
|
return (target_type) -1; \
|
|
} \
|
|
} \
|
|
return (target_type) value; \
|
|
}
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
#include "longintrepr.h"
|
|
#endif
|
|
#endif
|
|
static CYTHON_INLINE Py_intptr_t __Pyx_PyInt_As_Py_intptr_t(PyObject *x) {
|
|
const Py_intptr_t neg_one = (Py_intptr_t) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyInt_Check(x))) {
|
|
if (sizeof(Py_intptr_t) < sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(Py_intptr_t, long, PyInt_AS_LONG)
|
|
} else {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to Py_intptr_t");
|
|
return (Py_intptr_t) -1;
|
|
}
|
|
return (Py_intptr_t) val;
|
|
}
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(Py_intptr_t)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return (Py_intptr_t) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to Py_intptr_t");
|
|
return (Py_intptr_t) -1;
|
|
}
|
|
if (sizeof(Py_intptr_t) <= sizeof(unsigned long)) {
|
|
__PYX_VERIFY_RETURN_INT(Py_intptr_t, unsigned long, PyLong_AsUnsignedLong)
|
|
} else if (sizeof(Py_intptr_t) <= sizeof(unsigned long long)) {
|
|
__PYX_VERIFY_RETURN_INT(Py_intptr_t, unsigned long long, PyLong_AsUnsignedLongLong)
|
|
}
|
|
} else {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(Py_intptr_t)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return +(Py_intptr_t) ((PyLongObject*)x)->ob_digit[0];
|
|
case -1: return -(Py_intptr_t) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (sizeof(Py_intptr_t) <= sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(Py_intptr_t, long, PyLong_AsLong)
|
|
} else if (sizeof(Py_intptr_t) <= sizeof(long long)) {
|
|
__PYX_VERIFY_RETURN_INT(Py_intptr_t, long long, PyLong_AsLongLong)
|
|
}
|
|
}
|
|
{
|
|
#if CYTHON_COMPILING_IN_PYPY && !defined(_PyLong_AsByteArray)
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"_PyLong_AsByteArray() not available in PyPy, cannot convert large numbers");
|
|
#else
|
|
Py_intptr_t val;
|
|
PyObject *v = __Pyx_PyNumber_Int(x);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(v) && !PyLong_Check(v)) {
|
|
PyObject *tmp = v;
|
|
v = PyNumber_Long(tmp);
|
|
Py_DECREF(tmp);
|
|
}
|
|
#endif
|
|
if (likely(v)) {
|
|
int one = 1; int is_little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&val;
|
|
int ret = _PyLong_AsByteArray((PyLongObject *)v,
|
|
bytes, sizeof(val),
|
|
is_little, !is_unsigned);
|
|
Py_DECREF(v);
|
|
if (likely(!ret))
|
|
return val;
|
|
}
|
|
#endif
|
|
return (Py_intptr_t) -1;
|
|
}
|
|
} else {
|
|
Py_intptr_t val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (Py_intptr_t) -1;
|
|
val = __Pyx_PyInt_As_Py_intptr_t(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, int level) {
|
|
PyObject *empty_list = 0;
|
|
PyObject *module = 0;
|
|
PyObject *global_dict = 0;
|
|
PyObject *empty_dict = 0;
|
|
PyObject *list;
|
|
#if PY_VERSION_HEX < 0x03030000
|
|
PyObject *py_import;
|
|
py_import = __Pyx_PyObject_GetAttrStr(__pyx_b, __pyx_n_s_import);
|
|
if (!py_import)
|
|
goto bad;
|
|
#endif
|
|
if (from_list)
|
|
list = from_list;
|
|
else {
|
|
empty_list = PyList_New(0);
|
|
if (!empty_list)
|
|
goto bad;
|
|
list = empty_list;
|
|
}
|
|
global_dict = PyModule_GetDict(__pyx_m);
|
|
if (!global_dict)
|
|
goto bad;
|
|
empty_dict = PyDict_New();
|
|
if (!empty_dict)
|
|
goto bad;
|
|
#if PY_VERSION_HEX >= 0x02050000
|
|
{
|
|
#if PY_MAJOR_VERSION >= 3
|
|
if (level == -1) {
|
|
if (strchr(__Pyx_MODULE_NAME, '.')) {
|
|
#if PY_VERSION_HEX < 0x03030000
|
|
PyObject *py_level = PyInt_FromLong(1);
|
|
if (!py_level)
|
|
goto bad;
|
|
module = PyObject_CallFunctionObjArgs(py_import,
|
|
name, global_dict, empty_dict, list, py_level, NULL);
|
|
Py_DECREF(py_level);
|
|
#else
|
|
module = PyImport_ImportModuleLevelObject(
|
|
name, global_dict, empty_dict, list, 1);
|
|
#endif
|
|
if (!module) {
|
|
if (!PyErr_ExceptionMatches(PyExc_ImportError))
|
|
goto bad;
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
level = 0; /* try absolute import on failure */
|
|
}
|
|
#endif
|
|
if (!module) {
|
|
#if PY_VERSION_HEX < 0x03030000
|
|
PyObject *py_level = PyInt_FromLong(level);
|
|
if (!py_level)
|
|
goto bad;
|
|
module = PyObject_CallFunctionObjArgs(py_import,
|
|
name, global_dict, empty_dict, list, py_level, NULL);
|
|
Py_DECREF(py_level);
|
|
#else
|
|
module = PyImport_ImportModuleLevelObject(
|
|
name, global_dict, empty_dict, list, level);
|
|
#endif
|
|
}
|
|
}
|
|
#else
|
|
if (level>0) {
|
|
PyErr_SetString(PyExc_RuntimeError, "Relative import is not supported for Python <=2.4.");
|
|
goto bad;
|
|
}
|
|
module = PyObject_CallFunctionObjArgs(py_import,
|
|
name, global_dict, empty_dict, list, NULL);
|
|
#endif
|
|
bad:
|
|
#if PY_VERSION_HEX < 0x03030000
|
|
Py_XDECREF(py_import);
|
|
#endif
|
|
Py_XDECREF(empty_list);
|
|
Py_XDECREF(empty_dict);
|
|
return module;
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION < 3
|
|
static int __Pyx_GetBuffer(PyObject *obj, Py_buffer *view, int flags) {
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
if (PyObject_CheckBuffer(obj)) return PyObject_GetBuffer(obj, view, flags);
|
|
#endif
|
|
if (PyObject_TypeCheck(obj, __pyx_ptype_5numpy_ndarray)) return __pyx_pw_5numpy_7ndarray_1__getbuffer__(obj, view, flags);
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
if (obj->ob_type->tp_dict) {
|
|
PyObject *getbuffer_cobj = PyObject_GetItem(
|
|
obj->ob_type->tp_dict, __pyx_n_s_pyx_getbuffer);
|
|
if (getbuffer_cobj) {
|
|
getbufferproc func = (getbufferproc) PyCObject_AsVoidPtr(getbuffer_cobj);
|
|
Py_DECREF(getbuffer_cobj);
|
|
if (!func)
|
|
goto fail;
|
|
return func(obj, view, flags);
|
|
} else {
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
#endif
|
|
PyErr_Format(PyExc_TypeError, "'%.200s' does not have the buffer interface", Py_TYPE(obj)->tp_name);
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
fail:
|
|
#endif
|
|
return -1;
|
|
}
|
|
static void __Pyx_ReleaseBuffer(Py_buffer *view) {
|
|
PyObject *obj = view->obj;
|
|
if (!obj) return;
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
if (PyObject_CheckBuffer(obj)) {
|
|
PyBuffer_Release(view);
|
|
return;
|
|
}
|
|
#endif
|
|
if (PyObject_TypeCheck(obj, __pyx_ptype_5numpy_ndarray)) { __pyx_pw_5numpy_7ndarray_3__releasebuffer__(obj, view); return; }
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
if (obj->ob_type->tp_dict) {
|
|
PyObject *releasebuffer_cobj = PyObject_GetItem(
|
|
obj->ob_type->tp_dict, __pyx_n_s_pyx_releasebuffer);
|
|
if (releasebuffer_cobj) {
|
|
releasebufferproc func = (releasebufferproc) PyCObject_AsVoidPtr(releasebuffer_cobj);
|
|
Py_DECREF(releasebuffer_cobj);
|
|
if (!func)
|
|
goto fail;
|
|
func(obj, view);
|
|
return;
|
|
} else {
|
|
PyErr_Clear();
|
|
}
|
|
}
|
|
#endif
|
|
goto nofail;
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
fail:
|
|
#endif
|
|
PyErr_WriteUnraisable(obj);
|
|
nofail:
|
|
Py_DECREF(obj);
|
|
view->obj = NULL;
|
|
}
|
|
#endif /* PY_MAJOR_VERSION < 3 */
|
|
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
#include "longintrepr.h"
|
|
#endif
|
|
#endif
|
|
static CYTHON_INLINE int __Pyx_PyInt_As_int(PyObject *x) {
|
|
const int neg_one = (int) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyInt_Check(x))) {
|
|
if (sizeof(int) < sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(int, long, PyInt_AS_LONG)
|
|
} else {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to int");
|
|
return (int) -1;
|
|
}
|
|
return (int) val;
|
|
}
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(int)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return (int) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to int");
|
|
return (int) -1;
|
|
}
|
|
if (sizeof(int) <= sizeof(unsigned long)) {
|
|
__PYX_VERIFY_RETURN_INT(int, unsigned long, PyLong_AsUnsignedLong)
|
|
} else if (sizeof(int) <= sizeof(unsigned long long)) {
|
|
__PYX_VERIFY_RETURN_INT(int, unsigned long long, PyLong_AsUnsignedLongLong)
|
|
}
|
|
} else {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(int)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return +(int) ((PyLongObject*)x)->ob_digit[0];
|
|
case -1: return -(int) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (sizeof(int) <= sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(int, long, PyLong_AsLong)
|
|
} else if (sizeof(int) <= sizeof(long long)) {
|
|
__PYX_VERIFY_RETURN_INT(int, long long, PyLong_AsLongLong)
|
|
}
|
|
}
|
|
{
|
|
#if CYTHON_COMPILING_IN_PYPY && !defined(_PyLong_AsByteArray)
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"_PyLong_AsByteArray() not available in PyPy, cannot convert large numbers");
|
|
#else
|
|
int val;
|
|
PyObject *v = __Pyx_PyNumber_Int(x);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(v) && !PyLong_Check(v)) {
|
|
PyObject *tmp = v;
|
|
v = PyNumber_Long(tmp);
|
|
Py_DECREF(tmp);
|
|
}
|
|
#endif
|
|
if (likely(v)) {
|
|
int one = 1; int is_little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&val;
|
|
int ret = _PyLong_AsByteArray((PyLongObject *)v,
|
|
bytes, sizeof(val),
|
|
is_little, !is_unsigned);
|
|
Py_DECREF(v);
|
|
if (likely(!ret))
|
|
return val;
|
|
}
|
|
#endif
|
|
return (int) -1;
|
|
}
|
|
} else {
|
|
int val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (int) -1;
|
|
val = __Pyx_PyInt_As_int(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_int(int value) {
|
|
const int neg_one = (int) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (is_unsigned) {
|
|
if (sizeof(int) < sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(int) <= sizeof(unsigned long)) {
|
|
return PyLong_FromUnsignedLong((unsigned long) value);
|
|
} else if (sizeof(int) <= sizeof(unsigned long long)) {
|
|
return PyLong_FromUnsignedLongLong((unsigned long long) value);
|
|
}
|
|
} else {
|
|
if (sizeof(int) <= sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(int) <= sizeof(long long)) {
|
|
return PyLong_FromLongLong((long long) value);
|
|
}
|
|
}
|
|
{
|
|
int one = 1; int little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&value;
|
|
return _PyLong_FromByteArray(bytes, sizeof(int),
|
|
little, !is_unsigned);
|
|
}
|
|
}
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
#include "longintrepr.h"
|
|
#endif
|
|
#endif
|
|
static CYTHON_INLINE npy_uint32 __Pyx_PyInt_As_npy_uint32(PyObject *x) {
|
|
const npy_uint32 neg_one = (npy_uint32) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyInt_Check(x))) {
|
|
if (sizeof(npy_uint32) < sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_uint32, long, PyInt_AS_LONG)
|
|
} else {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to npy_uint32");
|
|
return (npy_uint32) -1;
|
|
}
|
|
return (npy_uint32) val;
|
|
}
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(npy_uint32)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return (npy_uint32) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to npy_uint32");
|
|
return (npy_uint32) -1;
|
|
}
|
|
if (sizeof(npy_uint32) <= sizeof(unsigned long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_uint32, unsigned long, PyLong_AsUnsignedLong)
|
|
} else if (sizeof(npy_uint32) <= sizeof(unsigned long long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_uint32, unsigned long long, PyLong_AsUnsignedLongLong)
|
|
}
|
|
} else {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(npy_uint32)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return +(npy_uint32) ((PyLongObject*)x)->ob_digit[0];
|
|
case -1: return -(npy_uint32) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (sizeof(npy_uint32) <= sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_uint32, long, PyLong_AsLong)
|
|
} else if (sizeof(npy_uint32) <= sizeof(long long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_uint32, long long, PyLong_AsLongLong)
|
|
}
|
|
}
|
|
{
|
|
#if CYTHON_COMPILING_IN_PYPY && !defined(_PyLong_AsByteArray)
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"_PyLong_AsByteArray() not available in PyPy, cannot convert large numbers");
|
|
#else
|
|
npy_uint32 val;
|
|
PyObject *v = __Pyx_PyNumber_Int(x);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(v) && !PyLong_Check(v)) {
|
|
PyObject *tmp = v;
|
|
v = PyNumber_Long(tmp);
|
|
Py_DECREF(tmp);
|
|
}
|
|
#endif
|
|
if (likely(v)) {
|
|
int one = 1; int is_little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&val;
|
|
int ret = _PyLong_AsByteArray((PyLongObject *)v,
|
|
bytes, sizeof(val),
|
|
is_little, !is_unsigned);
|
|
Py_DECREF(v);
|
|
if (likely(!ret))
|
|
return val;
|
|
}
|
|
#endif
|
|
return (npy_uint32) -1;
|
|
}
|
|
} else {
|
|
npy_uint32 val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (npy_uint32) -1;
|
|
val = __Pyx_PyInt_As_npy_uint32(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_long(long value) {
|
|
const long neg_one = (long) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (is_unsigned) {
|
|
if (sizeof(long) < sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(long) <= sizeof(unsigned long)) {
|
|
return PyLong_FromUnsignedLong((unsigned long) value);
|
|
} else if (sizeof(long) <= sizeof(unsigned long long)) {
|
|
return PyLong_FromUnsignedLongLong((unsigned long long) value);
|
|
}
|
|
} else {
|
|
if (sizeof(long) <= sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(long) <= sizeof(long long)) {
|
|
return PyLong_FromLongLong((long long) value);
|
|
}
|
|
}
|
|
{
|
|
int one = 1; int little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&value;
|
|
return _PyLong_FromByteArray(bytes, sizeof(long),
|
|
little, !is_unsigned);
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyInt_From_npy_int32(npy_int32 value) {
|
|
const npy_int32 neg_one = (npy_int32) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
if (is_unsigned) {
|
|
if (sizeof(npy_int32) < sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(npy_int32) <= sizeof(unsigned long)) {
|
|
return PyLong_FromUnsignedLong((unsigned long) value);
|
|
} else if (sizeof(npy_int32) <= sizeof(unsigned long long)) {
|
|
return PyLong_FromUnsignedLongLong((unsigned long long) value);
|
|
}
|
|
} else {
|
|
if (sizeof(npy_int32) <= sizeof(long)) {
|
|
return PyInt_FromLong((long) value);
|
|
} else if (sizeof(npy_int32) <= sizeof(long long)) {
|
|
return PyLong_FromLongLong((long long) value);
|
|
}
|
|
}
|
|
{
|
|
int one = 1; int little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&value;
|
|
return _PyLong_FromByteArray(bytes, sizeof(npy_int32),
|
|
little, !is_unsigned);
|
|
}
|
|
}
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
#include "longintrepr.h"
|
|
#endif
|
|
#endif
|
|
static CYTHON_INLINE npy_int32 __Pyx_PyInt_As_npy_int32(PyObject *x) {
|
|
const npy_int32 neg_one = (npy_int32) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyInt_Check(x))) {
|
|
if (sizeof(npy_int32) < sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_int32, long, PyInt_AS_LONG)
|
|
} else {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to npy_int32");
|
|
return (npy_int32) -1;
|
|
}
|
|
return (npy_int32) val;
|
|
}
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(npy_int32)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return (npy_int32) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to npy_int32");
|
|
return (npy_int32) -1;
|
|
}
|
|
if (sizeof(npy_int32) <= sizeof(unsigned long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_int32, unsigned long, PyLong_AsUnsignedLong)
|
|
} else if (sizeof(npy_int32) <= sizeof(unsigned long long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_int32, unsigned long long, PyLong_AsUnsignedLongLong)
|
|
}
|
|
} else {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(npy_int32)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return +(npy_int32) ((PyLongObject*)x)->ob_digit[0];
|
|
case -1: return -(npy_int32) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (sizeof(npy_int32) <= sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_int32, long, PyLong_AsLong)
|
|
} else if (sizeof(npy_int32) <= sizeof(long long)) {
|
|
__PYX_VERIFY_RETURN_INT(npy_int32, long long, PyLong_AsLongLong)
|
|
}
|
|
}
|
|
{
|
|
#if CYTHON_COMPILING_IN_PYPY && !defined(_PyLong_AsByteArray)
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"_PyLong_AsByteArray() not available in PyPy, cannot convert large numbers");
|
|
#else
|
|
npy_int32 val;
|
|
PyObject *v = __Pyx_PyNumber_Int(x);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(v) && !PyLong_Check(v)) {
|
|
PyObject *tmp = v;
|
|
v = PyNumber_Long(tmp);
|
|
Py_DECREF(tmp);
|
|
}
|
|
#endif
|
|
if (likely(v)) {
|
|
int one = 1; int is_little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&val;
|
|
int ret = _PyLong_AsByteArray((PyLongObject *)v,
|
|
bytes, sizeof(val),
|
|
is_little, !is_unsigned);
|
|
Py_DECREF(v);
|
|
if (likely(!ret))
|
|
return val;
|
|
}
|
|
#endif
|
|
return (npy_int32) -1;
|
|
}
|
|
} else {
|
|
npy_int32 val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (npy_int32) -1;
|
|
val = __Pyx_PyInt_As_npy_int32(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static CYTHON_INLINE long __Pyx_pow_long(long b, long e) {
|
|
long t = b;
|
|
switch (e) {
|
|
case 3:
|
|
t *= b;
|
|
case 2:
|
|
t *= b;
|
|
case 1:
|
|
return t;
|
|
case 0:
|
|
return 1;
|
|
}
|
|
#if 1
|
|
if (unlikely(e<0)) return 0;
|
|
#endif
|
|
t = 1;
|
|
while (likely(e)) {
|
|
t *= (b * (e&1)) | ((~e)&1); /* 1 or b */
|
|
b *= b;
|
|
e >>= 1;
|
|
}
|
|
return t;
|
|
}
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float x, float y) {
|
|
return ::std::complex< float >(x, y);
|
|
}
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float x, float y) {
|
|
return x + y*(__pyx_t_float_complex)_Complex_I;
|
|
}
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_float_complex __pyx_t_float_complex_from_parts(float x, float y) {
|
|
__pyx_t_float_complex z;
|
|
z.real = x;
|
|
z.imag = y;
|
|
return z;
|
|
}
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eqf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
return (a.real == b.real) && (a.imag == b.imag);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_sumf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real + b.real;
|
|
z.imag = a.imag + b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_difff(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real - b.real;
|
|
z.imag = a.imag - b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_prodf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real * b.real - a.imag * b.imag;
|
|
z.imag = a.real * b.imag + a.imag * b.real;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_quotf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
float denom = b.real * b.real + b.imag * b.imag;
|
|
z.real = (a.real * b.real + a.imag * b.imag) / denom;
|
|
z.imag = (a.imag * b.real - a.real * b.imag) / denom;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_negf(__pyx_t_float_complex a) {
|
|
__pyx_t_float_complex z;
|
|
z.real = -a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE int __Pyx_c_is_zerof(__pyx_t_float_complex a) {
|
|
return (a.real == 0) && (a.imag == 0);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_conjf(__pyx_t_float_complex a) {
|
|
__pyx_t_float_complex z;
|
|
z.real = a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
#if 1
|
|
static CYTHON_INLINE float __Pyx_c_absf(__pyx_t_float_complex z) {
|
|
#if !defined(HAVE_HYPOT) || defined(_MSC_VER)
|
|
return sqrtf(z.real*z.real + z.imag*z.imag);
|
|
#else
|
|
return hypotf(z.real, z.imag);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE __pyx_t_float_complex __Pyx_c_powf(__pyx_t_float_complex a, __pyx_t_float_complex b) {
|
|
__pyx_t_float_complex z;
|
|
float r, lnr, theta, z_r, z_theta;
|
|
if (b.imag == 0 && b.real == (int)b.real) {
|
|
if (b.real < 0) {
|
|
float denom = a.real * a.real + a.imag * a.imag;
|
|
a.real = a.real / denom;
|
|
a.imag = -a.imag / denom;
|
|
b.real = -b.real;
|
|
}
|
|
switch ((int)b.real) {
|
|
case 0:
|
|
z.real = 1;
|
|
z.imag = 0;
|
|
return z;
|
|
case 1:
|
|
return a;
|
|
case 2:
|
|
z = __Pyx_c_prodf(a, a);
|
|
return __Pyx_c_prodf(a, a);
|
|
case 3:
|
|
z = __Pyx_c_prodf(a, a);
|
|
return __Pyx_c_prodf(z, a);
|
|
case 4:
|
|
z = __Pyx_c_prodf(a, a);
|
|
return __Pyx_c_prodf(z, z);
|
|
}
|
|
}
|
|
if (a.imag == 0) {
|
|
if (a.real == 0) {
|
|
return a;
|
|
}
|
|
r = a.real;
|
|
theta = 0;
|
|
} else {
|
|
r = __Pyx_c_absf(a);
|
|
theta = atan2f(a.imag, a.real);
|
|
}
|
|
lnr = logf(r);
|
|
z_r = expf(lnr * b.real - theta * b.imag);
|
|
z_theta = theta * b.real + lnr * b.imag;
|
|
z.real = z_r * cosf(z_theta);
|
|
z.imag = z_r * sinf(z_theta);
|
|
return z;
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#ifdef __cplusplus
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double x, double y) {
|
|
return ::std::complex< double >(x, y);
|
|
}
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double x, double y) {
|
|
return x + y*(__pyx_t_double_complex)_Complex_I;
|
|
}
|
|
#endif
|
|
#else
|
|
static CYTHON_INLINE __pyx_t_double_complex __pyx_t_double_complex_from_parts(double x, double y) {
|
|
__pyx_t_double_complex z;
|
|
z.real = x;
|
|
z.imag = y;
|
|
return z;
|
|
}
|
|
#endif
|
|
|
|
#if CYTHON_CCOMPLEX
|
|
#else
|
|
static CYTHON_INLINE int __Pyx_c_eq(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
return (a.real == b.real) && (a.imag == b.imag);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_sum(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real + b.real;
|
|
z.imag = a.imag + b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_diff(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real - b.real;
|
|
z.imag = a.imag - b.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_prod(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real * b.real - a.imag * b.imag;
|
|
z.imag = a.real * b.imag + a.imag * b.real;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_quot(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
double denom = b.real * b.real + b.imag * b.imag;
|
|
z.real = (a.real * b.real + a.imag * b.imag) / denom;
|
|
z.imag = (a.imag * b.real - a.real * b.imag) / denom;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_neg(__pyx_t_double_complex a) {
|
|
__pyx_t_double_complex z;
|
|
z.real = -a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
static CYTHON_INLINE int __Pyx_c_is_zero(__pyx_t_double_complex a) {
|
|
return (a.real == 0) && (a.imag == 0);
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_conj(__pyx_t_double_complex a) {
|
|
__pyx_t_double_complex z;
|
|
z.real = a.real;
|
|
z.imag = -a.imag;
|
|
return z;
|
|
}
|
|
#if 1
|
|
static CYTHON_INLINE double __Pyx_c_abs(__pyx_t_double_complex z) {
|
|
#if !defined(HAVE_HYPOT) || defined(_MSC_VER)
|
|
return sqrt(z.real*z.real + z.imag*z.imag);
|
|
#else
|
|
return hypot(z.real, z.imag);
|
|
#endif
|
|
}
|
|
static CYTHON_INLINE __pyx_t_double_complex __Pyx_c_pow(__pyx_t_double_complex a, __pyx_t_double_complex b) {
|
|
__pyx_t_double_complex z;
|
|
double r, lnr, theta, z_r, z_theta;
|
|
if (b.imag == 0 && b.real == (int)b.real) {
|
|
if (b.real < 0) {
|
|
double denom = a.real * a.real + a.imag * a.imag;
|
|
a.real = a.real / denom;
|
|
a.imag = -a.imag / denom;
|
|
b.real = -b.real;
|
|
}
|
|
switch ((int)b.real) {
|
|
case 0:
|
|
z.real = 1;
|
|
z.imag = 0;
|
|
return z;
|
|
case 1:
|
|
return a;
|
|
case 2:
|
|
z = __Pyx_c_prod(a, a);
|
|
return __Pyx_c_prod(a, a);
|
|
case 3:
|
|
z = __Pyx_c_prod(a, a);
|
|
return __Pyx_c_prod(z, a);
|
|
case 4:
|
|
z = __Pyx_c_prod(a, a);
|
|
return __Pyx_c_prod(z, z);
|
|
}
|
|
}
|
|
if (a.imag == 0) {
|
|
if (a.real == 0) {
|
|
return a;
|
|
}
|
|
r = a.real;
|
|
theta = 0;
|
|
} else {
|
|
r = __Pyx_c_abs(a);
|
|
theta = atan2(a.imag, a.real);
|
|
}
|
|
lnr = log(r);
|
|
z_r = exp(lnr * b.real - theta * b.imag);
|
|
z_theta = theta * b.real + lnr * b.imag;
|
|
z.real = z_r * cos(z_theta);
|
|
z.imag = z_r * sin(z_theta);
|
|
return z;
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
#include "longintrepr.h"
|
|
#endif
|
|
#endif
|
|
static CYTHON_INLINE long __Pyx_PyInt_As_long(PyObject *x) {
|
|
const long neg_one = (long) -1, const_zero = 0;
|
|
const int is_unsigned = neg_one > const_zero;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyInt_Check(x))) {
|
|
if (sizeof(long) < sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(long, long, PyInt_AS_LONG)
|
|
} else {
|
|
long val = PyInt_AS_LONG(x);
|
|
if (is_unsigned && unlikely(val < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to long");
|
|
return (long) -1;
|
|
}
|
|
return (long) val;
|
|
}
|
|
} else
|
|
#endif
|
|
if (likely(PyLong_Check(x))) {
|
|
if (is_unsigned) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(long)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return (long) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (unlikely(Py_SIZE(x) < 0)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"can't convert negative value to long");
|
|
return (long) -1;
|
|
}
|
|
if (sizeof(long) <= sizeof(unsigned long)) {
|
|
__PYX_VERIFY_RETURN_INT(long, unsigned long, PyLong_AsUnsignedLong)
|
|
} else if (sizeof(long) <= sizeof(unsigned long long)) {
|
|
__PYX_VERIFY_RETURN_INT(long, unsigned long long, PyLong_AsUnsignedLongLong)
|
|
}
|
|
} else {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
if (sizeof(digit) <= sizeof(long)) {
|
|
switch (Py_SIZE(x)) {
|
|
case 0: return 0;
|
|
case 1: return +(long) ((PyLongObject*)x)->ob_digit[0];
|
|
case -1: return -(long) ((PyLongObject*)x)->ob_digit[0];
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|
|
if (sizeof(long) <= sizeof(long)) {
|
|
__PYX_VERIFY_RETURN_INT(long, long, PyLong_AsLong)
|
|
} else if (sizeof(long) <= sizeof(long long)) {
|
|
__PYX_VERIFY_RETURN_INT(long, long long, PyLong_AsLongLong)
|
|
}
|
|
}
|
|
{
|
|
#if CYTHON_COMPILING_IN_PYPY && !defined(_PyLong_AsByteArray)
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"_PyLong_AsByteArray() not available in PyPy, cannot convert large numbers");
|
|
#else
|
|
long val;
|
|
PyObject *v = __Pyx_PyNumber_Int(x);
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(v) && !PyLong_Check(v)) {
|
|
PyObject *tmp = v;
|
|
v = PyNumber_Long(tmp);
|
|
Py_DECREF(tmp);
|
|
}
|
|
#endif
|
|
if (likely(v)) {
|
|
int one = 1; int is_little = (int)*(unsigned char *)&one;
|
|
unsigned char *bytes = (unsigned char *)&val;
|
|
int ret = _PyLong_AsByteArray((PyLongObject *)v,
|
|
bytes, sizeof(val),
|
|
is_little, !is_unsigned);
|
|
Py_DECREF(v);
|
|
if (likely(!ret))
|
|
return val;
|
|
}
|
|
#endif
|
|
return (long) -1;
|
|
}
|
|
} else {
|
|
long val;
|
|
PyObject *tmp = __Pyx_PyNumber_Int(x);
|
|
if (!tmp) return (long) -1;
|
|
val = __Pyx_PyInt_As_long(tmp);
|
|
Py_DECREF(tmp);
|
|
return val;
|
|
}
|
|
}
|
|
|
|
static int __Pyx_check_binary_version(void) {
|
|
char ctversion[4], rtversion[4];
|
|
PyOS_snprintf(ctversion, 4, "%d.%d", PY_MAJOR_VERSION, PY_MINOR_VERSION);
|
|
PyOS_snprintf(rtversion, 4, "%s", Py_GetVersion());
|
|
if (ctversion[0] != rtversion[0] || ctversion[2] != rtversion[2]) {
|
|
char message[200];
|
|
PyOS_snprintf(message, sizeof(message),
|
|
"compiletime version %s of module '%.100s' "
|
|
"does not match runtime version %s",
|
|
ctversion, __Pyx_MODULE_NAME, rtversion);
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
return PyErr_Warn(NULL, message);
|
|
#else
|
|
return PyErr_WarnEx(NULL, message, 1);
|
|
#endif
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#ifndef __PYX_HAVE_RT_ImportModule
|
|
#define __PYX_HAVE_RT_ImportModule
|
|
static PyObject *__Pyx_ImportModule(const char *name) {
|
|
PyObject *py_name = 0;
|
|
PyObject *py_module = 0;
|
|
py_name = __Pyx_PyIdentifier_FromString(name);
|
|
if (!py_name)
|
|
goto bad;
|
|
py_module = PyImport_Import(py_name);
|
|
Py_DECREF(py_name);
|
|
return py_module;
|
|
bad:
|
|
Py_XDECREF(py_name);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
#ifndef __PYX_HAVE_RT_ImportType
|
|
#define __PYX_HAVE_RT_ImportType
|
|
static PyTypeObject *__Pyx_ImportType(const char *module_name, const char *class_name,
|
|
size_t size, int strict)
|
|
{
|
|
PyObject *py_module = 0;
|
|
PyObject *result = 0;
|
|
PyObject *py_name = 0;
|
|
char warning[200];
|
|
Py_ssize_t basicsize;
|
|
#ifdef Py_LIMITED_API
|
|
PyObject *py_basicsize;
|
|
#endif
|
|
py_module = __Pyx_ImportModule(module_name);
|
|
if (!py_module)
|
|
goto bad;
|
|
py_name = __Pyx_PyIdentifier_FromString(class_name);
|
|
if (!py_name)
|
|
goto bad;
|
|
result = PyObject_GetAttr(py_module, py_name);
|
|
Py_DECREF(py_name);
|
|
py_name = 0;
|
|
Py_DECREF(py_module);
|
|
py_module = 0;
|
|
if (!result)
|
|
goto bad;
|
|
if (!PyType_Check(result)) {
|
|
PyErr_Format(PyExc_TypeError,
|
|
"%.200s.%.200s is not a type object",
|
|
module_name, class_name);
|
|
goto bad;
|
|
}
|
|
#ifndef Py_LIMITED_API
|
|
basicsize = ((PyTypeObject *)result)->tp_basicsize;
|
|
#else
|
|
py_basicsize = PyObject_GetAttrString(result, "__basicsize__");
|
|
if (!py_basicsize)
|
|
goto bad;
|
|
basicsize = PyLong_AsSsize_t(py_basicsize);
|
|
Py_DECREF(py_basicsize);
|
|
py_basicsize = 0;
|
|
if (basicsize == (Py_ssize_t)-1 && PyErr_Occurred())
|
|
goto bad;
|
|
#endif
|
|
if (!strict && (size_t)basicsize > size) {
|
|
PyOS_snprintf(warning, sizeof(warning),
|
|
"%s.%s size changed, may indicate binary incompatibility",
|
|
module_name, class_name);
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (PyErr_Warn(NULL, warning) < 0) goto bad;
|
|
#else
|
|
if (PyErr_WarnEx(NULL, warning, 0) < 0) goto bad;
|
|
#endif
|
|
}
|
|
else if ((size_t)basicsize != size) {
|
|
PyErr_Format(PyExc_ValueError,
|
|
"%.200s.%.200s has the wrong size, try recompiling",
|
|
module_name, class_name);
|
|
goto bad;
|
|
}
|
|
return (PyTypeObject *)result;
|
|
bad:
|
|
Py_XDECREF(py_module);
|
|
Py_XDECREF(result);
|
|
return NULL;
|
|
}
|
|
#endif
|
|
|
|
static int __pyx_bisect_code_objects(__Pyx_CodeObjectCacheEntry* entries, int count, int code_line) {
|
|
int start = 0, mid = 0, end = count - 1;
|
|
if (end >= 0 && code_line > entries[end].code_line) {
|
|
return count;
|
|
}
|
|
while (start < end) {
|
|
mid = (start + end) / 2;
|
|
if (code_line < entries[mid].code_line) {
|
|
end = mid;
|
|
} else if (code_line > entries[mid].code_line) {
|
|
start = mid + 1;
|
|
} else {
|
|
return mid;
|
|
}
|
|
}
|
|
if (code_line <= entries[mid].code_line) {
|
|
return mid;
|
|
} else {
|
|
return mid + 1;
|
|
}
|
|
}
|
|
static PyCodeObject *__pyx_find_code_object(int code_line) {
|
|
PyCodeObject* code_object;
|
|
int pos;
|
|
if (unlikely(!code_line) || unlikely(!__pyx_code_cache.entries)) {
|
|
return NULL;
|
|
}
|
|
pos = __pyx_bisect_code_objects(__pyx_code_cache.entries, __pyx_code_cache.count, code_line);
|
|
if (unlikely(pos >= __pyx_code_cache.count) || unlikely(__pyx_code_cache.entries[pos].code_line != code_line)) {
|
|
return NULL;
|
|
}
|
|
code_object = __pyx_code_cache.entries[pos].code_object;
|
|
Py_INCREF(code_object);
|
|
return code_object;
|
|
}
|
|
static void __pyx_insert_code_object(int code_line, PyCodeObject* code_object) {
|
|
int pos, i;
|
|
__Pyx_CodeObjectCacheEntry* entries = __pyx_code_cache.entries;
|
|
if (unlikely(!code_line)) {
|
|
return;
|
|
}
|
|
if (unlikely(!entries)) {
|
|
entries = (__Pyx_CodeObjectCacheEntry*)PyMem_Malloc(64*sizeof(__Pyx_CodeObjectCacheEntry));
|
|
if (likely(entries)) {
|
|
__pyx_code_cache.entries = entries;
|
|
__pyx_code_cache.max_count = 64;
|
|
__pyx_code_cache.count = 1;
|
|
entries[0].code_line = code_line;
|
|
entries[0].code_object = code_object;
|
|
Py_INCREF(code_object);
|
|
}
|
|
return;
|
|
}
|
|
pos = __pyx_bisect_code_objects(__pyx_code_cache.entries, __pyx_code_cache.count, code_line);
|
|
if ((pos < __pyx_code_cache.count) && unlikely(__pyx_code_cache.entries[pos].code_line == code_line)) {
|
|
PyCodeObject* tmp = entries[pos].code_object;
|
|
entries[pos].code_object = code_object;
|
|
Py_DECREF(tmp);
|
|
return;
|
|
}
|
|
if (__pyx_code_cache.count == __pyx_code_cache.max_count) {
|
|
int new_max = __pyx_code_cache.max_count + 64;
|
|
entries = (__Pyx_CodeObjectCacheEntry*)PyMem_Realloc(
|
|
__pyx_code_cache.entries, (size_t)new_max*sizeof(__Pyx_CodeObjectCacheEntry));
|
|
if (unlikely(!entries)) {
|
|
return;
|
|
}
|
|
__pyx_code_cache.entries = entries;
|
|
__pyx_code_cache.max_count = new_max;
|
|
}
|
|
for (i=__pyx_code_cache.count; i>pos; i--) {
|
|
entries[i] = entries[i-1];
|
|
}
|
|
entries[pos].code_line = code_line;
|
|
entries[pos].code_object = code_object;
|
|
__pyx_code_cache.count++;
|
|
Py_INCREF(code_object);
|
|
}
|
|
|
|
#include "compile.h"
|
|
#include "frameobject.h"
|
|
#include "traceback.h"
|
|
static PyCodeObject* __Pyx_CreateCodeObjectForTraceback(
|
|
const char *funcname, int c_line,
|
|
int py_line, const char *filename) {
|
|
PyCodeObject *py_code = 0;
|
|
PyObject *py_srcfile = 0;
|
|
PyObject *py_funcname = 0;
|
|
#if PY_MAJOR_VERSION < 3
|
|
py_srcfile = PyString_FromString(filename);
|
|
#else
|
|
py_srcfile = PyUnicode_FromString(filename);
|
|
#endif
|
|
if (!py_srcfile) goto bad;
|
|
if (c_line) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
py_funcname = PyString_FromFormat( "%s (%s:%d)", funcname, __pyx_cfilenm, c_line);
|
|
#else
|
|
py_funcname = PyUnicode_FromFormat( "%s (%s:%d)", funcname, __pyx_cfilenm, c_line);
|
|
#endif
|
|
}
|
|
else {
|
|
#if PY_MAJOR_VERSION < 3
|
|
py_funcname = PyString_FromString(funcname);
|
|
#else
|
|
py_funcname = PyUnicode_FromString(funcname);
|
|
#endif
|
|
}
|
|
if (!py_funcname) goto bad;
|
|
py_code = __Pyx_PyCode_New(
|
|
0, /*int argcount,*/
|
|
0, /*int kwonlyargcount,*/
|
|
0, /*int nlocals,*/
|
|
0, /*int stacksize,*/
|
|
0, /*int flags,*/
|
|
__pyx_empty_bytes, /*PyObject *code,*/
|
|
__pyx_empty_tuple, /*PyObject *consts,*/
|
|
__pyx_empty_tuple, /*PyObject *names,*/
|
|
__pyx_empty_tuple, /*PyObject *varnames,*/
|
|
__pyx_empty_tuple, /*PyObject *freevars,*/
|
|
__pyx_empty_tuple, /*PyObject *cellvars,*/
|
|
py_srcfile, /*PyObject *filename,*/
|
|
py_funcname, /*PyObject *name,*/
|
|
py_line, /*int firstlineno,*/
|
|
__pyx_empty_bytes /*PyObject *lnotab*/
|
|
);
|
|
Py_DECREF(py_srcfile);
|
|
Py_DECREF(py_funcname);
|
|
return py_code;
|
|
bad:
|
|
Py_XDECREF(py_srcfile);
|
|
Py_XDECREF(py_funcname);
|
|
return NULL;
|
|
}
|
|
static void __Pyx_AddTraceback(const char *funcname, int c_line,
|
|
int py_line, const char *filename) {
|
|
PyCodeObject *py_code = 0;
|
|
PyObject *py_globals = 0;
|
|
PyFrameObject *py_frame = 0;
|
|
py_code = __pyx_find_code_object(c_line ? c_line : py_line);
|
|
if (!py_code) {
|
|
py_code = __Pyx_CreateCodeObjectForTraceback(
|
|
funcname, c_line, py_line, filename);
|
|
if (!py_code) goto bad;
|
|
__pyx_insert_code_object(c_line ? c_line : py_line, py_code);
|
|
}
|
|
py_globals = PyModule_GetDict(__pyx_m);
|
|
if (!py_globals) goto bad;
|
|
py_frame = PyFrame_New(
|
|
PyThreadState_GET(), /*PyThreadState *tstate,*/
|
|
py_code, /*PyCodeObject *code,*/
|
|
py_globals, /*PyObject *globals,*/
|
|
0 /*PyObject *locals*/
|
|
);
|
|
if (!py_frame) goto bad;
|
|
py_frame->f_lineno = py_line;
|
|
PyTraceBack_Here(py_frame);
|
|
bad:
|
|
Py_XDECREF(py_code);
|
|
Py_XDECREF(py_frame);
|
|
}
|
|
|
|
static int __Pyx_InitStrings(__Pyx_StringTabEntry *t) {
|
|
while (t->p) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (t->is_unicode) {
|
|
*t->p = PyUnicode_DecodeUTF8(t->s, t->n - 1, NULL);
|
|
} else if (t->intern) {
|
|
*t->p = PyString_InternFromString(t->s);
|
|
} else {
|
|
*t->p = PyString_FromStringAndSize(t->s, t->n - 1);
|
|
}
|
|
#else /* Python 3+ has unicode identifiers */
|
|
if (t->is_unicode | t->is_str) {
|
|
if (t->intern) {
|
|
*t->p = PyUnicode_InternFromString(t->s);
|
|
} else if (t->encoding) {
|
|
*t->p = PyUnicode_Decode(t->s, t->n - 1, t->encoding, NULL);
|
|
} else {
|
|
*t->p = PyUnicode_FromStringAndSize(t->s, t->n - 1);
|
|
}
|
|
} else {
|
|
*t->p = PyBytes_FromStringAndSize(t->s, t->n - 1);
|
|
}
|
|
#endif
|
|
if (!*t->p)
|
|
return -1;
|
|
++t;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static CYTHON_INLINE PyObject* __Pyx_PyUnicode_FromString(const char* c_str) {
|
|
return __Pyx_PyUnicode_FromStringAndSize(c_str, (Py_ssize_t)strlen(c_str));
|
|
}
|
|
static CYTHON_INLINE char* __Pyx_PyObject_AsString(PyObject* o) {
|
|
Py_ssize_t ignore;
|
|
return __Pyx_PyObject_AsStringAndSize(o, &ignore);
|
|
}
|
|
static CYTHON_INLINE char* __Pyx_PyObject_AsStringAndSize(PyObject* o, Py_ssize_t *length) {
|
|
#if __PYX_DEFAULT_STRING_ENCODING_IS_ASCII || __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT
|
|
if (
|
|
#if PY_MAJOR_VERSION < 3 && __PYX_DEFAULT_STRING_ENCODING_IS_ASCII
|
|
__Pyx_sys_getdefaultencoding_not_ascii &&
|
|
#endif
|
|
PyUnicode_Check(o)) {
|
|
#if PY_VERSION_HEX < 0x03030000
|
|
char* defenc_c;
|
|
PyObject* defenc = _PyUnicode_AsDefaultEncodedString(o, NULL);
|
|
if (!defenc) return NULL;
|
|
defenc_c = PyBytes_AS_STRING(defenc);
|
|
#if __PYX_DEFAULT_STRING_ENCODING_IS_ASCII
|
|
{
|
|
char* end = defenc_c + PyBytes_GET_SIZE(defenc);
|
|
char* c;
|
|
for (c = defenc_c; c < end; c++) {
|
|
if ((unsigned char) (*c) >= 128) {
|
|
PyUnicode_AsASCIIString(o);
|
|
return NULL;
|
|
}
|
|
}
|
|
}
|
|
#endif /*__PYX_DEFAULT_STRING_ENCODING_IS_ASCII*/
|
|
*length = PyBytes_GET_SIZE(defenc);
|
|
return defenc_c;
|
|
#else /* PY_VERSION_HEX < 0x03030000 */
|
|
if (PyUnicode_READY(o) == -1) return NULL;
|
|
#if __PYX_DEFAULT_STRING_ENCODING_IS_ASCII
|
|
if (PyUnicode_IS_ASCII(o)) {
|
|
*length = PyUnicode_GET_LENGTH(o);
|
|
return PyUnicode_AsUTF8(o);
|
|
} else {
|
|
PyUnicode_AsASCIIString(o);
|
|
return NULL;
|
|
}
|
|
#else /* __PYX_DEFAULT_STRING_ENCODING_IS_ASCII */
|
|
return PyUnicode_AsUTF8AndSize(o, length);
|
|
#endif /* __PYX_DEFAULT_STRING_ENCODING_IS_ASCII */
|
|
#endif /* PY_VERSION_HEX < 0x03030000 */
|
|
} else
|
|
#endif /* __PYX_DEFAULT_STRING_ENCODING_IS_ASCII || __PYX_DEFAULT_STRING_ENCODING_IS_DEFAULT */
|
|
#if !CYTHON_COMPILING_IN_PYPY
|
|
#if PY_VERSION_HEX >= 0x02060000
|
|
if (PyByteArray_Check(o)) {
|
|
*length = PyByteArray_GET_SIZE(o);
|
|
return PyByteArray_AS_STRING(o);
|
|
} else
|
|
#endif
|
|
#endif
|
|
{
|
|
char* result;
|
|
int r = PyBytes_AsStringAndSize(o, &result, length);
|
|
if (unlikely(r < 0)) {
|
|
return NULL;
|
|
} else {
|
|
return result;
|
|
}
|
|
}
|
|
}
|
|
static CYTHON_INLINE int __Pyx_PyObject_IsTrue(PyObject* x) {
|
|
int is_true = x == Py_True;
|
|
if (is_true | (x == Py_False) | (x == Py_None)) return is_true;
|
|
else return PyObject_IsTrue(x);
|
|
}
|
|
static CYTHON_INLINE PyObject* __Pyx_PyNumber_Int(PyObject* x) {
|
|
PyNumberMethods *m;
|
|
const char *name = NULL;
|
|
PyObject *res = NULL;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (PyInt_Check(x) || PyLong_Check(x))
|
|
#else
|
|
if (PyLong_Check(x))
|
|
#endif
|
|
return Py_INCREF(x), x;
|
|
m = Py_TYPE(x)->tp_as_number;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (m && m->nb_int) {
|
|
name = "int";
|
|
res = PyNumber_Int(x);
|
|
}
|
|
else if (m && m->nb_long) {
|
|
name = "long";
|
|
res = PyNumber_Long(x);
|
|
}
|
|
#else
|
|
if (m && m->nb_int) {
|
|
name = "int";
|
|
res = PyNumber_Long(x);
|
|
}
|
|
#endif
|
|
if (res) {
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (!PyInt_Check(res) && !PyLong_Check(res)) {
|
|
#else
|
|
if (!PyLong_Check(res)) {
|
|
#endif
|
|
PyErr_Format(PyExc_TypeError,
|
|
"__%.4s__ returned non-%.4s (type %.200s)",
|
|
name, name, Py_TYPE(res)->tp_name);
|
|
Py_DECREF(res);
|
|
return NULL;
|
|
}
|
|
}
|
|
else if (!PyErr_Occurred()) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"an integer is required");
|
|
}
|
|
return res;
|
|
}
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
#include "longintrepr.h"
|
|
#endif
|
|
#endif
|
|
static CYTHON_INLINE Py_ssize_t __Pyx_PyIndex_AsSsize_t(PyObject* b) {
|
|
Py_ssize_t ival;
|
|
PyObject *x;
|
|
#if PY_MAJOR_VERSION < 3
|
|
if (likely(PyInt_CheckExact(b)))
|
|
return PyInt_AS_LONG(b);
|
|
#endif
|
|
if (likely(PyLong_CheckExact(b))) {
|
|
#if CYTHON_COMPILING_IN_CPYTHON && PY_MAJOR_VERSION >= 3
|
|
#if CYTHON_USE_PYLONG_INTERNALS
|
|
switch (Py_SIZE(b)) {
|
|
case -1: return -(sdigit)((PyLongObject*)b)->ob_digit[0];
|
|
case 0: return 0;
|
|
case 1: return ((PyLongObject*)b)->ob_digit[0];
|
|
}
|
|
#endif
|
|
#endif
|
|
#if PY_VERSION_HEX < 0x02060000
|
|
return PyInt_AsSsize_t(b);
|
|
#else
|
|
return PyLong_AsSsize_t(b);
|
|
#endif
|
|
}
|
|
x = PyNumber_Index(b);
|
|
if (!x) return -1;
|
|
ival = PyInt_AsSsize_t(x);
|
|
Py_DECREF(x);
|
|
return ival;
|
|
}
|
|
static CYTHON_INLINE PyObject * __Pyx_PyInt_FromSize_t(size_t ival) {
|
|
#if PY_VERSION_HEX < 0x02050000
|
|
if (ival <= LONG_MAX)
|
|
return PyInt_FromLong((long)ival);
|
|
else {
|
|
unsigned char *bytes = (unsigned char *) &ival;
|
|
int one = 1; int little = (int)*(unsigned char*)&one;
|
|
return _PyLong_FromByteArray(bytes, sizeof(size_t), little, 0);
|
|
}
|
|
#else
|
|
return PyInt_FromSize_t(ival);
|
|
#endif
|
|
}
|
|
|
|
|
|
#endif /* Py_PYTHON_H */
|