baohan
/
scikit-learn
mirror of https://github.com/scikit-learn/scikit-learn.git
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
scikit-learn/sklearn/tree/_tree.c

12163 lines
493 KiB
C
Raw Blame History

/* Generated by Cython 0.15.1 on Tue Mar 20 14:57:24 2012 */
#define PY_SSIZE_T_CLEAN
#include "Python.h"
#ifndef Py_PYTHON_H
#error Python headers needed to compile C extensions, please install development version of Python.
#else
#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
#if PY_VERSION_HEX < 0x02040000
#define METH_COEXIST 0
#define PyDict_CheckExact(op) (Py_TYPE(op) == &PyDict_Type)
#define PyDict_Contains(d,o) PySequence_Contains(d,o)
#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 PyInt_FromSsize_t(z) PyInt_FromLong(z)
#define PyInt_AsSsize_t(o) __Pyx_PyInt_AsInt(o)
#define PyNumber_Index(o) PyNumber_Int(o)
#define PyIndex_Check(o) PyNumber_Check(o)
#define PyErr_WarnEx(category, message, stacklevel) PyErr_Warn(category, message)
#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)
#endif
#if PY_MAJOR_VERSION < 3
#define __Pyx_BUILTIN_MODULE_NAME "__builtin__"
#else
#define __Pyx_BUILTIN_MODULE_NAME "builtins"
#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_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_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
#endif
#if PY_MAJOR_VERSION >= 3
#define PyBoolObject PyLongObject
#endif
#if PY_VERSION_HEX < 0x03020000
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
#define __Pyx_PyNumber_Divide(x,y) PyNumber_TrueDivide(x,y)
#define __Pyx_PyNumber_InPlaceDivide(x,y) PyNumber_InPlaceTrueDivide(x,y)
#else
#define __Pyx_PyNumber_Divide(x,y) PyNumber_Divide(x,y)
#define __Pyx_PyNumber_InPlaceDivide(x,y) PyNumber_InPlaceDivide(x,y)
#endif
#if (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 __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 "stdio.h"
#include "stdlib.h"
#include "numpy/arrayobject.h"
#include "numpy/ufuncobject.h"
#include "math.h"
#include "float.h"
#ifdef _OPENMP
#include <omp.h>
#endif /* _OPENMP */
#ifdef PYREX_WITHOUT_ASSERTIONS
#define CYTHON_WITHOUT_ASSERTIONS
#endif
/* inline attribute */
#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
/* unused attribute */
#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)
# define CYTHON_UNUSED __attribute__ ((__unused__))
# else
# define CYTHON_UNUSED
# endif
#endif
typedef struct {PyObject **p; char *s; const long n; const char* encoding; const char is_unicode; const char is_str; const char intern; } __Pyx_StringTabEntry; /*proto*/
/* Type Conversion Predeclarations */
#define __Pyx_PyBytes_FromUString(s) PyBytes_FromString((char*)s)
#define __Pyx_PyBytes_AsUString(s) ((unsigned char*) PyBytes_AsString(s))
#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);
static CYTHON_INLINE size_t __Pyx_PyInt_AsSize_t(PyObject*);
#define __pyx_PyFloat_AsDouble(x) (PyFloat_CheckExact(x) ? PyFloat_AS_DOUBLE(x) : PyFloat_AsDouble(x))
#ifdef __GNUC__
/* Test for GCC > 2.95 */
#if __GNUC__ > 2 || (__GNUC__ == 2 && (__GNUC_MINOR__ > 95))
#define likely(x) __builtin_expect(!!(x), 1)
#define unlikely(x) __builtin_expect(!!(x), 0)
#else /* __GNUC__ > 2 ... */
#define likely(x) (x)
#define unlikely(x) (x)
#endif /* __GNUC__ > 2 ... */
#else /* __GNUC__ */
#define likely(x) (x)
#define unlikely(x) (x)
#endif /* __GNUC__ */
static PyObject *__pyx_m;
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",
"numpy.pxd",
};
/* "numpy.pxd":719
* # 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;
/* "numpy.pxd":720
*
* 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;
/* "numpy.pxd":721
* 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;
/* "numpy.pxd":722
* 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;
/* "numpy.pxd":726
* #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;
/* "numpy.pxd":727
*
* 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;
/* "numpy.pxd":728
* 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;
/* "numpy.pxd":729
* 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;
/* "numpy.pxd":733
* #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;
/* "numpy.pxd":734
*
* 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;
/* "numpy.pxd":743
* # 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;
/* "numpy.pxd":744
* # 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;
/* "numpy.pxd":745
* 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;
/* "numpy.pxd":747
* 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;
/* "numpy.pxd":748
*
* ctypedef npy_ulong uint_t
* ctypedef npy_ulonglong ulong_t # <<<<<<<<<<<<<<
* ctypedef npy_ulonglong ulonglong_t
*
*/
typedef npy_ulonglong __pyx_t_5numpy_ulong_t;
/* "numpy.pxd":749
* 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;
/* "numpy.pxd":751
* ctypedef npy_ulonglong ulonglong_t
*
* ctypedef npy_intp intp_t # <<<<<<<<<<<<<<
* ctypedef npy_uintp uintp_t
*
*/
typedef npy_intp __pyx_t_5numpy_intp_t;
/* "numpy.pxd":752
*
* ctypedef npy_intp intp_t
* ctypedef npy_uintp uintp_t # <<<<<<<<<<<<<<
*
* ctypedef npy_double float_t
*/
typedef npy_uintp __pyx_t_5numpy_uintp_t;
/* "numpy.pxd":754
* 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;
/* "numpy.pxd":755
*
* ctypedef npy_double float_t
* ctypedef npy_double double_t # <<<<<<<<<<<<<<
* ctypedef npy_longdouble longdouble_t
*
*/
typedef npy_double __pyx_t_5numpy_double_t;
/* "numpy.pxd":756
* 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/_tree.pyx":17
* # Define a datatype for the data array
* DTYPE = np.float32
* ctypedef np.float32_t DTYPE_t # <<<<<<<<<<<<<<
* ctypedef np.int8_t BOOL_t
*
*/
typedef __pyx_t_5numpy_float32_t __pyx_t_7sklearn_4tree_5_tree_DTYPE_t;
/* "sklearn/tree/_tree.pyx":18
* DTYPE = np.float32
* ctypedef np.float32_t DTYPE_t
* ctypedef np.int8_t BOOL_t # <<<<<<<<<<<<<<
*
* cdef extern from "math.h":
*/
typedef __pyx_t_5numpy_int8_t __pyx_t_7sklearn_4tree_5_tree_BOOL_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_5_tree_Criterion;
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion;
struct __pyx_obj_7sklearn_4tree_5_tree_Gini;
struct __pyx_obj_7sklearn_4tree_5_tree_Entropy;
struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion;
struct __pyx_obj_7sklearn_4tree_5_tree_MSE;
/* "numpy.pxd":758
* 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;
/* "numpy.pxd":759
*
* ctypedef npy_cfloat cfloat_t
* ctypedef npy_cdouble cdouble_t # <<<<<<<<<<<<<<
* ctypedef npy_clongdouble clongdouble_t
*
*/
typedef npy_cdouble __pyx_t_5numpy_cdouble_t;
/* "numpy.pxd":760
* 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;
/* "numpy.pxd":762
* ctypedef npy_clongdouble clongdouble_t
*
* ctypedef npy_cdouble complex_t # <<<<<<<<<<<<<<
*
* cdef inline object PyArray_MultiIterNew1(a):
*/
typedef npy_cdouble __pyx_t_5numpy_complex_t;
/* "sklearn/tree/_tree.pyx":40
* # be the proportion of class k observations in node m
*
* cdef class Criterion: # <<<<<<<<<<<<<<
* """Interface for splitting criteria (regression and classification)"""
*
*/
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion {
PyObject_HEAD
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *__pyx_vtab;
};
/* "sklearn/tree/_tree.pyx":67
*
*
* cdef class ClassificationCriterion(Criterion): # <<<<<<<<<<<<<<
* """Abstract criterion for classification.
*
*/
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion {
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion __pyx_base;
int n_classes;
int n_samples;
int *label_count_left;
int *label_count_right;
int *label_count_init;
int n_left;
int n_right;
PyObject *ndarray_label_count_left;
PyObject *ndarray_label_count_right;
PyObject *ndarray_label_count_init;
};
/* "sklearn/tree/_tree.pyx":181
*
*
* cdef class Gini(ClassificationCriterion): # <<<<<<<<<<<<<<
* """Gini Index splitting criteria.
*
*/
struct __pyx_obj_7sklearn_4tree_5_tree_Gini {
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_base;
};
/* "sklearn/tree/_tree.pyx":217
*
*
* cdef class Entropy(ClassificationCriterion): # <<<<<<<<<<<<<<
* """Entropy splitting criteria.
*
*/
struct __pyx_obj_7sklearn_4tree_5_tree_Entropy {
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion __pyx_base;
};
/* "sklearn/tree/_tree.pyx":245
*
*
* cdef class RegressionCriterion(Criterion): # <<<<<<<<<<<<<<
* """Abstract criterion for regression. Computes variance of the
* target values left and right of the split point.
*/
struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion {
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion __pyx_base;
int n_samples;
int n_right;
int n_left;
double mean_left;
double mean_right;
double mean_init;
double sq_sum_right;
double sq_sum_left;
double sq_sum_init;
double var_left;
double var_right;
};
/* "sklearn/tree/_tree.pyx":394
*
*
* 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":40
* # be the proportion of class k observations in node m
*
* cdef class Criterion: # <<<<<<<<<<<<<<
* """Interface for splitting criteria (regression and classification)"""
*
*/
struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion {
void (*init)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *, int, int);
void (*reset)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *);
int (*update)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int, int, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, int *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *);
double (*eval)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *);
PyArrayObject *(*init_value)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int __pyx_skip_dispatch);
};
static struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *__pyx_vtabptr_7sklearn_4tree_5_tree_Criterion;
/* "sklearn/tree/_tree.pyx":245
*
*
* cdef class RegressionCriterion(Criterion): # <<<<<<<<<<<<<<
* """Abstract criterion for regression. Computes variance of the
* target values left and right of the split point.
*/
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":394
*
*
* 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":67
*
*
* cdef class ClassificationCriterion(Criterion): # <<<<<<<<<<<<<<
* """Abstract criterion for classification.
*
*/
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":181
*
*
* cdef class Gini(ClassificationCriterion): # <<<<<<<<<<<<<<
* """Gini Index splitting 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":217
*
*
* cdef class Entropy(ClassificationCriterion): # <<<<<<<<<<<<<<
* """Entropy splitting 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;
#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;
#define __Pyx_RefNannySetupContext(name) __pyx_refnanny = __Pyx_RefNanny->SetupContext((name), __LINE__, __FILE__)
#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)
#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 */
static PyObject *__Pyx_GetName(PyObject *dict, PyObject *name); /*proto*/
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type); /*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 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*/
/* Run-time type information about structs used with buffers */
struct __Pyx_StructField_;
typedef struct {
const char* name; /* for error messages only */
struct __Pyx_StructField_* fields;
size_t size; /* sizeof(type) */
char typegroup; /* _R_eal, _C_omplex, Signed _I_nt, _U_nsigned int, _S_truct, _P_ointer, _O_bject */
} __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;
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);
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*/
static CYTHON_INLINE int __Pyx_CheckKeywordStrings(PyObject *kwdict,
const char* function_name, int kw_allowed); /*proto*/
#define __Pyx_BufPtrCContig1d(type, buf, i0, s0) ((type)buf + i0)
static int __Pyx_ArgTypeTest(PyObject *obj, PyTypeObject *type, int none_allowed,
const char *name, int exact); /*proto*/
#define __Pyx_BufPtrStrided2d(type, buf, i0, s0, i1, s1) (type)((char*)buf + i0 * s0 + i1 * s1)
#define __Pyx_BufPtrStrided1d(type, buf, i0, s0) (type)((char*)buf + i0 * s0)
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause); /*proto*/
static CYTHON_INLINE void __Pyx_RaiseNeedMoreValuesError(Py_ssize_t index);
static CYTHON_INLINE void __Pyx_RaiseTooManyValuesError(Py_ssize_t expected);
static CYTHON_INLINE void __Pyx_RaiseNoneNotIterableError(void);
static void __Pyx_UnpackTupleError(PyObject *, Py_ssize_t index); /*proto*/
#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
Py_ssize_t __Pyx_zeros[] = {0, 0};
Py_ssize_t __Pyx_minusones[] = {-1, -1};
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, long level); /*proto*/
#if CYTHON_CCOMPLEX
#ifdef __cplusplus
#define __Pyx_CREAL(z) ((z).real())
#define __Pyx_CIMAG(z) ((z).imag())
#else
#define __Pyx_CREAL(z) (__real__(z))
#define __Pyx_CIMAG(z) (__imag__(z))
#endif
#else
#define __Pyx_CREAL(z) ((z).real)
#define __Pyx_CIMAG(z) ((z).imag)
#endif
#if defined(_WIN32) && 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 unsigned char __Pyx_PyInt_AsUnsignedChar(PyObject *);
static CYTHON_INLINE unsigned short __Pyx_PyInt_AsUnsignedShort(PyObject *);
static CYTHON_INLINE unsigned int __Pyx_PyInt_AsUnsignedInt(PyObject *);
static CYTHON_INLINE char __Pyx_PyInt_AsChar(PyObject *);
static CYTHON_INLINE short __Pyx_PyInt_AsShort(PyObject *);
static CYTHON_INLINE int __Pyx_PyInt_AsInt(PyObject *);
static CYTHON_INLINE signed char __Pyx_PyInt_AsSignedChar(PyObject *);
static CYTHON_INLINE signed short __Pyx_PyInt_AsSignedShort(PyObject *);
static CYTHON_INLINE signed int __Pyx_PyInt_AsSignedInt(PyObject *);
static CYTHON_INLINE int __Pyx_PyInt_AsLongDouble(PyObject *);
static CYTHON_INLINE unsigned long __Pyx_PyInt_AsUnsignedLong(PyObject *);
static CYTHON_INLINE unsigned PY_LONG_LONG __Pyx_PyInt_AsUnsignedLongLong(PyObject *);
static CYTHON_INLINE long __Pyx_PyInt_AsLong(PyObject *);
static CYTHON_INLINE PY_LONG_LONG __Pyx_PyInt_AsLongLong(PyObject *);
static CYTHON_INLINE signed long __Pyx_PyInt_AsSignedLong(PyObject *);
static CYTHON_INLINE signed PY_LONG_LONG __Pyx_PyInt_AsSignedLongLong(PyObject *);
static int __Pyx_check_binary_version(void);
static int __Pyx_SetVtable(PyObject *dict, void *vtable); /*proto*/
static PyTypeObject *__Pyx_ImportType(const char *module_name, const char *class_name, size_t size, int strict); /*proto*/
static PyObject *__Pyx_ImportModule(const char *name); /*proto*/
static void __Pyx_AddTraceback(const char *funcname, int __pyx_clineno,
int __pyx_lineno, const char *__pyx_filename); /*proto*/
static int __Pyx_InitStrings(__Pyx_StringTabEntry *t); /*proto*/
/* Module declarations from 'cython.cython.view' */
/* Module declarations from 'cython' */
/* Module declarations from 'cpython.buffer' */
/* Module declarations from 'cpython.ref' */
/* Module declarations from 'libc.stdio' */
/* Module declarations from 'cpython.object' */
/* 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 PyObject *__pyx_f_5numpy_PyArray_MultiIterNew1(PyObject *); /*proto*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew2(PyObject *, PyObject *); /*proto*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew3(PyObject *, PyObject *, PyObject *); /*proto*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew4(PyObject *, PyObject *, PyObject *, PyObject *); /*proto*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_PyArray_MultiIterNew5(PyObject *, PyObject *, PyObject *, PyObject *, PyObject *); /*proto*/
static CYTHON_INLINE char *__pyx_f_5numpy__util_dtypestring(PyArray_Descr *, char *, char *, int *); /*proto*/
static CYTHON_INLINE void __pyx_f_5numpy_set_array_base(PyArrayObject *, PyObject *); /*proto*/
static CYTHON_INLINE PyObject *__pyx_f_5numpy_get_array_base(PyArrayObject *); /*proto*/
/* Module declarations from 'sklearn.tree._tree' */
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Criterion = 0;
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_ClassificationCriterion = 0;
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Gini = 0;
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_Entropy = 0;
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_RegressionCriterion = 0;
static PyTypeObject *__pyx_ptype_7sklearn_4tree_5_tree_MSE = 0;
static int __pyx_f_7sklearn_4tree_5_tree_smallest_sample_larger_than(int, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, int *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *, int); /*proto*/
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t = { "int32_t", NULL, sizeof(__pyx_t_5numpy_int32_t), 'I' };
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t = { "float64_t", NULL, sizeof(__pyx_t_5numpy_float64_t), 'R' };
static __Pyx_TypeInfo __Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_BOOL_t = { "BOOL_t", NULL, sizeof(__pyx_t_7sklearn_4tree_5_tree_BOOL_t), 'I' };
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), 'R' };
#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_xrange;
static PyObject *__pyx_builtin_ValueError;
static PyObject *__pyx_builtin_RuntimeError;
static char __pyx_k_1[] = "ndarray is not C contiguous";
static char __pyx_k_3[] = "ndarray is not Fortran contiguous";
static char __pyx_k_5[] = "Non-native byte order not supported";
static char __pyx_k_7[] = "unknown dtype code in numpy.pxd (%d)";
static char __pyx_k_8[] = "Format string allocated too short, see comment in numpy.pxd";
static char __pyx_k_11[] = "Format string allocated too short.";
static char __pyx_k_13[] = "sklearn.tree._tree";
static char __pyx_k_14[] = "_predict_regression_tree_inplace";
static char __pyx_k_15[] = "_find_best_random_split";
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__k[] = "k";
static char __pyx_k__l[] = "l";
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__out[] = "out";
static char __pyx_k__bool[] = "bool";
static char __pyx_k__int8[] = "int8";
static char __pyx_k__pred[] = "pred";
static char __pyx_k__rand[] = "rand";
static char __pyx_k__DTYPE[] = "DTYPE";
static char __pyx_k__dtype[] = "dtype";
static char __pyx_k__int32[] = "int32";
static char __pyx_k__numpy[] = "numpy";
static char __pyx_k__order[] = "order";
static char __pyx_k__range[] = "range";
static char __pyx_k__scale[] = "scale";
static char __pyx_k__zeros[] = "zeros";
static char __pyx_k__arange[] = "arange";
static char __pyx_k__astype[] = "astype";
static char __pyx_k__values[] = "values";
static char __pyx_k__xrange[] = "xrange";
static char __pyx_k__asarray[] = "asarray";
static char __pyx_k__feature[] = "feature";
static char __pyx_k__float32[] = "float32";
static char __pyx_k____main__[] = "__main__";
static char __pyx_k____test__[] = "__test__";
static char __pyx_k__children[] = "children";
static char __pyx_k__min_leaf[] = "min_leaf";
static char __pyx_k__criterion[] = "criterion";
static char __pyx_k__n_classes[] = "n_classes";
static char __pyx_k__n_samples[] = "n_samples";
static char __pyx_k__threshold[] = "threshold";
static char __pyx_k__ValueError[] = "ValueError";
static char __pyx_k__init_value[] = "init_value";
static char __pyx_k__X_argsorted[] = "X_argsorted";
static char __pyx_k___apply_tree[] = "_apply_tree";
static char __pyx_k__permutation[] = "permutation";
static char __pyx_k__sample_mask[] = "sample_mask";
static char __pyx_k__RuntimeError[] = "RuntimeError";
static char __pyx_k__max_features[] = "max_features";
static char __pyx_k__random_state[] = "random_state";
static char __pyx_k___predict_tree[] = "_predict_tree";
static char __pyx_k___error_at_leaf[] = "_error_at_leaf";
static char __pyx_k__n_total_in_bag[] = "n_total_in_bag";
static char __pyx_k__n_total_samples[] = "n_total_samples";
static char __pyx_k___find_best_split[] = "_find_best_split";
static char __pyx_k___random_sample_mask[] = "_random_sample_mask";
static PyObject *__pyx_kp_u_1;
static PyObject *__pyx_kp_u_11;
static PyObject *__pyx_n_s_13;
static PyObject *__pyx_n_s_14;
static PyObject *__pyx_n_s_15;
static PyObject *__pyx_kp_u_3;
static PyObject *__pyx_kp_u_5;
static PyObject *__pyx_kp_u_7;
static PyObject *__pyx_kp_u_8;
static PyObject *__pyx_n_s__C;
static PyObject *__pyx_n_s__DTYPE;
static PyObject *__pyx_n_s__RuntimeError;
static PyObject *__pyx_n_s__ValueError;
static PyObject *__pyx_n_s__X;
static PyObject *__pyx_n_s__X_argsorted;
static PyObject *__pyx_n_s____main__;
static PyObject *__pyx_n_s____test__;
static PyObject *__pyx_n_s___apply_tree;
static PyObject *__pyx_n_s___error_at_leaf;
static PyObject *__pyx_n_s___find_best_split;
static PyObject *__pyx_n_s___predict_tree;
static PyObject *__pyx_n_s___random_sample_mask;
static PyObject *__pyx_n_s__arange;
static PyObject *__pyx_n_s__asarray;
static PyObject *__pyx_n_s__astype;
static PyObject *__pyx_n_s__bool;
static PyObject *__pyx_n_s__children;
static PyObject *__pyx_n_s__criterion;
static PyObject *__pyx_n_s__dtype;
static PyObject *__pyx_n_s__feature;
static PyObject *__pyx_n_s__float32;
static PyObject *__pyx_n_s__inf;
static PyObject *__pyx_n_s__init_value;
static PyObject *__pyx_n_s__int32;
static PyObject *__pyx_n_s__int8;
static PyObject *__pyx_n_s__k;
static PyObject *__pyx_n_s__max_features;
static PyObject *__pyx_n_s__min_leaf;
static PyObject *__pyx_n_s__n_classes;
static PyObject *__pyx_n_s__n_samples;
static PyObject *__pyx_n_s__n_total_in_bag;
static PyObject *__pyx_n_s__n_total_samples;
static PyObject *__pyx_n_s__np;
static PyObject *__pyx_n_s__numpy;
static PyObject *__pyx_n_s__order;
static PyObject *__pyx_n_s__out;
static PyObject *__pyx_n_s__permutation;
static PyObject *__pyx_n_s__pred;
static PyObject *__pyx_n_s__rand;
static PyObject *__pyx_n_s__random_state;
static PyObject *__pyx_n_s__range;
static PyObject *__pyx_n_s__sample_mask;
static PyObject *__pyx_n_s__scale;
static PyObject *__pyx_n_s__threshold;
static PyObject *__pyx_n_s__values;
static PyObject *__pyx_n_s__xrange;
static PyObject *__pyx_n_s__y;
static PyObject *__pyx_n_s__zeros;
static PyObject *__pyx_int_1;
static PyObject *__pyx_int_15;
static PyObject *__pyx_k_tuple_2;
static PyObject *__pyx_k_tuple_4;
static PyObject *__pyx_k_tuple_6;
static PyObject *__pyx_k_tuple_9;
static PyObject *__pyx_k_tuple_10;
static PyObject *__pyx_k_tuple_12;
/* "sklearn/tree/_tree.pyx":43
* """Interface for splitting criteria (regression and classification)"""
*
* cdef void init(self, DTYPE_t *y, BOOL_t *sample_mask, int n_samples, # <<<<<<<<<<<<<<
* int n_total_samples):
* """Initialise the criterion class for new split point."""
*/
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_init(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask, int __pyx_v_n_samples, int __pyx_v_n_total_samples) {
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("init");
__Pyx_RefNannyFinishContext();
}
/* "sklearn/tree/_tree.pyx":48
* pass
*
* cdef void reset(self): # <<<<<<<<<<<<<<
* """Reset the criterion for a new feature index."""
* pass
*/
static void __pyx_f_7sklearn_4tree_5_tree_9Criterion_reset(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self) {
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("reset");
__Pyx_RefNannyFinishContext();
}
/* "sklearn/tree/_tree.pyx":52
* pass
*
* cdef int update(self, int a, int b, DTYPE_t *y, int *X_argsorted_i, # <<<<<<<<<<<<<<
* BOOL_t *sample_mask):
* """Update the criteria for each value in interval [a,b) (where a and b
*/
static int __pyx_f_7sklearn_4tree_5_tree_9Criterion_update(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, int __pyx_v_a, int __pyx_v_b, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y, int *__pyx_v_X_argsorted_i, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask) {
int __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("update");
__pyx_r = 0;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":58
* pass
*
* cdef double eval(self): # <<<<<<<<<<<<<<
* """Evaluate the criteria (aka the split error)."""
* pass
*/
static double __pyx_f_7sklearn_4tree_5_tree_9Criterion_eval(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self) {
double __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("eval");
__pyx_r = 0;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":62
* pass
*
* cpdef np.ndarray init_value(self): # <<<<<<<<<<<<<<
* """Get the init value of the criterion - `init` must be called before."""
* pass
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_9Criterion_init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_9Criterion_init_value(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_self, int __pyx_skip_dispatch) {
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("init_value");
/* Check if called by wrapper */
if (unlikely(__pyx_skip_dispatch)) ;
/* Check if overriden in Python */
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
__pyx_t_1 = PyObject_GetAttr(((PyObject *)__pyx_v_self), __pyx_n_s__init_value); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 62; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (void *)&__pyx_pf_7sklearn_4tree_5_tree_9Criterion_init_value)) {
__Pyx_XDECREF(((PyObject *)__pyx_r));
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 62; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 62; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_r = ((PyArrayObject *)__pyx_t_2);
__pyx_t_2 = 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":64
* cpdef np.ndarray init_value(self):
* """Get the init value of the criterion - `init` must be called before."""
* pass # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_2);
__Pyx_AddTraceback("sklearn.tree._tree.Criterion.init_value", __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":62
* pass
*
* cpdef np.ndarray init_value(self): # <<<<<<<<<<<<<<
* """Get the init value of the criterion - `init` must be called before."""
* pass
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_9Criterion_init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_9Criterion_init_value[] = "Get the init value of the criterion - `init` must be called before.";
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_9Criterion_init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
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("init_value");
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self)->__pyx_vtab)->init_value(((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self), 1)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 62; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_AddTraceback("sklearn.tree._tree.Criterion.init_value", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":107
* cdef ndarray_label_count_init
*
* def __init__(self, int n_classes): # <<<<<<<<<<<<<<
* cdef np.ndarray[np.int32_t, ndim=1] ndarray_label_count_left \
* = np.zeros((n_classes,), dtype=np.int32, order='C')
*/
static int __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion___init__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static int __pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion___init__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
int __pyx_v_n_classes;
PyArrayObject *__pyx_v_ndarray_label_count_left = 0;
PyArrayObject *__pyx_v_ndarray_label_count_right = 0;
PyArrayObject *__pyx_v_ndarray_label_count_init = 0;
Py_buffer __pyx_bstruct_ndarray_label_count_right;
Py_ssize_t __pyx_bstride_0_ndarray_label_count_right = 0;
Py_ssize_t __pyx_bshape_0_ndarray_label_count_right = 0;
Py_buffer __pyx_bstruct_ndarray_label_count_left;
Py_ssize_t __pyx_bstride_0_ndarray_label_count_left = 0;
Py_ssize_t __pyx_bshape_0_ndarray_label_count_left = 0;
Py_buffer __pyx_bstruct_ndarray_label_count_init;
Py_ssize_t __pyx_bstride_0_ndarray_label_count_init = 0;
Py_ssize_t __pyx_bshape_0_ndarray_label_count_init = 0;
int __pyx_r;
__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;
PyArrayObject *__pyx_t_6 = NULL;
PyArrayObject *__pyx_t_7 = NULL;
PyArrayObject *__pyx_t_8 = NULL;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__n_classes,0};
__Pyx_RefNannySetupContext("__init__");
{
PyObject* values[1] = {0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
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;
}
kw_args = PyDict_Size(__pyx_kwds);
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__n_classes);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "__init__") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 107; __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_classes = __Pyx_PyInt_AsInt(values[0]); if (unlikely((__pyx_v_n_classes == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 107; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("__init__", 1, 1, 1, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 107; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.__init__", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return -1;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_ndarray_label_count_left.buf = NULL;
__pyx_bstruct_ndarray_label_count_right.buf = NULL;
__pyx_bstruct_ndarray_label_count_init.buf = NULL;
/* "sklearn/tree/_tree.pyx":109
* def __init__(self, int n_classes):
* cdef np.ndarray[np.int32_t, ndim=1] ndarray_label_count_left \
* = np.zeros((n_classes,), dtype=np.int32, order='C') # <<<<<<<<<<<<<<
* cdef np.ndarray[np.int32_t, ndim=1] ndarray_label_count_right \
* = np.zeros((n_classes,), dtype=np.int32, order='C')
*/
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__zeros); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = PyInt_FromLong(__pyx_v_n_classes); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __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 = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
PyTuple_SET_ITEM(__pyx_t_3, 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 = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
PyTuple_SET_ITEM(__pyx_t_1, 0, ((PyObject *)__pyx_t_3));
__Pyx_GIVEREF(((PyObject *)__pyx_t_3));
__pyx_t_3 = 0;
__pyx_t_3 = PyDict_New(); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
__pyx_t_4 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__pyx_t_5 = PyObject_GetAttr(__pyx_t_4, __pyx_n_s__int32); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__dtype), __pyx_t_5) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (PyDict_SetItem(__pyx_t_3, ((PyObject *)__pyx_n_s__order), ((PyObject *)__pyx_n_s__C)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_5 = PyEval_CallObjectWithKeywords(__pyx_t_2, ((PyObject *)__pyx_t_1), ((PyObject *)__pyx_t_3)); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 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 = 109; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_6 = ((PyArrayObject *)__pyx_t_5);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_ndarray_label_count_left, (PyObject*)__pyx_t_6, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
__pyx_v_ndarray_label_count_left = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_bstruct_ndarray_label_count_left.buf = NULL;
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 108; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
} else {__pyx_bstride_0_ndarray_label_count_left = __pyx_bstruct_ndarray_label_count_left.strides[0];
__pyx_bshape_0_ndarray_label_count_left = __pyx_bstruct_ndarray_label_count_left.shape[0];
}
}
__pyx_t_6 = 0;
__pyx_v_ndarray_label_count_left = ((PyArrayObject *)__pyx_t_5);
__pyx_t_5 = 0;
/* "sklearn/tree/_tree.pyx":111
* = np.zeros((n_classes,), dtype=np.int32, order='C')
* cdef np.ndarray[np.int32_t, ndim=1] ndarray_label_count_right \
* = np.zeros((n_classes,), dtype=np.int32, order='C') # <<<<<<<<<<<<<<
* cdef np.ndarray[np.int32_t, ndim=1] ndarray_label_count_init \
* = np.zeros((n_classes,), dtype=np.int32, order='C')
*/
__pyx_t_5 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_GetAttr(__pyx_t_5, __pyx_n_s__zeros); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_t_5 = PyInt_FromLong(__pyx_v_n_classes); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
PyTuple_SET_ITEM(__pyx_t_1, 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 = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_5));
PyTuple_SET_ITEM(__pyx_t_5, 0, ((PyObject *)__pyx_t_1));
__Pyx_GIVEREF(((PyObject *)__pyx_t_1));
__pyx_t_1 = 0;
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_4 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__int32); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
if (PyDict_SetItem(__pyx_t_1, ((PyObject *)__pyx_n_s__dtype), __pyx_t_4) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
if (PyDict_SetItem(__pyx_t_1, ((PyObject *)__pyx_n_s__order), ((PyObject *)__pyx_n_s__C)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_4 = PyEval_CallObjectWithKeywords(__pyx_t_3, ((PyObject *)__pyx_t_5), ((PyObject *)__pyx_t_1)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_5)); __pyx_t_5 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
if (!(likely(((__pyx_t_4) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_4, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 111; __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_bstruct_ndarray_label_count_right, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
__pyx_v_ndarray_label_count_right = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_bstruct_ndarray_label_count_right.buf = NULL;
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 110; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
} else {__pyx_bstride_0_ndarray_label_count_right = __pyx_bstruct_ndarray_label_count_right.strides[0];
__pyx_bshape_0_ndarray_label_count_right = __pyx_bstruct_ndarray_label_count_right.shape[0];
}
}
__pyx_t_7 = 0;
__pyx_v_ndarray_label_count_right = ((PyArrayObject *)__pyx_t_4);
__pyx_t_4 = 0;
/* "sklearn/tree/_tree.pyx":113
* = np.zeros((n_classes,), dtype=np.int32, order='C')
* cdef np.ndarray[np.int32_t, ndim=1] ndarray_label_count_init \
* = np.zeros((n_classes,), dtype=np.int32, order='C') # <<<<<<<<<<<<<<
*
* self.n_classes = n_classes
*/
__pyx_t_4 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__pyx_t_1 = PyObject_GetAttr(__pyx_t_4, __pyx_n_s__zeros); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
__pyx_t_4 = PyInt_FromLong(__pyx_v_n_classes); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__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 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_4));
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_t_5));
__Pyx_GIVEREF(((PyObject *)__pyx_t_5));
__pyx_t_5 = 0;
__pyx_t_5 = PyDict_New(); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_5));
__pyx_t_3 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_3, __pyx_n_s__int32); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (PyDict_SetItem(__pyx_t_5, ((PyObject *)__pyx_n_s__dtype), __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
if (PyDict_SetItem(__pyx_t_5, ((PyObject *)__pyx_n_s__order), ((PyObject *)__pyx_n_s__C)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_2 = PyEval_CallObjectWithKeywords(__pyx_t_1, ((PyObject *)__pyx_t_4), ((PyObject *)__pyx_t_5)); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_4)); __pyx_t_4 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_5)); __pyx_t_5 = 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 = 113; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_8 = ((PyArrayObject *)__pyx_t_2);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_ndarray_label_count_init, (PyObject*)__pyx_t_8, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {
__pyx_v_ndarray_label_count_init = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_bstruct_ndarray_label_count_init.buf = NULL;
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 112; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
} else {__pyx_bstride_0_ndarray_label_count_init = __pyx_bstruct_ndarray_label_count_init.strides[0];
__pyx_bshape_0_ndarray_label_count_init = __pyx_bstruct_ndarray_label_count_init.shape[0];
}
}
__pyx_t_8 = 0;
__pyx_v_ndarray_label_count_init = ((PyArrayObject *)__pyx_t_2);
__pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":115
* = np.zeros((n_classes,), dtype=np.int32, order='C')
*
* self.n_classes = n_classes # <<<<<<<<<<<<<<
* self.n_samples = 0
* self.n_left = 0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->n_classes = __pyx_v_n_classes;
/* "sklearn/tree/_tree.pyx":116
*
* self.n_classes = n_classes
* self.n_samples = 0 # <<<<<<<<<<<<<<
* self.n_left = 0
* self.n_right = 0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->n_samples = 0;
/* "sklearn/tree/_tree.pyx":117
* self.n_classes = n_classes
* self.n_samples = 0
* self.n_left = 0 # <<<<<<<<<<<<<<
* self.n_right = 0
* self.label_count_left = <int*>ndarray_label_count_left.data
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->n_left = 0;
/* "sklearn/tree/_tree.pyx":118
* self.n_samples = 0
* self.n_left = 0
* self.n_right = 0 # <<<<<<<<<<<<<<
* self.label_count_left = <int*>ndarray_label_count_left.data
* self.label_count_right = <int*>ndarray_label_count_right.data
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->n_right = 0;
/* "sklearn/tree/_tree.pyx":119
* self.n_left = 0
* self.n_right = 0
* self.label_count_left = <int*>ndarray_label_count_left.data # <<<<<<<<<<<<<<
* self.label_count_right = <int*>ndarray_label_count_right.data
* self.label_count_init = <int*>ndarray_label_count_init.data
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->label_count_left = ((int *)__pyx_v_ndarray_label_count_left->data);
/* "sklearn/tree/_tree.pyx":120
* self.n_right = 0
* self.label_count_left = <int*>ndarray_label_count_left.data
* self.label_count_right = <int*>ndarray_label_count_right.data # <<<<<<<<<<<<<<
* self.label_count_init = <int*>ndarray_label_count_init.data
* self.ndarray_label_count_left = ndarray_label_count_left
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->label_count_right = ((int *)__pyx_v_ndarray_label_count_right->data);
/* "sklearn/tree/_tree.pyx":121
* self.label_count_left = <int*>ndarray_label_count_left.data
* self.label_count_right = <int*>ndarray_label_count_right.data
* self.label_count_init = <int*>ndarray_label_count_init.data # <<<<<<<<<<<<<<
* self.ndarray_label_count_left = ndarray_label_count_left
* self.ndarray_label_count_right = ndarray_label_count_right
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->label_count_init = ((int *)__pyx_v_ndarray_label_count_init->data);
/* "sklearn/tree/_tree.pyx":122
* self.label_count_right = <int*>ndarray_label_count_right.data
* self.label_count_init = <int*>ndarray_label_count_init.data
* self.ndarray_label_count_left = ndarray_label_count_left # <<<<<<<<<<<<<<
* self.ndarray_label_count_right = ndarray_label_count_right
* self.ndarray_label_count_init = ndarray_label_count_init
*/
__Pyx_INCREF(((PyObject *)__pyx_v_ndarray_label_count_left));
__Pyx_GIVEREF(((PyObject *)__pyx_v_ndarray_label_count_left));
__Pyx_GOTREF(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_left);
__Pyx_DECREF(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_left);
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_left = ((PyObject *)__pyx_v_ndarray_label_count_left);
/* "sklearn/tree/_tree.pyx":123
* self.label_count_init = <int*>ndarray_label_count_init.data
* self.ndarray_label_count_left = ndarray_label_count_left
* self.ndarray_label_count_right = ndarray_label_count_right # <<<<<<<<<<<<<<
* self.ndarray_label_count_init = ndarray_label_count_init
*
*/
__Pyx_INCREF(((PyObject *)__pyx_v_ndarray_label_count_right));
__Pyx_GIVEREF(((PyObject *)__pyx_v_ndarray_label_count_right));
__Pyx_GOTREF(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_right);
__Pyx_DECREF(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_right);
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_right = ((PyObject *)__pyx_v_ndarray_label_count_right);
/* "sklearn/tree/_tree.pyx":124
* self.ndarray_label_count_left = ndarray_label_count_left
* self.ndarray_label_count_right = ndarray_label_count_right
* self.ndarray_label_count_init = ndarray_label_count_init # <<<<<<<<<<<<<<
*
* cdef void init(self, DTYPE_t *y, BOOL_t *sample_mask, int n_samples,
*/
__Pyx_INCREF(((PyObject *)__pyx_v_ndarray_label_count_init));
__Pyx_GIVEREF(((PyObject *)__pyx_v_ndarray_label_count_init));
__Pyx_GOTREF(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_init);
__Pyx_DECREF(((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_init);
((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->ndarray_label_count_init = ((PyObject *)__pyx_v_ndarray_label_count_init);
__pyx_r = 0;
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);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_ndarray_label_count_right);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_ndarray_label_count_left);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_ndarray_label_count_init);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.__init__", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = -1;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_ndarray_label_count_right);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_ndarray_label_count_left);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_ndarray_label_count_init);
__pyx_L2:;
__Pyx_XDECREF((PyObject *)__pyx_v_ndarray_label_count_left);
__Pyx_XDECREF((PyObject *)__pyx_v_ndarray_label_count_right);
__Pyx_XDECREF((PyObject *)__pyx_v_ndarray_label_count_init);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":126
* self.ndarray_label_count_init = ndarray_label_count_init
*
* cdef void init(self, DTYPE_t *y, BOOL_t *sample_mask, int n_samples, # <<<<<<<<<<<<<<
* int n_total_samples):
* """Initialise the criterion class."""
*/
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_DTYPE_t *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask, int __pyx_v_n_samples, int __pyx_v_n_total_samples) {
int __pyx_v_c;
int __pyx_v_j;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
int __pyx_t_3;
__Pyx_RefNannySetupContext("init");
/* "sklearn/tree/_tree.pyx":129
* int n_total_samples):
* """Initialise the criterion class."""
* cdef int c = 0 # <<<<<<<<<<<<<<
* cdef int j = 0
*
*/
__pyx_v_c = 0;
/* "sklearn/tree/_tree.pyx":130
* """Initialise the criterion class."""
* cdef int c = 0
* cdef int j = 0 # <<<<<<<<<<<<<<
*
* self.n_samples = n_samples
*/
__pyx_v_j = 0;
/* "sklearn/tree/_tree.pyx":132
* cdef int j = 0
*
* self.n_samples = n_samples # <<<<<<<<<<<<<<
*
* for c from 0 <= c < self.n_classes:
*/
__pyx_v_self->n_samples = __pyx_v_n_samples;
/* "sklearn/tree/_tree.pyx":134
* self.n_samples = n_samples
*
* for c from 0 <= c < self.n_classes: # <<<<<<<<<<<<<<
* self.label_count_init[c] = 0
*
*/
__pyx_t_1 = __pyx_v_self->n_classes;
for (__pyx_v_c = 0; __pyx_v_c < __pyx_t_1; __pyx_v_c++) {
/* "sklearn/tree/_tree.pyx":135
*
* for c from 0 <= c < self.n_classes:
* self.label_count_init[c] = 0 # <<<<<<<<<<<<<<
*
* for j from 0 <= j < n_total_samples:
*/
(__pyx_v_self->label_count_init[__pyx_v_c]) = 0;
}
/* "sklearn/tree/_tree.pyx":137
* self.label_count_init[c] = 0
*
* for j from 0 <= j < n_total_samples: # <<<<<<<<<<<<<<
* if sample_mask[j] == 0:
* continue
*/
__pyx_t_1 = __pyx_v_n_total_samples;
for (__pyx_v_j = 0; __pyx_v_j < __pyx_t_1; __pyx_v_j++) {
/* "sklearn/tree/_tree.pyx":138
*
* for j from 0 <= j < n_total_samples:
* if sample_mask[j] == 0: # <<<<<<<<<<<<<<
* continue
* c = <int>(y[j])
*/
__pyx_t_2 = ((__pyx_v_sample_mask[__pyx_v_j]) == 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":139
* for j from 0 <= j < n_total_samples:
* if sample_mask[j] == 0:
* continue # <<<<<<<<<<<<<<
* c = <int>(y[j])
* self.label_count_init[c] += 1
*/
goto __pyx_L5_continue;
goto __pyx_L7;
}
__pyx_L7:;
/* "sklearn/tree/_tree.pyx":140
* if sample_mask[j] == 0:
* continue
* c = <int>(y[j]) # <<<<<<<<<<<<<<
* self.label_count_init[c] += 1
*
*/
__pyx_v_c = ((int)(__pyx_v_y[__pyx_v_j]));
/* "sklearn/tree/_tree.pyx":141
* continue
* c = <int>(y[j])
* self.label_count_init[c] += 1 # <<<<<<<<<<<<<<
*
* self.reset()
*/
__pyx_t_3 = __pyx_v_c;
(__pyx_v_self->label_count_init[__pyx_t_3]) = ((__pyx_v_self->label_count_init[__pyx_t_3]) + 1);
__pyx_L5_continue:;
}
/* "sklearn/tree/_tree.pyx":143
* self.label_count_init[c] += 1
*
* self.reset() # <<<<<<<<<<<<<<
*
* cdef void reset(self):
*/
((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));
__Pyx_RefNannyFinishContext();
}
/* "sklearn/tree/_tree.pyx":145
* self.reset()
*
* cdef void reset(self): # <<<<<<<<<<<<<<
* """Reset label_counts by setting `label_count_left to zero
* and copying the init array into the right."""
*/
static void __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_reset(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
int __pyx_v_c;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
__Pyx_RefNannySetupContext("reset");
/* "sklearn/tree/_tree.pyx":148
* """Reset label_counts by setting `label_count_left to zero
* and copying the init array into the right."""
* cdef int c = 0 # <<<<<<<<<<<<<<
* self.n_left = 0
* self.n_right = self.n_samples
*/
__pyx_v_c = 0;
/* "sklearn/tree/_tree.pyx":149
* and copying the init array into the right."""
* cdef int c = 0
* self.n_left = 0 # <<<<<<<<<<<<<<
* self.n_right = self.n_samples
*
*/
__pyx_v_self->n_left = 0;
/* "sklearn/tree/_tree.pyx":150
* cdef int c = 0
* self.n_left = 0
* self.n_right = self.n_samples # <<<<<<<<<<<<<<
*
* for c from 0 <= c < self.n_classes:
*/
__pyx_v_self->n_right = __pyx_v_self->n_samples;
/* "sklearn/tree/_tree.pyx":152
* self.n_right = self.n_samples
*
* for c from 0 <= c < self.n_classes: # <<<<<<<<<<<<<<
* self.label_count_left[c] = 0
* self.label_count_right[c] = self.label_count_init[c]
*/
__pyx_t_1 = __pyx_v_self->n_classes;
for (__pyx_v_c = 0; __pyx_v_c < __pyx_t_1; __pyx_v_c++) {
/* "sklearn/tree/_tree.pyx":153
*
* for c from 0 <= c < self.n_classes:
* self.label_count_left[c] = 0 # <<<<<<<<<<<<<<
* self.label_count_right[c] = self.label_count_init[c]
*
*/
(__pyx_v_self->label_count_left[__pyx_v_c]) = 0;
/* "sklearn/tree/_tree.pyx":154
* for c from 0 <= c < self.n_classes:
* self.label_count_left[c] = 0
* self.label_count_right[c] = self.label_count_init[c] # <<<<<<<<<<<<<<
*
* cdef int update(self, int a, int b, DTYPE_t *y, int *X_argsorted_i,
*/
(__pyx_v_self->label_count_right[__pyx_v_c]) = (__pyx_v_self->label_count_init[__pyx_v_c]);
}
__Pyx_RefNannyFinishContext();
}
/* "sklearn/tree/_tree.pyx":156
* self.label_count_right[c] = self.label_count_init[c]
*
* cdef int update(self, int a, int b, DTYPE_t *y, int *X_argsorted_i, # <<<<<<<<<<<<<<
* BOOL_t *sample_mask):
* """Update the criteria for each value in interval [a,b) (where a and b
*/
static int __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_update(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, int __pyx_v_a, int __pyx_v_b, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y, int *__pyx_v_X_argsorted_i, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask) {
int __pyx_v_c;
int __pyx_v_idx;
int __pyx_v_s;
int __pyx_r;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
int __pyx_t_3;
__Pyx_RefNannySetupContext("update");
/* "sklearn/tree/_tree.pyx":162
* cdef int c
* # post condition: all samples from [0:b) are on the left side
* for idx from a <= idx < b: # <<<<<<<<<<<<<<
* s = X_argsorted_i[idx]
* if sample_mask[s] == 0:
*/
__pyx_t_1 = __pyx_v_b;
for (__pyx_v_idx = __pyx_v_a; __pyx_v_idx < __pyx_t_1; __pyx_v_idx++) {
/* "sklearn/tree/_tree.pyx":163
* # post condition: all samples from [0:b) are on the left side
* for idx from a <= idx < b:
* s = X_argsorted_i[idx] # <<<<<<<<<<<<<<
* if sample_mask[s] == 0:
* continue
*/
__pyx_v_s = (__pyx_v_X_argsorted_i[__pyx_v_idx]);
/* "sklearn/tree/_tree.pyx":164
* for idx from a <= idx < b:
* s = X_argsorted_i[idx]
* if sample_mask[s] == 0: # <<<<<<<<<<<<<<
* continue
* c = <int>(y[s])
*/
__pyx_t_2 = ((__pyx_v_sample_mask[__pyx_v_s]) == 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":165
* s = X_argsorted_i[idx]
* if sample_mask[s] == 0:
* continue # <<<<<<<<<<<<<<
* c = <int>(y[s])
* self.label_count_right[c] -= 1
*/
goto __pyx_L3_continue;
goto __pyx_L5;
}
__pyx_L5:;
/* "sklearn/tree/_tree.pyx":166
* if sample_mask[s] == 0:
* continue
* c = <int>(y[s]) # <<<<<<<<<<<<<<
* self.label_count_right[c] -= 1
* self.label_count_left[c] += 1
*/
__pyx_v_c = ((int)(__pyx_v_y[__pyx_v_s]));
/* "sklearn/tree/_tree.pyx":167
* continue
* c = <int>(y[s])
* self.label_count_right[c] -= 1 # <<<<<<<<<<<<<<
* self.label_count_left[c] += 1
* self.n_right -= 1
*/
__pyx_t_3 = __pyx_v_c;
(__pyx_v_self->label_count_right[__pyx_t_3]) = ((__pyx_v_self->label_count_right[__pyx_t_3]) - 1);
/* "sklearn/tree/_tree.pyx":168
* c = <int>(y[s])
* self.label_count_right[c] -= 1
* self.label_count_left[c] += 1 # <<<<<<<<<<<<<<
* self.n_right -= 1
* self.n_left += 1
*/
__pyx_t_3 = __pyx_v_c;
(__pyx_v_self->label_count_left[__pyx_t_3]) = ((__pyx_v_self->label_count_left[__pyx_t_3]) + 1);
/* "sklearn/tree/_tree.pyx":169
* self.label_count_right[c] -= 1
* self.label_count_left[c] += 1
* self.n_right -= 1 # <<<<<<<<<<<<<<
* self.n_left += 1
*
*/
__pyx_v_self->n_right = (__pyx_v_self->n_right - 1);
/* "sklearn/tree/_tree.pyx":170
* self.label_count_left[c] += 1
* self.n_right -= 1
* self.n_left += 1 # <<<<<<<<<<<<<<
*
* return self.n_left
*/
__pyx_v_self->n_left = (__pyx_v_self->n_left + 1);
__pyx_L3_continue:;
}
/* "sklearn/tree/_tree.pyx":172
* self.n_left += 1
*
* return self.n_left # <<<<<<<<<<<<<<
*
* cdef double eval(self):
*/
__pyx_r = __pyx_v_self->n_left;
goto __pyx_L0;
__pyx_r = 0;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":174
* return self.n_left
*
* cdef double eval(self): # <<<<<<<<<<<<<<
* pass
*
*/
static double __pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_eval(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self) {
double __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("eval");
__pyx_r = 0;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":177
* pass
*
* cpdef np.ndarray init_value(self): # <<<<<<<<<<<<<<
* return self.ndarray_label_count_init
*
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_1init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_init_value(struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *__pyx_v_self, int __pyx_skip_dispatch) {
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("init_value");
/* Check if called by wrapper */
if (unlikely(__pyx_skip_dispatch)) ;
/* Check if overriden in Python */
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
__pyx_t_1 = PyObject_GetAttr(((PyObject *)__pyx_v_self), __pyx_n_s__init_value); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 177; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (void *)&__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_1init_value)) {
__Pyx_XDECREF(((PyObject *)__pyx_r));
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 177; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 177; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_r = ((PyArrayObject *)__pyx_t_2);
__pyx_t_2 = 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":178
*
* cpdef np.ndarray init_value(self):
* return self.ndarray_label_count_init # <<<<<<<<<<<<<<
*
*
*/
__Pyx_XDECREF(((PyObject *)__pyx_r));
if (!(likely(((__pyx_v_self->ndarray_label_count_init) == Py_None) || likely(__Pyx_TypeTest(__pyx_v_self->ndarray_label_count_init, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 178; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_INCREF(__pyx_v_self->ndarray_label_count_init);
__pyx_r = ((PyArrayObject *)__pyx_v_self->ndarray_label_count_init);
goto __pyx_L0;
__pyx_r = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_2);
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.init_value", __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":177
* pass
*
* cpdef np.ndarray init_value(self): # <<<<<<<<<<<<<<
* return self.ndarray_label_count_init
*
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_1init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_1init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
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("init_value");
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_ClassificationCriterion *)((struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)__pyx_v_self)->__pyx_base.__pyx_vtab)->__pyx_base.init_value(((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self), 1)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 177; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_AddTraceback("sklearn.tree._tree.ClassificationCriterion.init_value", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":188
* """
*
* cdef double eval(self): # <<<<<<<<<<<<<<
* """Returns Gini index of left branch + Gini index of right branch. """
* cdef double n_left = <double> self.n_left
*/
static double __pyx_f_7sklearn_4tree_5_tree_4Gini_eval(struct __pyx_obj_7sklearn_4tree_5_tree_Gini *__pyx_v_self) {
double __pyx_v_n_left;
double __pyx_v_n_right;
double __pyx_v_H_left;
double __pyx_v_H_right;
int __pyx_v_k;
int __pyx_v_count_left;
int __pyx_v_count_right;
double __pyx_r;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
__Pyx_RefNannySetupContext("eval");
/* "sklearn/tree/_tree.pyx":190
* cdef double eval(self):
* """Returns Gini index of left branch + Gini index of right branch. """
* cdef double n_left = <double> self.n_left # <<<<<<<<<<<<<<
* cdef double n_right = <double> self.n_right
* cdef double H_left = n_left * n_left
*/
__pyx_v_n_left = ((double)__pyx_v_self->__pyx_base.n_left);
/* "sklearn/tree/_tree.pyx":191
* """Returns Gini index of left branch + Gini index of right branch. """
* cdef double n_left = <double> self.n_left
* cdef double n_right = <double> self.n_right # <<<<<<<<<<<<<<
* cdef double H_left = n_left * n_left
* cdef double H_right = n_right * n_right
*/
__pyx_v_n_right = ((double)__pyx_v_self->__pyx_base.n_right);
/* "sklearn/tree/_tree.pyx":192
* cdef double n_left = <double> self.n_left
* cdef double n_right = <double> self.n_right
* cdef double H_left = n_left * n_left # <<<<<<<<<<<<<<
* cdef double H_right = n_right * n_right
* cdef int k, count_left, count_right
*/
__pyx_v_H_left = (__pyx_v_n_left * __pyx_v_n_left);
/* "sklearn/tree/_tree.pyx":193
* cdef double n_right = <double> self.n_right
* cdef double H_left = n_left * n_left
* cdef double H_right = n_right * n_right # <<<<<<<<<<<<<<
* cdef int k, count_left, count_right
*
*/
__pyx_v_H_right = (__pyx_v_n_right * __pyx_v_n_right);
/* "sklearn/tree/_tree.pyx":196
* cdef int k, count_left, count_right
*
* for k from 0 <= k < self.n_classes: # <<<<<<<<<<<<<<
* count_left = self.label_count_left[k]
* if count_left > 0:
*/
__pyx_t_1 = __pyx_v_self->__pyx_base.n_classes;
for (__pyx_v_k = 0; __pyx_v_k < __pyx_t_1; __pyx_v_k++) {
/* "sklearn/tree/_tree.pyx":197
*
* for k from 0 <= k < self.n_classes:
* count_left = self.label_count_left[k] # <<<<<<<<<<<<<<
* if count_left > 0:
* H_left -= (count_left * count_left)
*/
__pyx_v_count_left = (__pyx_v_self->__pyx_base.label_count_left[__pyx_v_k]);
/* "sklearn/tree/_tree.pyx":198
* for k from 0 <= k < self.n_classes:
* count_left = self.label_count_left[k]
* if count_left > 0: # <<<<<<<<<<<<<<
* H_left -= (count_left * count_left)
* count_right = self.label_count_right[k]
*/
__pyx_t_2 = (__pyx_v_count_left > 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":199
* count_left = self.label_count_left[k]
* if count_left > 0:
* H_left -= (count_left * count_left) # <<<<<<<<<<<<<<
* count_right = self.label_count_right[k]
* if count_right > 0:
*/
__pyx_v_H_left = (__pyx_v_H_left - (__pyx_v_count_left * __pyx_v_count_left));
goto __pyx_L5;
}
__pyx_L5:;
/* "sklearn/tree/_tree.pyx":200
* if count_left > 0:
* H_left -= (count_left * count_left)
* count_right = self.label_count_right[k] # <<<<<<<<<<<<<<
* if count_right > 0:
* H_right -= (count_right * count_right)
*/
__pyx_v_count_right = (__pyx_v_self->__pyx_base.label_count_right[__pyx_v_k]);
/* "sklearn/tree/_tree.pyx":201
* H_left -= (count_left * count_left)
* count_right = self.label_count_right[k]
* if count_right > 0: # <<<<<<<<<<<<<<
* H_right -= (count_right * count_right)
*
*/
__pyx_t_2 = (__pyx_v_count_right > 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":202
* count_right = self.label_count_right[k]
* if count_right > 0:
* H_right -= (count_right * count_right) # <<<<<<<<<<<<<<
*
* if n_left == 0:
*/
__pyx_v_H_right = (__pyx_v_H_right - (__pyx_v_count_right * __pyx_v_count_right));
goto __pyx_L6;
}
__pyx_L6:;
}
/* "sklearn/tree/_tree.pyx":204
* H_right -= (count_right * count_right)
*
* if n_left == 0: # <<<<<<<<<<<<<<
* H_left = 0
* else:
*/
__pyx_t_2 = (__pyx_v_n_left == 0.0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":205
*
* if n_left == 0:
* H_left = 0 # <<<<<<<<<<<<<<
* else:
* H_left /= n_left
*/
__pyx_v_H_left = 0.0;
goto __pyx_L7;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":207
* H_left = 0
* else:
* H_left /= n_left # <<<<<<<<<<<<<<
*
* if n_right == 0:
*/
__pyx_v_H_left = (__pyx_v_H_left / __pyx_v_n_left);
}
__pyx_L7:;
/* "sklearn/tree/_tree.pyx":209
* H_left /= n_left
*
* if n_right == 0: # <<<<<<<<<<<<<<
* H_right = 0
* else:
*/
__pyx_t_2 = (__pyx_v_n_right == 0.0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":210
*
* if n_right == 0:
* H_right = 0 # <<<<<<<<<<<<<<
* else:
* H_right /= n_right
*/
__pyx_v_H_right = 0.0;
goto __pyx_L8;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":212
* H_right = 0
* else:
* H_right /= n_right # <<<<<<<<<<<<<<
*
* return (H_left + H_right) / self.n_samples
*/
__pyx_v_H_right = (__pyx_v_H_right / __pyx_v_n_right);
}
__pyx_L8:;
/* "sklearn/tree/_tree.pyx":214
* H_right /= n_right
*
* return (H_left + H_right) / self.n_samples # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = ((__pyx_v_H_left + __pyx_v_H_right) / __pyx_v_self->__pyx_base.n_samples);
goto __pyx_L0;
__pyx_r = 0;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":223
* """
*
* cdef double eval(self): # <<<<<<<<<<<<<<
* """Returns Entropy of left branch + Entropy index of right branch. """
* cdef double H_left = 0.0
*/
static double __pyx_f_7sklearn_4tree_5_tree_7Entropy_eval(struct __pyx_obj_7sklearn_4tree_5_tree_Entropy *__pyx_v_self) {
double __pyx_v_H_left;
double __pyx_v_H_right;
int __pyx_v_k;
double __pyx_v_e1;
double __pyx_v_e2;
double __pyx_v_n_left;
double __pyx_v_n_right;
double __pyx_r;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
__Pyx_RefNannySetupContext("eval");
/* "sklearn/tree/_tree.pyx":225
* cdef double eval(self):
* """Returns Entropy of left branch + Entropy index of right branch. """
* cdef double H_left = 0.0 # <<<<<<<<<<<<<<
* cdef double H_right = 0.0
* cdef int k
*/
__pyx_v_H_left = 0.0;
/* "sklearn/tree/_tree.pyx":226
* """Returns Entropy of left branch + Entropy index of right branch. """
* cdef double H_left = 0.0
* cdef double H_right = 0.0 # <<<<<<<<<<<<<<
* cdef int k
* cdef double e1, e2
*/
__pyx_v_H_right = 0.0;
/* "sklearn/tree/_tree.pyx":229
* cdef int k
* cdef double e1, e2
* cdef double n_left = <double> self.n_left # <<<<<<<<<<<<<<
* cdef double n_right = <double> self.n_right
*
*/
__pyx_v_n_left = ((double)__pyx_v_self->__pyx_base.n_left);
/* "sklearn/tree/_tree.pyx":230
* cdef double e1, e2
* cdef double n_left = <double> self.n_left
* cdef double n_right = <double> self.n_right # <<<<<<<<<<<<<<
*
* for k from 0 <= k < self.n_classes:
*/
__pyx_v_n_right = ((double)__pyx_v_self->__pyx_base.n_right);
/* "sklearn/tree/_tree.pyx":232
* cdef double n_right = <double> self.n_right
*
* for k from 0 <= k < self.n_classes: # <<<<<<<<<<<<<<
* if self.label_count_left[k] > 0:
* H_left -= ((self.label_count_left[k] / n_left)
*/
__pyx_t_1 = __pyx_v_self->__pyx_base.n_classes;
for (__pyx_v_k = 0; __pyx_v_k < __pyx_t_1; __pyx_v_k++) {
/* "sklearn/tree/_tree.pyx":233
*
* for k from 0 <= k < self.n_classes:
* if self.label_count_left[k] > 0: # <<<<<<<<<<<<<<
* H_left -= ((self.label_count_left[k] / n_left)
* * log(self.label_count_left[k] / n_left))
*/
__pyx_t_2 = ((__pyx_v_self->__pyx_base.label_count_left[__pyx_v_k]) > 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":235
* if self.label_count_left[k] > 0:
* H_left -= ((self.label_count_left[k] / n_left)
* * log(self.label_count_left[k] / n_left)) # <<<<<<<<<<<<<<
* if self.label_count_right[k] > 0:
* H_right -= ((self.label_count_right[k] / n_right)
*/
__pyx_v_H_left = (__pyx_v_H_left - (((__pyx_v_self->__pyx_base.label_count_left[__pyx_v_k]) / __pyx_v_n_left) * log(((__pyx_v_self->__pyx_base.label_count_left[__pyx_v_k]) / __pyx_v_n_left))));
goto __pyx_L5;
}
__pyx_L5:;
/* "sklearn/tree/_tree.pyx":236
* H_left -= ((self.label_count_left[k] / n_left)
* * log(self.label_count_left[k] / n_left))
* if self.label_count_right[k] > 0: # <<<<<<<<<<<<<<
* H_right -= ((self.label_count_right[k] / n_right)
* * log(self.label_count_right[k] / n_right))
*/
__pyx_t_2 = ((__pyx_v_self->__pyx_base.label_count_right[__pyx_v_k]) > 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":238
* if self.label_count_right[k] > 0:
* H_right -= ((self.label_count_right[k] / n_right)
* * log(self.label_count_right[k] / n_right)) # <<<<<<<<<<<<<<
*
* e1 = (n_left / self.n_samples) * H_left
*/
__pyx_v_H_right = (__pyx_v_H_right - (((__pyx_v_self->__pyx_base.label_count_right[__pyx_v_k]) / __pyx_v_n_right) * log(((__pyx_v_self->__pyx_base.label_count_right[__pyx_v_k]) / __pyx_v_n_right))));
goto __pyx_L6;
}
__pyx_L6:;
}
/* "sklearn/tree/_tree.pyx":240
* * log(self.label_count_right[k] / n_right))
*
* e1 = (n_left / self.n_samples) * H_left # <<<<<<<<<<<<<<
* e2 = (n_right / self.n_samples) * H_right
* return e1 + e2
*/
__pyx_v_e1 = ((__pyx_v_n_left / __pyx_v_self->__pyx_base.n_samples) * __pyx_v_H_left);
/* "sklearn/tree/_tree.pyx":241
*
* e1 = (n_left / self.n_samples) * H_left
* e2 = (n_right / self.n_samples) * H_right # <<<<<<<<<<<<<<
* return e1 + e2
*
*/
__pyx_v_e2 = ((__pyx_v_n_right / __pyx_v_self->__pyx_base.n_samples) * __pyx_v_H_right);
/* "sklearn/tree/_tree.pyx":242
* e1 = (n_left / self.n_samples) * H_left
* e2 = (n_right / self.n_samples) * H_right
* return e1 + e2 # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = (__pyx_v_e1 + __pyx_v_e2);
goto __pyx_L0;
__pyx_r = 0;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":299
* cdef double var_right
*
* def __init__(self): # <<<<<<<<<<<<<<
* self.n_samples = 0
* self.n_left = 0
*/
static int __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion___init__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static int __pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion___init__(PyObject *__pyx_v_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
int __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("__init__");
if (unlikely(PyTuple_GET_SIZE(__pyx_args) > 0)) {
__Pyx_RaiseArgtupleInvalid("__init__", 1, 0, 0, PyTuple_GET_SIZE(__pyx_args)); return -1;}
if (unlikely(__pyx_kwds) && unlikely(PyDict_Size(__pyx_kwds) > 0) && unlikely(!__Pyx_CheckKeywordStrings(__pyx_kwds, "__init__", 0))) return -1;
/* "sklearn/tree/_tree.pyx":300
*
* def __init__(self):
* self.n_samples = 0 # <<<<<<<<<<<<<<
* self.n_left = 0
* self.n_right = 0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->n_samples = 0;
/* "sklearn/tree/_tree.pyx":301
* def __init__(self):
* self.n_samples = 0
* self.n_left = 0 # <<<<<<<<<<<<<<
* self.n_right = 0
* self.mean_left = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->n_left = 0;
/* "sklearn/tree/_tree.pyx":302
* self.n_samples = 0
* self.n_left = 0
* self.n_right = 0 # <<<<<<<<<<<<<<
* self.mean_left = 0.0
* self.mean_right = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->n_right = 0;
/* "sklearn/tree/_tree.pyx":303
* self.n_left = 0
* self.n_right = 0
* self.mean_left = 0.0 # <<<<<<<<<<<<<<
* self.mean_right = 0.0
* self.mean_init = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->mean_left = 0.0;
/* "sklearn/tree/_tree.pyx":304
* self.n_right = 0
* self.mean_left = 0.0
* self.mean_right = 0.0 # <<<<<<<<<<<<<<
* self.mean_init = 0.0
* self.sq_sum_right = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->mean_right = 0.0;
/* "sklearn/tree/_tree.pyx":305
* self.mean_left = 0.0
* self.mean_right = 0.0
* self.mean_init = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_right = 0.0
* self.sq_sum_left = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->mean_init = 0.0;
/* "sklearn/tree/_tree.pyx":306
* self.mean_right = 0.0
* self.mean_init = 0.0
* self.sq_sum_right = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_left = 0.0
* self.sq_sum_init = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->sq_sum_right = 0.0;
/* "sklearn/tree/_tree.pyx":307
* self.mean_init = 0.0
* self.sq_sum_right = 0.0
* self.sq_sum_left = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_init = 0.0
* self.var_left = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->sq_sum_left = 0.0;
/* "sklearn/tree/_tree.pyx":308
* self.sq_sum_right = 0.0
* self.sq_sum_left = 0.0
* self.sq_sum_init = 0.0 # <<<<<<<<<<<<<<
* self.var_left = 0.0
* self.var_right = 0.0
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->sq_sum_init = 0.0;
/* "sklearn/tree/_tree.pyx":309
* self.sq_sum_left = 0.0
* self.sq_sum_init = 0.0
* self.var_left = 0.0 # <<<<<<<<<<<<<<
* self.var_right = 0.0
*
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->var_left = 0.0;
/* "sklearn/tree/_tree.pyx":310
* self.sq_sum_init = 0.0
* self.var_left = 0.0
* self.var_right = 0.0 # <<<<<<<<<<<<<<
*
* cdef void init(self, DTYPE_t *y, BOOL_t *sample_mask, int n_samples,
*/
((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->var_right = 0.0;
__pyx_r = 0;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":312
* self.var_right = 0.0
*
* cdef void init(self, DTYPE_t *y, BOOL_t *sample_mask, int n_samples, # <<<<<<<<<<<<<<
* int n_total_samples):
* """Initialise the criterion class; assume all samples
*/
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_DTYPE_t *__pyx_v_y, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask, int __pyx_v_n_samples, int __pyx_v_n_total_samples) {
int __pyx_v_j;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
__Pyx_RefNannySetupContext("init");
/* "sklearn/tree/_tree.pyx":317
* are in the right branch and store the mean and squared
* sum in `self.mean_init` and `self.sq_sum_init`. """
* self.mean_left = 0.0 # <<<<<<<<<<<<<<
* self.mean_right = 0.0
* self.mean_init = 0.0
*/
__pyx_v_self->mean_left = 0.0;
/* "sklearn/tree/_tree.pyx":318
* sum in `self.mean_init` and `self.sq_sum_init`. """
* self.mean_left = 0.0
* self.mean_right = 0.0 # <<<<<<<<<<<<<<
* self.mean_init = 0.0
* self.sq_sum_right = 0.0
*/
__pyx_v_self->mean_right = 0.0;
/* "sklearn/tree/_tree.pyx":319
* self.mean_left = 0.0
* self.mean_right = 0.0
* self.mean_init = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_right = 0.0
* self.sq_sum_left = 0.0
*/
__pyx_v_self->mean_init = 0.0;
/* "sklearn/tree/_tree.pyx":320
* self.mean_right = 0.0
* self.mean_init = 0.0
* self.sq_sum_right = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_left = 0.0
* self.sq_sum_init = 0.0
*/
__pyx_v_self->sq_sum_right = 0.0;
/* "sklearn/tree/_tree.pyx":321
* self.mean_init = 0.0
* self.sq_sum_right = 0.0
* self.sq_sum_left = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_init = 0.0
* self.var_left = 0.0
*/
__pyx_v_self->sq_sum_left = 0.0;
/* "sklearn/tree/_tree.pyx":322
* self.sq_sum_right = 0.0
* self.sq_sum_left = 0.0
* self.sq_sum_init = 0.0 # <<<<<<<<<<<<<<
* self.var_left = 0.0
* self.var_right = 0.0
*/
__pyx_v_self->sq_sum_init = 0.0;
/* "sklearn/tree/_tree.pyx":323
* self.sq_sum_left = 0.0
* self.sq_sum_init = 0.0
* self.var_left = 0.0 # <<<<<<<<<<<<<<
* self.var_right = 0.0
* self.n_samples = n_samples
*/
__pyx_v_self->var_left = 0.0;
/* "sklearn/tree/_tree.pyx":324
* self.sq_sum_init = 0.0
* self.var_left = 0.0
* self.var_right = 0.0 # <<<<<<<<<<<<<<
* self.n_samples = n_samples
*
*/
__pyx_v_self->var_right = 0.0;
/* "sklearn/tree/_tree.pyx":325
* self.var_left = 0.0
* self.var_right = 0.0
* self.n_samples = n_samples # <<<<<<<<<<<<<<
*
* cdef int j = 0
*/
__pyx_v_self->n_samples = __pyx_v_n_samples;
/* "sklearn/tree/_tree.pyx":327
* self.n_samples = n_samples
*
* cdef int j = 0 # <<<<<<<<<<<<<<
* for j from 0 <= j < n_total_samples:
* if sample_mask[j] == 0:
*/
__pyx_v_j = 0;
/* "sklearn/tree/_tree.pyx":328
*
* cdef int j = 0
* for j from 0 <= j < n_total_samples: # <<<<<<<<<<<<<<
* if sample_mask[j] == 0:
* continue
*/
__pyx_t_1 = __pyx_v_n_total_samples;
for (__pyx_v_j = 0; __pyx_v_j < __pyx_t_1; __pyx_v_j++) {
/* "sklearn/tree/_tree.pyx":329
* cdef int j = 0
* for j from 0 <= j < n_total_samples:
* if sample_mask[j] == 0: # <<<<<<<<<<<<<<
* continue
* self.sq_sum_init += (y[j] * y[j])
*/
__pyx_t_2 = ((__pyx_v_sample_mask[__pyx_v_j]) == 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":330
* for j from 0 <= j < n_total_samples:
* if sample_mask[j] == 0:
* continue # <<<<<<<<<<<<<<
* self.sq_sum_init += (y[j] * y[j])
* self.mean_init += y[j]
*/
goto __pyx_L3_continue;
goto __pyx_L5;
}
__pyx_L5:;
/* "sklearn/tree/_tree.pyx":331
* if sample_mask[j] == 0:
* continue
* self.sq_sum_init += (y[j] * y[j]) # <<<<<<<<<<<<<<
* self.mean_init += y[j]
*
*/
__pyx_v_self->sq_sum_init = (__pyx_v_self->sq_sum_init + ((__pyx_v_y[__pyx_v_j]) * (__pyx_v_y[__pyx_v_j])));
/* "sklearn/tree/_tree.pyx":332
* continue
* self.sq_sum_init += (y[j] * y[j])
* self.mean_init += y[j] # <<<<<<<<<<<<<<
*
* self.mean_init = self.mean_init / self.n_samples
*/
__pyx_v_self->mean_init = (__pyx_v_self->mean_init + (__pyx_v_y[__pyx_v_j]));
__pyx_L3_continue:;
}
/* "sklearn/tree/_tree.pyx":334
* self.mean_init += y[j]
*
* self.mean_init = self.mean_init / self.n_samples # <<<<<<<<<<<<<<
*
* self.reset()
*/
__pyx_v_self->mean_init = (__pyx_v_self->mean_init / __pyx_v_self->n_samples);
/* "sklearn/tree/_tree.pyx":336
* self.mean_init = self.mean_init / self.n_samples
*
* self.reset() # <<<<<<<<<<<<<<
*
* cdef void reset(self):
*/
((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));
__Pyx_RefNannyFinishContext();
}
/* "sklearn/tree/_tree.pyx":338
* self.reset()
*
* cdef void reset(self): # <<<<<<<<<<<<<<
* """Reset criterion for new feature.
*
*/
static void __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_reset(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("reset");
/* "sklearn/tree/_tree.pyx":345
* right branch.
* """
* self.n_right = self.n_samples # <<<<<<<<<<<<<<
* self.n_left = 0
* self.mean_right = self.mean_init
*/
__pyx_v_self->n_right = __pyx_v_self->n_samples;
/* "sklearn/tree/_tree.pyx":346
* """
* self.n_right = self.n_samples
* self.n_left = 0 # <<<<<<<<<<<<<<
* self.mean_right = self.mean_init
* self.mean_left = 0.0
*/
__pyx_v_self->n_left = 0;
/* "sklearn/tree/_tree.pyx":347
* self.n_right = self.n_samples
* self.n_left = 0
* self.mean_right = self.mean_init # <<<<<<<<<<<<<<
* self.mean_left = 0.0
* self.sq_sum_right = self.sq_sum_init
*/
__pyx_v_self->mean_right = __pyx_v_self->mean_init;
/* "sklearn/tree/_tree.pyx":348
* self.n_left = 0
* self.mean_right = self.mean_init
* self.mean_left = 0.0 # <<<<<<<<<<<<<<
* self.sq_sum_right = self.sq_sum_init
* self.sq_sum_left = 0.0
*/
__pyx_v_self->mean_left = 0.0;
/* "sklearn/tree/_tree.pyx":349
* self.mean_right = self.mean_init
* self.mean_left = 0.0
* self.sq_sum_right = self.sq_sum_init # <<<<<<<<<<<<<<
* self.sq_sum_left = 0.0
* self.var_left = 0.0
*/
__pyx_v_self->sq_sum_right = __pyx_v_self->sq_sum_init;
/* "sklearn/tree/_tree.pyx":350
* self.mean_left = 0.0
* self.sq_sum_right = self.sq_sum_init
* self.sq_sum_left = 0.0 # <<<<<<<<<<<<<<
* self.var_left = 0.0
* self.var_right = self.sq_sum_right - \
*/
__pyx_v_self->sq_sum_left = 0.0;
/* "sklearn/tree/_tree.pyx":351
* self.sq_sum_right = self.sq_sum_init
* self.sq_sum_left = 0.0
* self.var_left = 0.0 # <<<<<<<<<<<<<<
* self.var_right = self.sq_sum_right - \
* self.n_samples * (self.mean_right * self.mean_right)
*/
__pyx_v_self->var_left = 0.0;
/* "sklearn/tree/_tree.pyx":352
* self.sq_sum_left = 0.0
* self.var_left = 0.0
* self.var_right = self.sq_sum_right - \ # <<<<<<<<<<<<<<
* self.n_samples * (self.mean_right * self.mean_right)
*
*/
__pyx_v_self->var_right = (__pyx_v_self->sq_sum_right - (__pyx_v_self->n_samples * (__pyx_v_self->mean_right * __pyx_v_self->mean_right)));
__Pyx_RefNannyFinishContext();
}
/* "sklearn/tree/_tree.pyx":355
* self.n_samples * (self.mean_right * self.mean_right)
*
* cdef int update(self, int a, int b, DTYPE_t *y, int *X_argsorted_i, # <<<<<<<<<<<<<<
* BOOL_t *sample_mask):
* """Update the criteria for each value in interval [a,b) (where a and b
*/
static int __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_update(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, int __pyx_v_a, int __pyx_v_b, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y, int *__pyx_v_X_argsorted_i, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask) {
double __pyx_v_y_idx;
int __pyx_v_idx;
int __pyx_v_j;
int __pyx_r;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
__Pyx_RefNannySetupContext("update");
/* "sklearn/tree/_tree.pyx":359
* """Update the criteria for each value in interval [a,b) (where a and b
* are indices in `X_argsorted_i`)."""
* cdef double y_idx = 0.0 # <<<<<<<<<<<<<<
* cdef int idx, j
* # post condition: all samples from [0:b) are on the left side
*/
__pyx_v_y_idx = 0.0;
/* "sklearn/tree/_tree.pyx":362
* cdef int idx, j
* # post condition: all samples from [0:b) are on the left side
* for idx from a <= idx < b: # <<<<<<<<<<<<<<
* j = X_argsorted_i[idx]
* if sample_mask[j] == 0:
*/
__pyx_t_1 = __pyx_v_b;
for (__pyx_v_idx = __pyx_v_a; __pyx_v_idx < __pyx_t_1; __pyx_v_idx++) {
/* "sklearn/tree/_tree.pyx":363
* # post condition: all samples from [0:b) are on the left side
* for idx from a <= idx < b:
* j = X_argsorted_i[idx] # <<<<<<<<<<<<<<
* if sample_mask[j] == 0:
* continue
*/
__pyx_v_j = (__pyx_v_X_argsorted_i[__pyx_v_idx]);
/* "sklearn/tree/_tree.pyx":364
* for idx from a <= idx < b:
* j = X_argsorted_i[idx]
* if sample_mask[j] == 0: # <<<<<<<<<<<<<<
* continue
* y_idx = y[j]
*/
__pyx_t_2 = ((__pyx_v_sample_mask[__pyx_v_j]) == 0);
if (__pyx_t_2) {
/* "sklearn/tree/_tree.pyx":365
* j = X_argsorted_i[idx]
* if sample_mask[j] == 0:
* continue # <<<<<<<<<<<<<<
* y_idx = y[j]
* self.sq_sum_left = self.sq_sum_left + (y_idx * y_idx)
*/
goto __pyx_L3_continue;
goto __pyx_L5;
}
__pyx_L5:;
/* "sklearn/tree/_tree.pyx":366
* if sample_mask[j] == 0:
* continue
* y_idx = y[j] # <<<<<<<<<<<<<<
* self.sq_sum_left = self.sq_sum_left + (y_idx * y_idx)
* self.sq_sum_right = self.sq_sum_right - (y_idx * y_idx)
*/
__pyx_v_y_idx = (__pyx_v_y[__pyx_v_j]);
/* "sklearn/tree/_tree.pyx":367
* continue
* y_idx = y[j]
* self.sq_sum_left = self.sq_sum_left + (y_idx * y_idx) # <<<<<<<<<<<<<<
* self.sq_sum_right = self.sq_sum_right - (y_idx * y_idx)
*
*/
__pyx_v_self->sq_sum_left = (__pyx_v_self->sq_sum_left + (__pyx_v_y_idx * __pyx_v_y_idx));
/* "sklearn/tree/_tree.pyx":368
* y_idx = y[j]
* self.sq_sum_left = self.sq_sum_left + (y_idx * y_idx)
* self.sq_sum_right = self.sq_sum_right - (y_idx * y_idx) # <<<<<<<<<<<<<<
*
* self.mean_left = (self.n_left * self.mean_left + y_idx) / \
*/
__pyx_v_self->sq_sum_right = (__pyx_v_self->sq_sum_right - (__pyx_v_y_idx * __pyx_v_y_idx));
/* "sklearn/tree/_tree.pyx":370
* self.sq_sum_right = self.sq_sum_right - (y_idx * y_idx)
*
* self.mean_left = (self.n_left * self.mean_left + y_idx) / \ # <<<<<<<<<<<<<<
* <double>(self.n_left + 1)
* self.mean_right = ((self.n_samples - self.n_left) * \
*/
__pyx_v_self->mean_left = (((__pyx_v_self->n_left * __pyx_v_self->mean_left) + __pyx_v_y_idx) / ((double)(__pyx_v_self->n_left + 1)));
/* "sklearn/tree/_tree.pyx":372
* self.mean_left = (self.n_left * self.mean_left + y_idx) / \
* <double>(self.n_left + 1)
* self.mean_right = ((self.n_samples - self.n_left) * \ # <<<<<<<<<<<<<<
* self.mean_right - y_idx) / \
* <double>(self.n_samples - self.n_left - 1)
*/
__pyx_v_self->mean_right = ((((__pyx_v_self->n_samples - __pyx_v_self->n_left) * __pyx_v_self->mean_right) - __pyx_v_y_idx) / ((double)((__pyx_v_self->n_samples - __pyx_v_self->n_left) - 1)));
/* "sklearn/tree/_tree.pyx":376
* <double>(self.n_samples - self.n_left - 1)
*
* self.n_right -= 1 # <<<<<<<<<<<<<<
* self.n_left += 1
*
*/
__pyx_v_self->n_right = (__pyx_v_self->n_right - 1);
/* "sklearn/tree/_tree.pyx":377
*
* self.n_right -= 1
* self.n_left += 1 # <<<<<<<<<<<<<<
*
* self.var_left = self.sq_sum_left - \
*/
__pyx_v_self->n_left = (__pyx_v_self->n_left + 1);
/* "sklearn/tree/_tree.pyx":379
* self.n_left += 1
*
* self.var_left = self.sq_sum_left - \ # <<<<<<<<<<<<<<
* self.n_left * (self.mean_left * self.mean_left)
* self.var_right = self.sq_sum_right - \
*/
__pyx_v_self->var_left = (__pyx_v_self->sq_sum_left - (__pyx_v_self->n_left * (__pyx_v_self->mean_left * __pyx_v_self->mean_left)));
/* "sklearn/tree/_tree.pyx":381
* self.var_left = self.sq_sum_left - \
* self.n_left * (self.mean_left * self.mean_left)
* self.var_right = self.sq_sum_right - \ # <<<<<<<<<<<<<<
* self.n_right * (self.mean_right * self.mean_right)
*
*/
__pyx_v_self->var_right = (__pyx_v_self->sq_sum_right - (__pyx_v_self->n_right * (__pyx_v_self->mean_right * __pyx_v_self->mean_right)));
__pyx_L3_continue:;
}
/* "sklearn/tree/_tree.pyx":384
* self.n_right * (self.mean_right * self.mean_right)
*
* return self.n_left # <<<<<<<<<<<<<<
*
* cdef double eval(self):
*/
__pyx_r = __pyx_v_self->n_left;
goto __pyx_L0;
__pyx_r = 0;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":386
* return self.n_left
*
* cdef double eval(self): # <<<<<<<<<<<<<<
* pass
*
*/
static double __pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_eval(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self) {
double __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("eval");
__pyx_r = 0;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":389
* pass
*
* cpdef np.ndarray init_value(self): # <<<<<<<<<<<<<<
* ## TODO is calling np.asarray a performance issue?
* return np.asarray(self.mean_init)
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_1init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
static PyArrayObject *__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_init_value(struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *__pyx_v_self, 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_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
__Pyx_RefNannySetupContext("init_value");
/* Check if called by wrapper */
if (unlikely(__pyx_skip_dispatch)) ;
/* Check if overriden in Python */
else if (unlikely(Py_TYPE(((PyObject *)__pyx_v_self))->tp_dictoffset != 0)) {
__pyx_t_1 = PyObject_GetAttr(((PyObject *)__pyx_v_self), __pyx_n_s__init_value); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 389; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
if (!PyCFunction_Check(__pyx_t_1) || (PyCFunction_GET_FUNCTION(__pyx_t_1) != (void *)&__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_1init_value)) {
__Pyx_XDECREF(((PyObject *)__pyx_r));
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 389; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (!(likely(((__pyx_t_2) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_2, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 389; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_r = ((PyArrayObject *)__pyx_t_2);
__pyx_t_2 = 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":391
* cpdef np.ndarray init_value(self):
* ## TODO is calling np.asarray a performance issue?
* return np.asarray(self.mean_init) # <<<<<<<<<<<<<<
*
*
*/
__Pyx_XDECREF(((PyObject *)__pyx_r));
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 391; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__asarray); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 391; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_1 = PyFloat_FromDouble(__pyx_v_self->mean_init); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 391; __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 = 391; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_1);
__Pyx_GIVEREF(__pyx_t_1);
__pyx_t_1 = 0;
__pyx_t_1 = PyObject_Call(__pyx_t_2, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 391; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
__Pyx_DECREF(((PyObject *)__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 = 391; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_r = ((PyArrayObject *)__pyx_t_1);
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = ((PyArrayObject *)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.RegressionCriterion.init_value", __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":389
* pass
*
* cpdef np.ndarray init_value(self): # <<<<<<<<<<<<<<
* ## TODO is calling np.asarray a performance issue?
* return np.asarray(self.mean_init)
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_1init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused); /*proto*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_1init_value(PyObject *__pyx_v_self, CYTHON_UNUSED PyObject *unused) {
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("init_value");
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = ((PyObject *)((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_RegressionCriterion *)((struct __pyx_obj_7sklearn_4tree_5_tree_RegressionCriterion *)__pyx_v_self)->__pyx_base.__pyx_vtab)->__pyx_base.init_value(((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)__pyx_v_self), 1)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 389; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_AddTraceback("sklearn.tree._tree.RegressionCriterion.init_value", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
__pyx_L0:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":400
* """
*
* cdef double eval(self): # <<<<<<<<<<<<<<
* return self.var_left + self.var_right
*
*/
static double __pyx_f_7sklearn_4tree_5_tree_3MSE_eval(struct __pyx_obj_7sklearn_4tree_5_tree_MSE *__pyx_v_self) {
double __pyx_r;
__Pyx_RefNannyDeclarations
__Pyx_RefNannySetupContext("eval");
/* "sklearn/tree/_tree.pyx":401
*
* cdef double eval(self):
* return self.var_left + self.var_right # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = (__pyx_v_self->__pyx_base.var_left + __pyx_v_self->__pyx_base.var_right);
goto __pyx_L0;
__pyx_r = 0;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":409
*
*
* def _random_sample_mask(int n_total_samples, int n_total_in_bag, random_state): # <<<<<<<<<<<<<<
* """Create a random sample mask where ``n_total_in_bag`` elements are set.
*
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree__random_sample_mask(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree__random_sample_mask[] = "Create a random sample mask where ``n_total_in_bag`` elements are set.\n\n Parameters\n ----------\n n_total_samples : int\n The length of the resulting mask.\n n_total_in_bag : int\n The number of elements in the sample mask which are set to 1.\n random_state : np.RandomState\n A numpy ``RandomState`` object.\n\n Returns\n -------\n sample_mask : np.ndarray, shape=[n_total_samples]\n An ndarray where ``n_total_in_bag`` elements are set to ``True``\n the others are ``False``.\n ";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree__random_sample_mask = {__Pyx_NAMESTR("_random_sample_mask"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree__random_sample_mask, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree__random_sample_mask)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree__random_sample_mask(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
int __pyx_v_n_total_samples;
int __pyx_v_n_total_in_bag;
PyObject *__pyx_v_random_state = 0;
PyArrayObject *__pyx_v_rand = 0;
PyArrayObject *__pyx_v_sample_mask = 0;
int __pyx_v_n_bagged;
int __pyx_v_i;
Py_buffer __pyx_bstruct_rand;
Py_ssize_t __pyx_bstride_0_rand = 0;
Py_ssize_t __pyx_bshape_0_rand = 0;
Py_buffer __pyx_bstruct_sample_mask;
Py_ssize_t __pyx_bstride_0_sample_mask = 0;
Py_ssize_t __pyx_bshape_0_sample_mask = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
PyObject *__pyx_t_2 = NULL;
PyObject *__pyx_t_3 = NULL;
PyArrayObject *__pyx_t_4 = NULL;
PyObject *__pyx_t_5 = NULL;
PyObject *__pyx_t_6 = NULL;
PyArrayObject *__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_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__n_total_samples,&__pyx_n_s__n_total_in_bag,&__pyx_n_s__random_state,0};
__Pyx_RefNannySetupContext("_random_sample_mask");
__pyx_self = __pyx_self;
{
PyObject* values[3] = {0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
switch (PyTuple_GET_SIZE(__pyx_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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__n_total_samples);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__n_total_in_bag);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_random_sample_mask", 1, 3, 3, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__random_state);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_random_sample_mask", 1, 3, 3, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_random_sample_mask") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __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_total_samples = __Pyx_PyInt_AsInt(values[0]); if (unlikely((__pyx_v_n_total_samples == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_n_total_in_bag = __Pyx_PyInt_AsInt(values[1]); if (unlikely((__pyx_v_n_total_in_bag == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_random_state = values[2];
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_random_sample_mask", 1, 3, 3, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._random_sample_mask", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_rand.buf = NULL;
__pyx_bstruct_sample_mask.buf = NULL;
/* "sklearn/tree/_tree.pyx":428
* """
* cdef np.ndarray[np.float64_t, ndim=1, mode="c"] rand = \
* random_state.rand(n_total_samples) # <<<<<<<<<<<<<<
* cdef np.ndarray[BOOL_t, ndim=1, mode="c"] sample_mask = \
* np.zeros((n_total_samples,), dtype=np.int8)
*/
__pyx_t_1 = PyObject_GetAttr(__pyx_v_random_state, __pyx_n_s__rand); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 428; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyInt_FromLong(__pyx_v_n_total_samples); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 428; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 428; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_2);
__Pyx_GIVEREF(__pyx_t_2);
__pyx_t_2 = 0;
__pyx_t_2 = PyObject_Call(__pyx_t_1, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 428; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 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 = 428; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_4 = ((PyArrayObject *)__pyx_t_2);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_rand, (PyObject*)__pyx_t_4, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {
__pyx_v_rand = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_bstruct_rand.buf = NULL;
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 427; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
} else {__pyx_bstride_0_rand = __pyx_bstruct_rand.strides[0];
__pyx_bshape_0_rand = __pyx_bstruct_rand.shape[0];
}
}
__pyx_t_4 = 0;
__pyx_v_rand = ((PyArrayObject *)__pyx_t_2);
__pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":430
* random_state.rand(n_total_samples)
* cdef np.ndarray[BOOL_t, ndim=1, mode="c"] sample_mask = \
* np.zeros((n_total_samples,), dtype=np.int8) # <<<<<<<<<<<<<<
*
* cdef int n_bagged = 0
*/
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_3 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__zeros); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
__pyx_t_2 = PyInt_FromLong(__pyx_v_n_total_samples); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_1 = PyTuple_New(1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__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 = PyTuple_New(1); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_2));
PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_t_1));
__Pyx_GIVEREF(((PyObject *)__pyx_t_1));
__pyx_t_1 = 0;
__pyx_t_1 = PyDict_New(); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
__pyx_t_5 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_6 = PyObject_GetAttr(__pyx_t_5, __pyx_n_s__int8); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (PyDict_SetItem(__pyx_t_1, ((PyObject *)__pyx_n_s__dtype), __pyx_t_6) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
__pyx_t_6 = PyEval_CallObjectWithKeywords(__pyx_t_3, ((PyObject *)__pyx_t_2), ((PyObject *)__pyx_t_1)); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
if (!(likely(((__pyx_t_6) == Py_None) || likely(__Pyx_TypeTest(__pyx_t_6, __pyx_ptype_5numpy_ndarray))))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 430; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_7 = ((PyArrayObject *)__pyx_t_6);
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_sample_mask, (PyObject*)__pyx_t_7, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_BOOL_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {
__pyx_v_sample_mask = ((PyArrayObject *)Py_None); __Pyx_INCREF(Py_None); __pyx_bstruct_sample_mask.buf = NULL;
{__pyx_filename = __pyx_f[0]; __pyx_lineno = 429; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
} else {__pyx_bstride_0_sample_mask = __pyx_bstruct_sample_mask.strides[0];
__pyx_bshape_0_sample_mask = __pyx_bstruct_sample_mask.shape[0];
}
}
__pyx_t_7 = 0;
__pyx_v_sample_mask = ((PyArrayObject *)__pyx_t_6);
__pyx_t_6 = 0;
/* "sklearn/tree/_tree.pyx":432
* np.zeros((n_total_samples,), dtype=np.int8)
*
* cdef int n_bagged = 0 # <<<<<<<<<<<<<<
* cdef int i = 0
* for i in range(n_total_samples):
*/
__pyx_v_n_bagged = 0;
/* "sklearn/tree/_tree.pyx":433
*
* cdef int n_bagged = 0
* cdef int i = 0 # <<<<<<<<<<<<<<
* for i in range(n_total_samples):
* if rand[i] * (n_total_samples - i) < (n_total_in_bag - n_bagged):
*/
__pyx_v_i = 0;
/* "sklearn/tree/_tree.pyx":434
* cdef int n_bagged = 0
* cdef int i = 0
* for i in range(n_total_samples): # <<<<<<<<<<<<<<
* if rand[i] * (n_total_samples - i) < (n_total_in_bag - n_bagged):
* sample_mask[i] = 1
*/
__pyx_t_8 = __pyx_v_n_total_samples;
for (__pyx_t_9 = 0; __pyx_t_9 < __pyx_t_8; __pyx_t_9+=1) {
__pyx_v_i = __pyx_t_9;
/* "sklearn/tree/_tree.pyx":435
* cdef int i = 0
* for i in range(n_total_samples):
* if rand[i] * (n_total_samples - i) < (n_total_in_bag - n_bagged): # <<<<<<<<<<<<<<
* sample_mask[i] = 1
* n_bagged += 1
*/
__pyx_t_10 = __pyx_v_i;
__pyx_t_11 = (((*__Pyx_BufPtrCContig1d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_rand.buf, __pyx_t_10, __pyx_bstride_0_rand)) * (__pyx_v_n_total_samples - __pyx_v_i)) < (__pyx_v_n_total_in_bag - __pyx_v_n_bagged));
if (__pyx_t_11) {
/* "sklearn/tree/_tree.pyx":436
* for i in range(n_total_samples):
* if rand[i] * (n_total_samples - i) < (n_total_in_bag - n_bagged):
* sample_mask[i] = 1 # <<<<<<<<<<<<<<
* n_bagged += 1
*
*/
__pyx_t_12 = __pyx_v_i;
*__Pyx_BufPtrCContig1d(__pyx_t_7sklearn_4tree_5_tree_BOOL_t *, __pyx_bstruct_sample_mask.buf, __pyx_t_12, __pyx_bstride_0_sample_mask) = 1;
/* "sklearn/tree/_tree.pyx":437
* if rand[i] * (n_total_samples - i) < (n_total_in_bag - n_bagged):
* sample_mask[i] = 1
* n_bagged += 1 # <<<<<<<<<<<<<<
*
* return sample_mask.astype(np.bool)
*/
__pyx_v_n_bagged = (__pyx_v_n_bagged + 1);
goto __pyx_L8;
}
__pyx_L8:;
}
/* "sklearn/tree/_tree.pyx":439
* n_bagged += 1
*
* return sample_mask.astype(np.bool) # <<<<<<<<<<<<<<
*
*
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_6 = PyObject_GetAttr(((PyObject *)__pyx_v_sample_mask), __pyx_n_s__astype); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 439; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 439; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__bool); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 439; __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 = 439; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__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 = PyObject_Call(__pyx_t_6, ((PyObject *)__pyx_t_1), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 439; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_1)); __pyx_t_1 = 0;
__pyx_r = __pyx_t_2;
__pyx_t_2 = 0;
goto __pyx_L0;
__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_5);
__Pyx_XDECREF(__pyx_t_6);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_rand);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_sample_mask);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._random_sample_mask", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_rand);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_sample_mask);
__pyx_L2:;
__Pyx_XDECREF((PyObject *)__pyx_v_rand);
__Pyx_XDECREF((PyObject *)__pyx_v_sample_mask);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":442
*
*
* def _apply_tree(np.ndarray[DTYPE_t, ndim=2] X, # <<<<<<<<<<<<<<
* np.ndarray[np.int32_t, ndim=2] children,
* np.ndarray[np.int32_t, ndim=1] feature,
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_1_apply_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_1_apply_tree[] = "Finds the terminal region (=leaf node) for each sample in\n `X` and sets the corresponding element in `out` to its node id.";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_1_apply_tree = {__Pyx_NAMESTR("_apply_tree"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_1_apply_tree, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_1_apply_tree)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_1_apply_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_X = 0;
PyArrayObject *__pyx_v_children = 0;
PyArrayObject *__pyx_v_feature = 0;
PyArrayObject *__pyx_v_threshold = 0;
PyArrayObject *__pyx_v_out = 0;
int __pyx_v_i;
int __pyx_v_n;
int __pyx_v_node_id;
Py_buffer __pyx_bstruct_feature;
Py_ssize_t __pyx_bstride_0_feature = 0;
Py_ssize_t __pyx_bshape_0_feature = 0;
Py_buffer __pyx_bstruct_threshold;
Py_ssize_t __pyx_bstride_0_threshold = 0;
Py_ssize_t __pyx_bshape_0_threshold = 0;
Py_buffer __pyx_bstruct_X;
Py_ssize_t __pyx_bstride_0_X = 0;
Py_ssize_t __pyx_bstride_1_X = 0;
Py_ssize_t __pyx_bshape_0_X = 0;
Py_ssize_t __pyx_bshape_1_X = 0;
Py_buffer __pyx_bstruct_children;
Py_ssize_t __pyx_bstride_0_children = 0;
Py_ssize_t __pyx_bstride_1_children = 0;
Py_ssize_t __pyx_bshape_0_children = 0;
Py_ssize_t __pyx_bshape_1_children = 0;
Py_buffer __pyx_bstruct_out;
Py_ssize_t __pyx_bstride_0_out = 0;
Py_ssize_t __pyx_bshape_0_out = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
int __pyx_t_3;
long __pyx_t_4;
int __pyx_t_5;
int __pyx_t_6;
long __pyx_t_7;
int __pyx_t_8;
int __pyx_t_9;
int __pyx_t_10;
int __pyx_t_11;
__pyx_t_5numpy_int32_t __pyx_t_12;
int __pyx_t_13;
int __pyx_t_14;
long __pyx_t_15;
int __pyx_t_16;
long __pyx_t_17;
int __pyx_t_18;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__X,&__pyx_n_s__children,&__pyx_n_s__feature,&__pyx_n_s__threshold,&__pyx_n_s__out,0};
__Pyx_RefNannySetupContext("_apply_tree");
__pyx_self = __pyx_self;
{
PyObject* values[5] = {0,0,0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
switch (PyTuple_GET_SIZE(__pyx_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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__children);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_apply_tree", 1, 5, 5, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__feature);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_apply_tree", 1, 5, 5, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 3:
values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__threshold);
if (likely(values[3])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_apply_tree", 1, 5, 5, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 4:
values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__out);
if (likely(values[4])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_apply_tree", 1, 5, 5, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_apply_tree") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __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_X = ((PyArrayObject *)values[0]);
__pyx_v_children = ((PyArrayObject *)values[1]);
__pyx_v_feature = ((PyArrayObject *)values[2]);
__pyx_v_threshold = ((PyArrayObject *)values[3]);
__pyx_v_out = ((PyArrayObject *)values[4]);
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_apply_tree", 1, 5, 5, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._apply_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_X.buf = NULL;
__pyx_bstruct_children.buf = NULL;
__pyx_bstruct_feature.buf = NULL;
__pyx_bstruct_threshold.buf = NULL;
__pyx_bstruct_out.buf = NULL;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X), __pyx_ptype_5numpy_ndarray, 1, "X", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_children), __pyx_ptype_5numpy_ndarray, 1, "children", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 443; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_feature), __pyx_ptype_5numpy_ndarray, 1, "feature", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 444; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_threshold), __pyx_ptype_5numpy_ndarray, 1, "threshold", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 445; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_out), __pyx_ptype_5numpy_ndarray, 1, "out", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 446; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X, (PyObject*)__pyx_v_X, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X = __pyx_bstruct_X.strides[0]; __pyx_bstride_1_X = __pyx_bstruct_X.strides[1];
__pyx_bshape_0_X = __pyx_bstruct_X.shape[0]; __pyx_bshape_1_X = __pyx_bstruct_X.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_children, (PyObject*)__pyx_v_children, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_children = __pyx_bstruct_children.strides[0]; __pyx_bstride_1_children = __pyx_bstruct_children.strides[1];
__pyx_bshape_0_children = __pyx_bstruct_children.shape[0]; __pyx_bshape_1_children = __pyx_bstruct_children.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_feature, (PyObject*)__pyx_v_feature, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_feature = __pyx_bstruct_feature.strides[0];
__pyx_bshape_0_feature = __pyx_bstruct_feature.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_threshold, (PyObject*)__pyx_v_threshold, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_threshold = __pyx_bstruct_threshold.strides[0];
__pyx_bshape_0_threshold = __pyx_bstruct_threshold.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_out, (PyObject*)__pyx_v_out, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES| PyBUF_WRITABLE, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_out = __pyx_bstruct_out.strides[0];
__pyx_bshape_0_out = __pyx_bstruct_out.shape[0];
/* "sklearn/tree/_tree.pyx":449
* """Finds the terminal region (=leaf node) for each sample in
* `X` and sets the corresponding element in `out` to its node id."""
* cdef int i = 0 # <<<<<<<<<<<<<<
* cdef int n = X.shape[0]
* cdef int node_id = 0
*/
__pyx_v_i = 0;
/* "sklearn/tree/_tree.pyx":450
* `X` and sets the corresponding element in `out` to its node id."""
* cdef int i = 0
* cdef int n = X.shape[0] # <<<<<<<<<<<<<<
* cdef int node_id = 0
* for i in xrange(n):
*/
__pyx_v_n = (__pyx_v_X->dimensions[0]);
/* "sklearn/tree/_tree.pyx":451
* cdef int i = 0
* cdef int n = X.shape[0]
* cdef int node_id = 0 # <<<<<<<<<<<<<<
* for i in xrange(n):
* node_id = 0
*/
__pyx_v_node_id = 0;
/* "sklearn/tree/_tree.pyx":452
* cdef int n = X.shape[0]
* cdef int node_id = 0
* for i in xrange(n): # <<<<<<<<<<<<<<
* node_id = 0
* # While node_id not a leaf
*/
__pyx_t_1 = __pyx_v_n;
for (__pyx_t_2 = 0; __pyx_t_2 < __pyx_t_1; __pyx_t_2+=1) {
__pyx_v_i = __pyx_t_2;
/* "sklearn/tree/_tree.pyx":453
* cdef int node_id = 0
* for i in xrange(n):
* node_id = 0 # <<<<<<<<<<<<<<
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
*/
__pyx_v_node_id = 0;
/* "sklearn/tree/_tree.pyx":455
* node_id = 0
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1: # <<<<<<<<<<<<<<
* if X[i, feature[node_id]] <= threshold[node_id]:
* node_id = children[node_id, 0]
*/
while (1) {
__pyx_t_3 = __pyx_v_node_id;
__pyx_t_4 = 0;
__pyx_t_5 = ((*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_3, __pyx_bstride_0_children, __pyx_t_4, __pyx_bstride_1_children)) != -1);
if (__pyx_t_5) {
__pyx_t_6 = __pyx_v_node_id;
__pyx_t_7 = 1;
__pyx_t_8 = ((*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_6, __pyx_bstride_0_children, __pyx_t_7, __pyx_bstride_1_children)) != -1);
__pyx_t_9 = __pyx_t_8;
} else {
__pyx_t_9 = __pyx_t_5;
}
if (!__pyx_t_9) break;
/* "sklearn/tree/_tree.pyx":456
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
* if X[i, feature[node_id]] <= threshold[node_id]: # <<<<<<<<<<<<<<
* node_id = children[node_id, 0]
* else:
*/
__pyx_t_10 = __pyx_v_node_id;
__pyx_t_11 = __pyx_v_i;
__pyx_t_12 = (*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_feature.buf, __pyx_t_10, __pyx_bstride_0_feature));
__pyx_t_13 = __pyx_v_node_id;
__pyx_t_9 = ((*__Pyx_BufPtrStrided2d(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_bstruct_X.buf, __pyx_t_11, __pyx_bstride_0_X, __pyx_t_12, __pyx_bstride_1_X)) <= (*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_threshold.buf, __pyx_t_13, __pyx_bstride_0_threshold)));
if (__pyx_t_9) {
/* "sklearn/tree/_tree.pyx":457
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
* if X[i, feature[node_id]] <= threshold[node_id]:
* node_id = children[node_id, 0] # <<<<<<<<<<<<<<
* else:
* node_id = children[node_id, 1]
*/
__pyx_t_14 = __pyx_v_node_id;
__pyx_t_15 = 0;
__pyx_v_node_id = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_14, __pyx_bstride_0_children, __pyx_t_15, __pyx_bstride_1_children));
goto __pyx_L10;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":459
* node_id = children[node_id, 0]
* else:
* node_id = children[node_id, 1] # <<<<<<<<<<<<<<
* out[i] = node_id
*
*/
__pyx_t_16 = __pyx_v_node_id;
__pyx_t_17 = 1;
__pyx_v_node_id = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_16, __pyx_bstride_0_children, __pyx_t_17, __pyx_bstride_1_children));
}
__pyx_L10:;
}
/* "sklearn/tree/_tree.pyx":460
* else:
* node_id = children[node_id, 1]
* out[i] = node_id # <<<<<<<<<<<<<<
*
*
*/
__pyx_t_18 = __pyx_v_i;
*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_out.buf, __pyx_t_18, __pyx_bstride_0_out) = __pyx_v_node_id;
}
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_feature);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_threshold);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_children);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_out);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._apply_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_feature);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_threshold);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_children);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_out);
__pyx_L2:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":463
*
*
* def _predict_tree(np.ndarray[DTYPE_t, ndim=2] X, # <<<<<<<<<<<<<<
* np.ndarray[np.int32_t, ndim=2] children,
* np.ndarray[np.int32_t, ndim=1] feature,
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_2_predict_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_2_predict_tree[] = "Finds the terminal region (=leaf node) values for each sample. ";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_2_predict_tree = {__Pyx_NAMESTR("_predict_tree"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_2_predict_tree, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_2_predict_tree)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_2_predict_tree(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_X = 0;
PyArrayObject *__pyx_v_children = 0;
PyArrayObject *__pyx_v_feature = 0;
PyArrayObject *__pyx_v_threshold = 0;
PyArrayObject *__pyx_v_values = 0;
PyArrayObject *__pyx_v_pred = 0;
int __pyx_v_i;
int __pyx_v_n;
int __pyx_v_node_id;
int __pyx_v_K;
int __pyx_v_k;
Py_buffer __pyx_bstruct_pred;
Py_ssize_t __pyx_bstride_0_pred = 0;
Py_ssize_t __pyx_bstride_1_pred = 0;
Py_ssize_t __pyx_bshape_0_pred = 0;
Py_ssize_t __pyx_bshape_1_pred = 0;
Py_buffer __pyx_bstruct_feature;
Py_ssize_t __pyx_bstride_0_feature = 0;
Py_ssize_t __pyx_bshape_0_feature = 0;
Py_buffer __pyx_bstruct_values;
Py_ssize_t __pyx_bstride_0_values = 0;
Py_ssize_t __pyx_bstride_1_values = 0;
Py_ssize_t __pyx_bshape_0_values = 0;
Py_ssize_t __pyx_bshape_1_values = 0;
Py_buffer __pyx_bstruct_threshold;
Py_ssize_t __pyx_bstride_0_threshold = 0;
Py_ssize_t __pyx_bshape_0_threshold = 0;
Py_buffer __pyx_bstruct_X;
Py_ssize_t __pyx_bstride_0_X = 0;
Py_ssize_t __pyx_bstride_1_X = 0;
Py_ssize_t __pyx_bshape_0_X = 0;
Py_ssize_t __pyx_bshape_1_X = 0;
Py_buffer __pyx_bstruct_children;
Py_ssize_t __pyx_bstride_0_children = 0;
Py_ssize_t __pyx_bstride_1_children = 0;
Py_ssize_t __pyx_bshape_0_children = 0;
Py_ssize_t __pyx_bshape_1_children = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
int __pyx_t_3;
long __pyx_t_4;
int __pyx_t_5;
int __pyx_t_6;
long __pyx_t_7;
int __pyx_t_8;
int __pyx_t_9;
int __pyx_t_10;
int __pyx_t_11;
__pyx_t_5numpy_int32_t __pyx_t_12;
int __pyx_t_13;
int __pyx_t_14;
long __pyx_t_15;
int __pyx_t_16;
long __pyx_t_17;
int __pyx_t_18;
int __pyx_t_19;
int __pyx_t_20;
int __pyx_t_21;
int __pyx_t_22;
int __pyx_t_23;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__X,&__pyx_n_s__children,&__pyx_n_s__feature,&__pyx_n_s__threshold,&__pyx_n_s__values,&__pyx_n_s__pred,0};
__Pyx_RefNannySetupContext("_predict_tree");
__pyx_self = __pyx_self;
{
PyObject* values[6] = {0,0,0,0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
switch (PyTuple_GET_SIZE(__pyx_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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__children);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_tree", 1, 6, 6, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__feature);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_tree", 1, 6, 6, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 3:
values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__threshold);
if (likely(values[3])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_tree", 1, 6, 6, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 4:
values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__values);
if (likely(values[4])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_tree", 1, 6, 6, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 5:
values[5] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__pred);
if (likely(values[5])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_tree", 1, 6, 6, 5); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_predict_tree") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __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_X = ((PyArrayObject *)values[0]);
__pyx_v_children = ((PyArrayObject *)values[1]);
__pyx_v_feature = ((PyArrayObject *)values[2]);
__pyx_v_threshold = ((PyArrayObject *)values[3]);
__pyx_v_values = ((PyArrayObject *)values[4]);
__pyx_v_pred = ((PyArrayObject *)values[5]);
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_predict_tree", 1, 6, 6, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._predict_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_X.buf = NULL;
__pyx_bstruct_children.buf = NULL;
__pyx_bstruct_feature.buf = NULL;
__pyx_bstruct_threshold.buf = NULL;
__pyx_bstruct_values.buf = NULL;
__pyx_bstruct_pred.buf = NULL;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X), __pyx_ptype_5numpy_ndarray, 1, "X", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_children), __pyx_ptype_5numpy_ndarray, 1, "children", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 464; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_feature), __pyx_ptype_5numpy_ndarray, 1, "feature", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 465; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_threshold), __pyx_ptype_5numpy_ndarray, 1, "threshold", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 466; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_values), __pyx_ptype_5numpy_ndarray, 1, "values", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 467; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_pred), __pyx_ptype_5numpy_ndarray, 1, "pred", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 468; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X, (PyObject*)__pyx_v_X, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X = __pyx_bstruct_X.strides[0]; __pyx_bstride_1_X = __pyx_bstruct_X.strides[1];
__pyx_bshape_0_X = __pyx_bstruct_X.shape[0]; __pyx_bshape_1_X = __pyx_bstruct_X.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_children, (PyObject*)__pyx_v_children, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_children = __pyx_bstruct_children.strides[0]; __pyx_bstride_1_children = __pyx_bstruct_children.strides[1];
__pyx_bshape_0_children = __pyx_bstruct_children.shape[0]; __pyx_bshape_1_children = __pyx_bstruct_children.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_feature, (PyObject*)__pyx_v_feature, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_feature = __pyx_bstruct_feature.strides[0];
__pyx_bshape_0_feature = __pyx_bstruct_feature.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_threshold, (PyObject*)__pyx_v_threshold, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_threshold = __pyx_bstruct_threshold.strides[0];
__pyx_bshape_0_threshold = __pyx_bstruct_threshold.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_values, (PyObject*)__pyx_v_values, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_values = __pyx_bstruct_values.strides[0]; __pyx_bstride_1_values = __pyx_bstruct_values.strides[1];
__pyx_bshape_0_values = __pyx_bstruct_values.shape[0]; __pyx_bshape_1_values = __pyx_bstruct_values.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_pred, (PyObject*)__pyx_v_pred, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES| PyBUF_WRITABLE, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_pred = __pyx_bstruct_pred.strides[0]; __pyx_bstride_1_pred = __pyx_bstruct_pred.strides[1];
__pyx_bshape_0_pred = __pyx_bstruct_pred.shape[0]; __pyx_bshape_1_pred = __pyx_bstruct_pred.shape[1];
/* "sklearn/tree/_tree.pyx":470
* np.ndarray[np.float64_t, ndim=2] pred):
* """Finds the terminal region (=leaf node) values for each sample. """
* cdef int i = 0 # <<<<<<<<<<<<<<
* cdef int n = X.shape[0]
* cdef int node_id = 0
*/
__pyx_v_i = 0;
/* "sklearn/tree/_tree.pyx":471
* """Finds the terminal region (=leaf node) values for each sample. """
* cdef int i = 0
* cdef int n = X.shape[0] # <<<<<<<<<<<<<<
* cdef int node_id = 0
* cdef int K = values.shape[1]
*/
__pyx_v_n = (__pyx_v_X->dimensions[0]);
/* "sklearn/tree/_tree.pyx":472
* cdef int i = 0
* cdef int n = X.shape[0]
* cdef int node_id = 0 # <<<<<<<<<<<<<<
* cdef int K = values.shape[1]
* for i in xrange(n):
*/
__pyx_v_node_id = 0;
/* "sklearn/tree/_tree.pyx":473
* cdef int n = X.shape[0]
* cdef int node_id = 0
* cdef int K = values.shape[1] # <<<<<<<<<<<<<<
* for i in xrange(n):
* node_id = 0
*/
__pyx_v_K = (__pyx_v_values->dimensions[1]);
/* "sklearn/tree/_tree.pyx":474
* cdef int node_id = 0
* cdef int K = values.shape[1]
* for i in xrange(n): # <<<<<<<<<<<<<<
* node_id = 0
* # While node_id not a leaf
*/
__pyx_t_1 = __pyx_v_n;
for (__pyx_t_2 = 0; __pyx_t_2 < __pyx_t_1; __pyx_t_2+=1) {
__pyx_v_i = __pyx_t_2;
/* "sklearn/tree/_tree.pyx":475
* cdef int K = values.shape[1]
* for i in xrange(n):
* node_id = 0 # <<<<<<<<<<<<<<
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
*/
__pyx_v_node_id = 0;
/* "sklearn/tree/_tree.pyx":477
* node_id = 0
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1: # <<<<<<<<<<<<<<
* if X[i, feature[node_id]] <= threshold[node_id]:
* node_id = children[node_id, 0]
*/
while (1) {
__pyx_t_3 = __pyx_v_node_id;
__pyx_t_4 = 0;
__pyx_t_5 = ((*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_3, __pyx_bstride_0_children, __pyx_t_4, __pyx_bstride_1_children)) != -1);
if (__pyx_t_5) {
__pyx_t_6 = __pyx_v_node_id;
__pyx_t_7 = 1;
__pyx_t_8 = ((*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_6, __pyx_bstride_0_children, __pyx_t_7, __pyx_bstride_1_children)) != -1);
__pyx_t_9 = __pyx_t_8;
} else {
__pyx_t_9 = __pyx_t_5;
}
if (!__pyx_t_9) break;
/* "sklearn/tree/_tree.pyx":478
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
* if X[i, feature[node_id]] <= threshold[node_id]: # <<<<<<<<<<<<<<
* node_id = children[node_id, 0]
* else:
*/
__pyx_t_10 = __pyx_v_node_id;
__pyx_t_11 = __pyx_v_i;
__pyx_t_12 = (*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_feature.buf, __pyx_t_10, __pyx_bstride_0_feature));
__pyx_t_13 = __pyx_v_node_id;
__pyx_t_9 = ((*__Pyx_BufPtrStrided2d(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_bstruct_X.buf, __pyx_t_11, __pyx_bstride_0_X, __pyx_t_12, __pyx_bstride_1_X)) <= (*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_threshold.buf, __pyx_t_13, __pyx_bstride_0_threshold)));
if (__pyx_t_9) {
/* "sklearn/tree/_tree.pyx":479
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
* if X[i, feature[node_id]] <= threshold[node_id]:
* node_id = children[node_id, 0] # <<<<<<<<<<<<<<
* else:
* node_id = children[node_id, 1]
*/
__pyx_t_14 = __pyx_v_node_id;
__pyx_t_15 = 0;
__pyx_v_node_id = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_14, __pyx_bstride_0_children, __pyx_t_15, __pyx_bstride_1_children));
goto __pyx_L10;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":481
* node_id = children[node_id, 0]
* else:
* node_id = children[node_id, 1] # <<<<<<<<<<<<<<
* for k in xrange(K):
* pred[i, k] = values[node_id, k]
*/
__pyx_t_16 = __pyx_v_node_id;
__pyx_t_17 = 1;
__pyx_v_node_id = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_16, __pyx_bstride_0_children, __pyx_t_17, __pyx_bstride_1_children));
}
__pyx_L10:;
}
/* "sklearn/tree/_tree.pyx":482
* else:
* node_id = children[node_id, 1]
* for k in xrange(K): # <<<<<<<<<<<<<<
* pred[i, k] = values[node_id, k]
*
*/
__pyx_t_18 = __pyx_v_K;
for (__pyx_t_19 = 0; __pyx_t_19 < __pyx_t_18; __pyx_t_19+=1) {
__pyx_v_k = __pyx_t_19;
/* "sklearn/tree/_tree.pyx":483
* node_id = children[node_id, 1]
* for k in xrange(K):
* pred[i, k] = values[node_id, k] # <<<<<<<<<<<<<<
*
*
*/
__pyx_t_20 = __pyx_v_node_id;
__pyx_t_21 = __pyx_v_k;
__pyx_t_22 = __pyx_v_i;
__pyx_t_23 = __pyx_v_k;
*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_pred.buf, __pyx_t_22, __pyx_bstride_0_pred, __pyx_t_23, __pyx_bstride_1_pred) = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_values.buf, __pyx_t_20, __pyx_bstride_0_values, __pyx_t_21, __pyx_bstride_1_values));
}
}
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_pred);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_feature);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_values);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_threshold);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_children);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._predict_tree", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_pred);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_feature);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_values);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_threshold);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_children);
__pyx_L2:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":486
*
*
* def _predict_regression_tree_inplace(np.ndarray[DTYPE_t, ndim=2] X, # <<<<<<<<<<<<<<
* np.ndarray[np.int32_t, ndim=2] children,
* np.ndarray[np.int32_t, ndim=1] feature,
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace[] = "Predicts output for regression tree and stores it in ``pred[i, k]`` ";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace = {__Pyx_NAMESTR("_predict_regression_tree_inplace"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_X = 0;
PyArrayObject *__pyx_v_children = 0;
PyArrayObject *__pyx_v_feature = 0;
PyArrayObject *__pyx_v_threshold = 0;
PyArrayObject *__pyx_v_values = 0;
double __pyx_v_scale;
int __pyx_v_k;
PyArrayObject *__pyx_v_pred = 0;
int __pyx_v_i;
int __pyx_v_n;
int __pyx_v_node_id;
int __pyx_v_K;
Py_buffer __pyx_bstruct_pred;
Py_ssize_t __pyx_bstride_0_pred = 0;
Py_ssize_t __pyx_bstride_1_pred = 0;
Py_ssize_t __pyx_bshape_0_pred = 0;
Py_ssize_t __pyx_bshape_1_pred = 0;
Py_buffer __pyx_bstruct_feature;
Py_ssize_t __pyx_bstride_0_feature = 0;
Py_ssize_t __pyx_bshape_0_feature = 0;
Py_buffer __pyx_bstruct_values;
Py_ssize_t __pyx_bstride_0_values = 0;
Py_ssize_t __pyx_bstride_1_values = 0;
Py_ssize_t __pyx_bshape_0_values = 0;
Py_ssize_t __pyx_bshape_1_values = 0;
Py_buffer __pyx_bstruct_threshold;
Py_ssize_t __pyx_bstride_0_threshold = 0;
Py_ssize_t __pyx_bshape_0_threshold = 0;
Py_buffer __pyx_bstruct_X;
Py_ssize_t __pyx_bstride_0_X = 0;
Py_ssize_t __pyx_bstride_1_X = 0;
Py_ssize_t __pyx_bshape_0_X = 0;
Py_ssize_t __pyx_bshape_1_X = 0;
Py_buffer __pyx_bstruct_children;
Py_ssize_t __pyx_bstride_0_children = 0;
Py_ssize_t __pyx_bstride_1_children = 0;
Py_ssize_t __pyx_bshape_0_children = 0;
Py_ssize_t __pyx_bshape_1_children = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
int __pyx_t_3;
long __pyx_t_4;
int __pyx_t_5;
int __pyx_t_6;
long __pyx_t_7;
int __pyx_t_8;
int __pyx_t_9;
int __pyx_t_10;
int __pyx_t_11;
__pyx_t_5numpy_int32_t __pyx_t_12;
int __pyx_t_13;
int __pyx_t_14;
long __pyx_t_15;
int __pyx_t_16;
long __pyx_t_17;
int __pyx_t_18;
long __pyx_t_19;
int __pyx_t_20;
int __pyx_t_21;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__X,&__pyx_n_s__children,&__pyx_n_s__feature,&__pyx_n_s__threshold,&__pyx_n_s__values,&__pyx_n_s__scale,&__pyx_n_s__k,&__pyx_n_s__pred,0};
__Pyx_RefNannySetupContext("_predict_regression_tree_inplace");
__pyx_self = __pyx_self;
{
PyObject* values[8] = {0,0,0,0,0,0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__children);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__feature);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 3:
values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__threshold);
if (likely(values[3])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 4:
values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__values);
if (likely(values[4])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 5:
values[5] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__scale);
if (likely(values[5])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 5); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 6:
values[6] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__k);
if (likely(values[6])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 6); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 7:
values[7] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__pred);
if (likely(values[7])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, 7); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_predict_regression_tree_inplace") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
} else if (PyTuple_GET_SIZE(__pyx_args) != 8) {
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);
values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
}
__pyx_v_X = ((PyArrayObject *)values[0]);
__pyx_v_children = ((PyArrayObject *)values[1]);
__pyx_v_feature = ((PyArrayObject *)values[2]);
__pyx_v_threshold = ((PyArrayObject *)values[3]);
__pyx_v_values = ((PyArrayObject *)values[4]);
__pyx_v_scale = __pyx_PyFloat_AsDouble(values[5]); if (unlikely((__pyx_v_scale == (double)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 491; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_k = __Pyx_PyInt_AsInt(values[6]); if (unlikely((__pyx_v_k == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 492; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_pred = ((PyArrayObject *)values[7]);
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_predict_regression_tree_inplace", 1, 8, 8, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._predict_regression_tree_inplace", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_X.buf = NULL;
__pyx_bstruct_children.buf = NULL;
__pyx_bstruct_feature.buf = NULL;
__pyx_bstruct_threshold.buf = NULL;
__pyx_bstruct_values.buf = NULL;
__pyx_bstruct_pred.buf = NULL;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X), __pyx_ptype_5numpy_ndarray, 1, "X", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_children), __pyx_ptype_5numpy_ndarray, 1, "children", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 487; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_feature), __pyx_ptype_5numpy_ndarray, 1, "feature", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 488; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_threshold), __pyx_ptype_5numpy_ndarray, 1, "threshold", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 489; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_values), __pyx_ptype_5numpy_ndarray, 1, "values", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 490; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_pred), __pyx_ptype_5numpy_ndarray, 1, "pred", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 493; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X, (PyObject*)__pyx_v_X, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X = __pyx_bstruct_X.strides[0]; __pyx_bstride_1_X = __pyx_bstruct_X.strides[1];
__pyx_bshape_0_X = __pyx_bstruct_X.shape[0]; __pyx_bshape_1_X = __pyx_bstruct_X.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_children, (PyObject*)__pyx_v_children, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_children = __pyx_bstruct_children.strides[0]; __pyx_bstride_1_children = __pyx_bstruct_children.strides[1];
__pyx_bshape_0_children = __pyx_bstruct_children.shape[0]; __pyx_bshape_1_children = __pyx_bstruct_children.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_feature, (PyObject*)__pyx_v_feature, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_feature = __pyx_bstruct_feature.strides[0];
__pyx_bshape_0_feature = __pyx_bstruct_feature.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_threshold, (PyObject*)__pyx_v_threshold, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_threshold = __pyx_bstruct_threshold.strides[0];
__pyx_bshape_0_threshold = __pyx_bstruct_threshold.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_values, (PyObject*)__pyx_v_values, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_values = __pyx_bstruct_values.strides[0]; __pyx_bstride_1_values = __pyx_bstruct_values.strides[1];
__pyx_bshape_0_values = __pyx_bstruct_values.shape[0]; __pyx_bshape_1_values = __pyx_bstruct_values.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_pred, (PyObject*)__pyx_v_pred, &__Pyx_TypeInfo_nn___pyx_t_5numpy_float64_t, PyBUF_FORMAT| PyBUF_STRIDES| PyBUF_WRITABLE, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_pred = __pyx_bstruct_pred.strides[0]; __pyx_bstride_1_pred = __pyx_bstruct_pred.strides[1];
__pyx_bshape_0_pred = __pyx_bstruct_pred.shape[0]; __pyx_bshape_1_pred = __pyx_bstruct_pred.shape[1];
/* "sklearn/tree/_tree.pyx":495
* np.ndarray[np.float64_t, ndim=2] pred):
* """Predicts output for regression tree and stores it in ``pred[i, k]`` """
* cdef int i = 0 # <<<<<<<<<<<<<<
* cdef int n = X.shape[0]
* cdef int node_id = 0
*/
__pyx_v_i = 0;
/* "sklearn/tree/_tree.pyx":496
* """Predicts output for regression tree and stores it in ``pred[i, k]`` """
* cdef int i = 0
* cdef int n = X.shape[0] # <<<<<<<<<<<<<<
* cdef int node_id = 0
* cdef int K = values.shape[1]
*/
__pyx_v_n = (__pyx_v_X->dimensions[0]);
/* "sklearn/tree/_tree.pyx":497
* cdef int i = 0
* cdef int n = X.shape[0]
* cdef int node_id = 0 # <<<<<<<<<<<<<<
* cdef int K = values.shape[1]
* for i in xrange(n):
*/
__pyx_v_node_id = 0;
/* "sklearn/tree/_tree.pyx":498
* cdef int n = X.shape[0]
* cdef int node_id = 0
* cdef int K = values.shape[1] # <<<<<<<<<<<<<<
* for i in xrange(n):
* node_id = 0
*/
__pyx_v_K = (__pyx_v_values->dimensions[1]);
/* "sklearn/tree/_tree.pyx":499
* cdef int node_id = 0
* cdef int K = values.shape[1]
* for i in xrange(n): # <<<<<<<<<<<<<<
* node_id = 0
* # While node_id not a leaf
*/
__pyx_t_1 = __pyx_v_n;
for (__pyx_t_2 = 0; __pyx_t_2 < __pyx_t_1; __pyx_t_2+=1) {
__pyx_v_i = __pyx_t_2;
/* "sklearn/tree/_tree.pyx":500
* cdef int K = values.shape[1]
* for i in xrange(n):
* node_id = 0 # <<<<<<<<<<<<<<
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
*/
__pyx_v_node_id = 0;
/* "sklearn/tree/_tree.pyx":502
* node_id = 0
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1: # <<<<<<<<<<<<<<
* if X[i, feature[node_id]] <= threshold[node_id]:
* node_id = children[node_id, 0]
*/
while (1) {
__pyx_t_3 = __pyx_v_node_id;
__pyx_t_4 = 0;
__pyx_t_5 = ((*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_3, __pyx_bstride_0_children, __pyx_t_4, __pyx_bstride_1_children)) != -1);
if (__pyx_t_5) {
__pyx_t_6 = __pyx_v_node_id;
__pyx_t_7 = 1;
__pyx_t_8 = ((*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_6, __pyx_bstride_0_children, __pyx_t_7, __pyx_bstride_1_children)) != -1);
__pyx_t_9 = __pyx_t_8;
} else {
__pyx_t_9 = __pyx_t_5;
}
if (!__pyx_t_9) break;
/* "sklearn/tree/_tree.pyx":503
* # While node_id not a leaf
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
* if X[i, feature[node_id]] <= threshold[node_id]: # <<<<<<<<<<<<<<
* node_id = children[node_id, 0]
* else:
*/
__pyx_t_10 = __pyx_v_node_id;
__pyx_t_11 = __pyx_v_i;
__pyx_t_12 = (*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_feature.buf, __pyx_t_10, __pyx_bstride_0_feature));
__pyx_t_13 = __pyx_v_node_id;
__pyx_t_9 = ((*__Pyx_BufPtrStrided2d(__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, __pyx_bstruct_X.buf, __pyx_t_11, __pyx_bstride_0_X, __pyx_t_12, __pyx_bstride_1_X)) <= (*__Pyx_BufPtrStrided1d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_threshold.buf, __pyx_t_13, __pyx_bstride_0_threshold)));
if (__pyx_t_9) {
/* "sklearn/tree/_tree.pyx":504
* while children[node_id, 0] != -1 and children[node_id, 1] != -1:
* if X[i, feature[node_id]] <= threshold[node_id]:
* node_id = children[node_id, 0] # <<<<<<<<<<<<<<
* else:
* node_id = children[node_id, 1]
*/
__pyx_t_14 = __pyx_v_node_id;
__pyx_t_15 = 0;
__pyx_v_node_id = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_14, __pyx_bstride_0_children, __pyx_t_15, __pyx_bstride_1_children));
goto __pyx_L10;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":506
* node_id = children[node_id, 0]
* else:
* node_id = children[node_id, 1] # <<<<<<<<<<<<<<
* pred[i, k] += scale * values[node_id, 0]
*
*/
__pyx_t_16 = __pyx_v_node_id;
__pyx_t_17 = 1;
__pyx_v_node_id = (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_int32_t *, __pyx_bstruct_children.buf, __pyx_t_16, __pyx_bstride_0_children, __pyx_t_17, __pyx_bstride_1_children));
}
__pyx_L10:;
}
/* "sklearn/tree/_tree.pyx":507
* else:
* node_id = children[node_id, 1]
* pred[i, k] += scale * values[node_id, 0] # <<<<<<<<<<<<<<
*
*
*/
__pyx_t_18 = __pyx_v_node_id;
__pyx_t_19 = 0;
__pyx_t_20 = __pyx_v_i;
__pyx_t_21 = __pyx_v_k;
*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_pred.buf, __pyx_t_20, __pyx_bstride_0_pred, __pyx_t_21, __pyx_bstride_1_pred) += (__pyx_v_scale * (*__Pyx_BufPtrStrided2d(__pyx_t_5numpy_float64_t *, __pyx_bstruct_values.buf, __pyx_t_18, __pyx_bstride_0_values, __pyx_t_19, __pyx_bstride_1_values)));
}
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_pred);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_feature);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_values);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_threshold);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_children);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._predict_regression_tree_inplace", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_pred);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_feature);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_values);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_threshold);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_children);
__pyx_L2:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":510
*
*
* def _error_at_leaf(np.ndarray[DTYPE_t, ndim=1, mode="c"] y, # <<<<<<<<<<<<<<
* np.ndarray sample_mask, Criterion criterion,
* int n_samples):
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4_error_at_leaf(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_4_error_at_leaf[] = "Compute criterion error at leaf with terminal region defined\n by `sample_mask`. ";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_4_error_at_leaf = {__Pyx_NAMESTR("_error_at_leaf"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_4_error_at_leaf, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_4_error_at_leaf)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_4_error_at_leaf(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_y = 0;
PyArrayObject *__pyx_v_sample_mask = 0;
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
int __pyx_v_n_samples;
int __pyx_v_n_total_samples;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y_ptr;
__pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask_ptr;
Py_buffer __pyx_bstruct_y;
Py_ssize_t __pyx_bstride_0_y = 0;
Py_ssize_t __pyx_bshape_0_y = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__y,&__pyx_n_s__sample_mask,&__pyx_n_s__criterion,&__pyx_n_s__n_samples,0};
__Pyx_RefNannySetupContext("_error_at_leaf");
__pyx_self = __pyx_self;
{
PyObject* values[4] = {0,0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
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);
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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__y);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__sample_mask);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_error_at_leaf", 1, 4, 4, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__criterion);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_error_at_leaf", 1, 4, 4, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 3:
values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__n_samples);
if (likely(values[3])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_error_at_leaf", 1, 4, 4, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_error_at_leaf") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
} else if (PyTuple_GET_SIZE(__pyx_args) != 4) {
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);
}
__pyx_v_y = ((PyArrayObject *)values[0]);
__pyx_v_sample_mask = ((PyArrayObject *)values[1]);
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[2]);
__pyx_v_n_samples = __Pyx_PyInt_AsInt(values[3]); if (unlikely((__pyx_v_n_samples == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 512; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_error_at_leaf", 1, 4, 4, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._error_at_leaf", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_y.buf = NULL;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_y), __pyx_ptype_5numpy_ndarray, 1, "y", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_sample_mask), __pyx_ptype_5numpy_ndarray, 1, "sample_mask", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 511; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 511; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_y, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_y = __pyx_bstruct_y.strides[0];
__pyx_bshape_0_y = __pyx_bstruct_y.shape[0];
/* "sklearn/tree/_tree.pyx":515
* """Compute criterion error at leaf with terminal region defined
* by `sample_mask`. """
* cdef int n_total_samples = y.shape[0] # <<<<<<<<<<<<<<
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
*/
__pyx_v_n_total_samples = (__pyx_v_y->dimensions[0]);
/* "sklearn/tree/_tree.pyx":516
* by `sample_mask`. """
* cdef int n_total_samples = y.shape[0]
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data # <<<<<<<<<<<<<<
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
*/
__pyx_v_y_ptr = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_y->data);
/* "sklearn/tree/_tree.pyx":517
* cdef int n_total_samples = y.shape[0]
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data # <<<<<<<<<<<<<<
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
* return criterion.eval()
*/
__pyx_v_sample_mask_ptr = ((__pyx_t_7sklearn_4tree_5_tree_BOOL_t *)__pyx_v_sample_mask->data);
/* "sklearn/tree/_tree.pyx":518
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples) # <<<<<<<<<<<<<<
* return criterion.eval()
*
*/
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->init(__pyx_v_criterion, __pyx_v_y_ptr, __pyx_v_sample_mask_ptr, __pyx_v_n_samples, __pyx_v_n_total_samples);
/* "sklearn/tree/_tree.pyx":519
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
* return criterion.eval() # <<<<<<<<<<<<<<
*
*
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = PyFloat_FromDouble(((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->eval(__pyx_v_criterion)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 519; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
{ PyObject *__pyx_type, *__pyx_value, *__pyx_tb;
__Pyx_ErrFetch(&__pyx_type, &__pyx_value, &__pyx_tb);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_y);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._error_at_leaf", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_y);
__pyx_L2:;
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":522
*
*
* cdef int smallest_sample_larger_than(int sample_idx, DTYPE_t *X_i, # <<<<<<<<<<<<<<
* int *X_argsorted_i, BOOL_t *sample_mask,
* int n_total_samples):
*/
static int __pyx_f_7sklearn_4tree_5_tree_smallest_sample_larger_than(int __pyx_v_sample_idx, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_i, int *__pyx_v_X_argsorted_i, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask, int __pyx_v_n_total_samples) {
int __pyx_v_idx;
int __pyx_v_j;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_threshold;
int __pyx_r;
__Pyx_RefNannyDeclarations
int __pyx_t_1;
int __pyx_t_2;
__Pyx_RefNannySetupContext("smallest_sample_larger_than");
/* "sklearn/tree/_tree.pyx":539
* -1 if no such element exists.
* """
* cdef int idx = 0, j # <<<<<<<<<<<<<<
* cdef DTYPE_t threshold = -DBL_MAX
*
*/
__pyx_v_idx = 0;
/* "sklearn/tree/_tree.pyx":540
* """
* cdef int idx = 0, j
* cdef DTYPE_t threshold = -DBL_MAX # <<<<<<<<<<<<<<
*
* if sample_idx > -1:
*/
__pyx_v_threshold = (-DBL_MAX);
/* "sklearn/tree/_tree.pyx":542
* cdef DTYPE_t threshold = -DBL_MAX
*
* if sample_idx > -1: # <<<<<<<<<<<<<<
* threshold = X_i[X_argsorted_i[sample_idx]]
*
*/
__pyx_t_1 = (__pyx_v_sample_idx > -1);
if (__pyx_t_1) {
/* "sklearn/tree/_tree.pyx":543
*
* if sample_idx > -1:
* threshold = X_i[X_argsorted_i[sample_idx]] # <<<<<<<<<<<<<<
*
* for idx from sample_idx < idx < n_total_samples:
*/
__pyx_v_threshold = (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_sample_idx])]);
goto __pyx_L3;
}
__pyx_L3:;
/* "sklearn/tree/_tree.pyx":545
* threshold = X_i[X_argsorted_i[sample_idx]]
*
* for idx from sample_idx < idx < n_total_samples: # <<<<<<<<<<<<<<
* j = X_argsorted_i[idx]
*
*/
__pyx_t_2 = __pyx_v_n_total_samples;
for (__pyx_v_idx = __pyx_v_sample_idx+1; __pyx_v_idx < __pyx_t_2; __pyx_v_idx++) {
/* "sklearn/tree/_tree.pyx":546
*
* for idx from sample_idx < idx < n_total_samples:
* j = X_argsorted_i[idx] # <<<<<<<<<<<<<<
*
* if sample_mask[j] == 0:
*/
__pyx_v_j = (__pyx_v_X_argsorted_i[__pyx_v_idx]);
/* "sklearn/tree/_tree.pyx":548
* j = X_argsorted_i[idx]
*
* if sample_mask[j] == 0: # <<<<<<<<<<<<<<
* continue
*
*/
__pyx_t_1 = ((__pyx_v_sample_mask[__pyx_v_j]) == 0);
if (__pyx_t_1) {
/* "sklearn/tree/_tree.pyx":549
*
* if sample_mask[j] == 0:
* continue # <<<<<<<<<<<<<<
*
* if X_i[j] > threshold + 1.e-7:
*/
goto __pyx_L4_continue;
goto __pyx_L6;
}
__pyx_L6:;
/* "sklearn/tree/_tree.pyx":551
* continue
*
* if X_i[j] > threshold + 1.e-7: # <<<<<<<<<<<<<<
* return idx
*
*/
__pyx_t_1 = ((__pyx_v_X_i[__pyx_v_j]) > (__pyx_v_threshold + 1.e-7));
if (__pyx_t_1) {
/* "sklearn/tree/_tree.pyx":552
*
* if X_i[j] > threshold + 1.e-7:
* return idx # <<<<<<<<<<<<<<
*
* return -1
*/
__pyx_r = __pyx_v_idx;
goto __pyx_L0;
goto __pyx_L7;
}
__pyx_L7:;
__pyx_L4_continue:;
}
/* "sklearn/tree/_tree.pyx":554
* return idx
*
* return -1 # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = -1;
goto __pyx_L0;
__pyx_r = 0;
__pyx_L0:;
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":557
*
*
* def _find_best_split(np.ndarray[DTYPE_t, ndim=2, mode="fortran"] X, # <<<<<<<<<<<<<<
* np.ndarray[DTYPE_t, ndim=1, mode="c"] y,
* np.ndarray[np.int32_t, ndim=2, mode="fortran"] X_argsorted,
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_5_find_best_split(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_5_find_best_split[] = "Find the best dimension and threshold that minimises the error.\n\n Parameters\n ----------\n X : ndarray, shape (n_total_samples, n_features), dtype=DTYPE_t\n The feature values.\n\n y : ndarray, shape (n_total_samples,), dtype=float\n The label to predict for each sample.\n\n X_argsorted : ndarray, shape (n_samples, n_features)\n Argsort of cols of `X`. `X_argsorted[0,j]` gives the example\n index of the smallest value of feature `j`.\n\n sample_mask : ndarray, shape (n_samples,), dtype=np.bool\n A mask for the samples to be considered. Only samples `j` for which\n sample_mask[j] != 0 are considered.\n\n n_samples : int\n The number of samples in the current sample_mask\n (i.e. `sample_mask.sum()`).\n\n min_leaf : int\n The minimum number of samples required to be at a leaf node.\n\n max_features : int\n The number of features to consider when looking for the best split.\n\n criterion : Criterion\n The criterion function to be minimized.\n\n random_state : RandomState\n The numpy random state to use.\n\n Returns\n -------\n best_i : int\n The split feature or -1 if criterion not smaller than\n `parent_split_error`.\n\n best_t : DTYPE_t\n The split threshold\n\n best_error : DTYPE_t\n The split error\n\n initial_error : DTYPE_t\n The initial error contained in the node.\n ";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_5_find_best_split = {__Pyx_NAMESTR("_find_best_split"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_5_find_best_split, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_5_find_best_split)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_5_find_best_split(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_X = 0;
PyArrayObject *__pyx_v_y = 0;
PyArrayObject *__pyx_v_X_argsorted = 0;
PyArrayObject *__pyx_v_sample_mask = 0;
int __pyx_v_n_samples;
int __pyx_v_min_leaf;
int __pyx_v_max_features;
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
PyObject *__pyx_v_random_state = 0;
int __pyx_v_n_total_samples;
int __pyx_v_n_features;
int __pyx_v_i;
int __pyx_v_a;
int __pyx_v_b;
int __pyx_v_best_i;
int __pyx_v_n_left;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_t;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_initial_error;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_error;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_best_error;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_best_t;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y_ptr;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_i;
int *__pyx_v_X_argsorted_i;
__pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask_ptr;
int __pyx_v_X_elem_stride;
int __pyx_v_X_col_stride;
int __pyx_v_X_stride;
int __pyx_v_X_argsorted_elem_stride;
int __pyx_v_X_argsorted_col_stride;
int __pyx_v_X_argsorted_stride;
PyObject *__pyx_v_features = NULL;
Py_buffer __pyx_bstruct_X;
Py_ssize_t __pyx_bstride_0_X = 0;
Py_ssize_t __pyx_bstride_1_X = 0;
Py_ssize_t __pyx_bshape_0_X = 0;
Py_ssize_t __pyx_bshape_1_X = 0;
Py_buffer __pyx_bstruct_X_argsorted;
Py_ssize_t __pyx_bstride_0_X_argsorted = 0;
Py_ssize_t __pyx_bstride_1_X_argsorted = 0;
Py_ssize_t __pyx_bshape_0_X_argsorted = 0;
Py_ssize_t __pyx_bshape_1_X_argsorted = 0;
Py_buffer __pyx_bstruct_y;
Py_ssize_t __pyx_bstride_0_y = 0;
Py_ssize_t __pyx_bshape_0_y = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
PyObject *__pyx_t_2 = NULL;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __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;
Py_ssize_t __pyx_t_10;
PyObject *(*__pyx_t_11)(PyObject *);
int __pyx_t_12;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__X,&__pyx_n_s__y,&__pyx_n_s__X_argsorted,&__pyx_n_s__sample_mask,&__pyx_n_s__n_samples,&__pyx_n_s__min_leaf,&__pyx_n_s__max_features,&__pyx_n_s__criterion,&__pyx_n_s__random_state,0};
__Pyx_RefNannySetupContext("_find_best_split");
__pyx_self = __pyx_self;
{
PyObject* values[9] = {0,0,0,0,0,0,0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__y);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X_argsorted);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 3:
values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__sample_mask);
if (likely(values[3])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 4:
values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__n_samples);
if (likely(values[4])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 5:
values[5] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__min_leaf);
if (likely(values[5])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 5); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 6:
values[6] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__max_features);
if (likely(values[6])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 6); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 7:
values[7] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__criterion);
if (likely(values[7])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 7); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 8:
values[8] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__random_state);
if (likely(values[8])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, 8); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_find_best_split") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
} else if (PyTuple_GET_SIZE(__pyx_args) != 9) {
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);
values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
}
__pyx_v_X = ((PyArrayObject *)values[0]);
__pyx_v_y = ((PyArrayObject *)values[1]);
__pyx_v_X_argsorted = ((PyArrayObject *)values[2]);
__pyx_v_sample_mask = ((PyArrayObject *)values[3]);
__pyx_v_n_samples = __Pyx_PyInt_AsInt(values[4]); if (unlikely((__pyx_v_n_samples == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 561; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_min_leaf = __Pyx_PyInt_AsInt(values[5]); if (unlikely((__pyx_v_min_leaf == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 562; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_max_features = __Pyx_PyInt_AsInt(values[6]); if (unlikely((__pyx_v_max_features == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 563; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[7]);
__pyx_v_random_state = values[8];
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_find_best_split", 1, 9, 9, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._find_best_split", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_X.buf = NULL;
__pyx_bstruct_y.buf = NULL;
__pyx_bstruct_X_argsorted.buf = NULL;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X), __pyx_ptype_5numpy_ndarray, 1, "X", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __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 = 558; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X_argsorted), __pyx_ptype_5numpy_ndarray, 1, "X_argsorted", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 559; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_sample_mask), __pyx_ptype_5numpy_ndarray, 1, "sample_mask", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 560; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 564; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X, (PyObject*)__pyx_v_X, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_F_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X = __pyx_bstruct_X.strides[0]; __pyx_bstride_1_X = __pyx_bstruct_X.strides[1];
__pyx_bshape_0_X = __pyx_bstruct_X.shape[0]; __pyx_bshape_1_X = __pyx_bstruct_X.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_y, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_y = __pyx_bstruct_y.strides[0];
__pyx_bshape_0_y = __pyx_bstruct_y.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X_argsorted, (PyObject*)__pyx_v_X_argsorted, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_F_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X_argsorted = __pyx_bstruct_X_argsorted.strides[0]; __pyx_bstride_1_X_argsorted = __pyx_bstruct_X_argsorted.strides[1];
__pyx_bshape_0_X_argsorted = __pyx_bstruct_X_argsorted.shape[0]; __pyx_bshape_1_X_argsorted = __pyx_bstruct_X_argsorted.shape[1];
/* "sklearn/tree/_tree.pyx":616
* """
* # Variables declarations
* cdef int n_total_samples = X.shape[0] # <<<<<<<<<<<<<<
* cdef int n_features = X.shape[1]
* cdef int i, a, b, best_i = -1
*/
__pyx_v_n_total_samples = (__pyx_v_X->dimensions[0]);
/* "sklearn/tree/_tree.pyx":617
* # Variables declarations
* cdef int n_total_samples = X.shape[0]
* cdef int n_features = X.shape[1] # <<<<<<<<<<<<<<
* cdef int i, a, b, best_i = -1
* cdef int n_left = 0
*/
__pyx_v_n_features = (__pyx_v_X->dimensions[1]);
/* "sklearn/tree/_tree.pyx":618
* cdef int n_total_samples = X.shape[0]
* cdef int n_features = X.shape[1]
* cdef int i, a, b, best_i = -1 # <<<<<<<<<<<<<<
* cdef int n_left = 0
* cdef DTYPE_t t, initial_error, error
*/
__pyx_v_best_i = -1;
/* "sklearn/tree/_tree.pyx":619
* cdef int n_features = X.shape[1]
* cdef int i, a, b, best_i = -1
* cdef int n_left = 0 # <<<<<<<<<<<<<<
* cdef DTYPE_t t, initial_error, error
* cdef DTYPE_t best_error = np.inf, best_t = np.inf
*/
__pyx_v_n_left = 0;
/* "sklearn/tree/_tree.pyx":621
* cdef int n_left = 0
* cdef DTYPE_t t, initial_error, error
* cdef DTYPE_t best_error = np.inf, best_t = np.inf # <<<<<<<<<<<<<<
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef DTYPE_t *X_i = NULL
*/
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 621; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__inf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 621; __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_PyFloat_AsDouble(__pyx_t_2); if (unlikely((__pyx_t_3 == (npy_float32)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 621; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
__pyx_v_best_error = __pyx_t_3;
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 621; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__inf); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 621; __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 == (npy_float32)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 621; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_v_best_t = __pyx_t_3;
/* "sklearn/tree/_tree.pyx":622
* cdef DTYPE_t t, initial_error, error
* cdef DTYPE_t best_error = np.inf, best_t = np.inf
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data # <<<<<<<<<<<<<<
* cdef DTYPE_t *X_i = NULL
* cdef int *X_argsorted_i = NULL
*/
__pyx_v_y_ptr = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_y->data);
/* "sklearn/tree/_tree.pyx":623
* cdef DTYPE_t best_error = np.inf, best_t = np.inf
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef DTYPE_t *X_i = NULL # <<<<<<<<<<<<<<
* cdef int *X_argsorted_i = NULL
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
*/
__pyx_v_X_i = NULL;
/* "sklearn/tree/_tree.pyx":624
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef DTYPE_t *X_i = NULL
* cdef int *X_argsorted_i = NULL # <<<<<<<<<<<<<<
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
*
*/
__pyx_v_X_argsorted_i = NULL;
/* "sklearn/tree/_tree.pyx":625
* cdef DTYPE_t *X_i = NULL
* cdef int *X_argsorted_i = NULL
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data # <<<<<<<<<<<<<<
*
* # Compute the column strides (increment in pointer elements to get
*/
__pyx_v_sample_mask_ptr = ((__pyx_t_7sklearn_4tree_5_tree_BOOL_t *)__pyx_v_sample_mask->data);
/* "sklearn/tree/_tree.pyx":629
* # Compute the column strides (increment in pointer elements to get
* # from column i to i + 1) for `X` and `X_argsorted`
* cdef int X_elem_stride = X.strides[0] # <<<<<<<<<<<<<<
* cdef int X_col_stride = X.strides[1]
* cdef int X_stride = X_col_stride / X_elem_stride
*/
__pyx_v_X_elem_stride = (__pyx_v_X->strides[0]);
/* "sklearn/tree/_tree.pyx":630
* # from column i to i + 1) for `X` and `X_argsorted`
* cdef int X_elem_stride = X.strides[0]
* cdef int X_col_stride = X.strides[1] # <<<<<<<<<<<<<<
* cdef int X_stride = X_col_stride / X_elem_stride
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
*/
__pyx_v_X_col_stride = (__pyx_v_X->strides[1]);
/* "sklearn/tree/_tree.pyx":631
* cdef int X_elem_stride = X.strides[0]
* cdef int X_col_stride = X.strides[1]
* cdef int X_stride = X_col_stride / X_elem_stride # <<<<<<<<<<<<<<
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
* cdef int X_argsorted_col_stride = X_argsorted.strides[1]
*/
__pyx_v_X_stride = (__pyx_v_X_col_stride / __pyx_v_X_elem_stride);
/* "sklearn/tree/_tree.pyx":632
* cdef int X_col_stride = X.strides[1]
* cdef int X_stride = X_col_stride / X_elem_stride
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0] # <<<<<<<<<<<<<<
* cdef int X_argsorted_col_stride = X_argsorted.strides[1]
* cdef int X_argsorted_stride = X_argsorted_col_stride / X_argsorted_elem_stride
*/
__pyx_v_X_argsorted_elem_stride = (__pyx_v_X_argsorted->strides[0]);
/* "sklearn/tree/_tree.pyx":633
* cdef int X_stride = X_col_stride / X_elem_stride
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
* cdef int X_argsorted_col_stride = X_argsorted.strides[1] # <<<<<<<<<<<<<<
* cdef int X_argsorted_stride = X_argsorted_col_stride / X_argsorted_elem_stride
*
*/
__pyx_v_X_argsorted_col_stride = (__pyx_v_X_argsorted->strides[1]);
/* "sklearn/tree/_tree.pyx":634
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
* cdef int X_argsorted_col_stride = X_argsorted.strides[1]
* cdef int X_argsorted_stride = X_argsorted_col_stride / X_argsorted_elem_stride # <<<<<<<<<<<<<<
*
* # Compute the initial criterion value in the node
*/
__pyx_v_X_argsorted_stride = (__pyx_v_X_argsorted_col_stride / __pyx_v_X_argsorted_elem_stride);
/* "sklearn/tree/_tree.pyx":637
*
* # Compute the initial criterion value in the node
* X_argsorted_i = <int *>X_argsorted.data # <<<<<<<<<<<<<<
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
* initial_error = criterion.eval()
*/
__pyx_v_X_argsorted_i = ((int *)__pyx_v_X_argsorted->data);
/* "sklearn/tree/_tree.pyx":638
* # Compute the initial criterion value in the node
* X_argsorted_i = <int *>X_argsorted.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples) # <<<<<<<<<<<<<<
* initial_error = criterion.eval()
*
*/
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->init(__pyx_v_criterion, __pyx_v_y_ptr, __pyx_v_sample_mask_ptr, __pyx_v_n_samples, __pyx_v_n_total_samples);
/* "sklearn/tree/_tree.pyx":639
* X_argsorted_i = <int *>X_argsorted.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
* initial_error = criterion.eval() # <<<<<<<<<<<<<<
*
* if initial_error == 0: # break early if the node is pure
*/
__pyx_v_initial_error = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->eval(__pyx_v_criterion);
/* "sklearn/tree/_tree.pyx":641
* initial_error = criterion.eval()
*
* if initial_error == 0: # break early if the node is pure # <<<<<<<<<<<<<<
* return best_i, best_t, initial_error, initial_error
*
*/
__pyx_t_4 = (__pyx_v_initial_error == 0.0);
if (__pyx_t_4) {
/* "sklearn/tree/_tree.pyx":642
*
* if initial_error == 0: # break early if the node is pure
* return best_i, best_t, initial_error, initial_error # <<<<<<<<<<<<<<
*
* best_error = initial_error
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = PyInt_FromLong(__pyx_v_best_i); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 642; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyFloat_FromDouble(__pyx_v_best_t); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 642; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_5 = PyFloat_FromDouble(__pyx_v_initial_error); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 642; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_6 = PyFloat_FromDouble(__pyx_v_initial_error); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 642; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_7 = PyTuple_New(4); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 642; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_7));
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_1);
__Pyx_GIVEREF(__pyx_t_1);
PyTuple_SET_ITEM(__pyx_t_7, 1, __pyx_t_2);
__Pyx_GIVEREF(__pyx_t_2);
PyTuple_SET_ITEM(__pyx_t_7, 2, __pyx_t_5);
__Pyx_GIVEREF(__pyx_t_5);
PyTuple_SET_ITEM(__pyx_t_7, 3, __pyx_t_6);
__Pyx_GIVEREF(__pyx_t_6);
__pyx_t_1 = 0;
__pyx_t_2 = 0;
__pyx_t_5 = 0;
__pyx_t_6 = 0;
__pyx_r = ((PyObject *)__pyx_t_7);
__pyx_t_7 = 0;
goto __pyx_L0;
goto __pyx_L6;
}
__pyx_L6:;
/* "sklearn/tree/_tree.pyx":644
* return best_i, best_t, initial_error, initial_error
*
* best_error = initial_error # <<<<<<<<<<<<<<
*
* # Features to consider
*/
__pyx_v_best_error = __pyx_v_initial_error;
/* "sklearn/tree/_tree.pyx":647
*
* # Features to consider
* if max_features < 0 or max_features == n_features: # <<<<<<<<<<<<<<
* features = np.arange(n_features)
* else:
*/
__pyx_t_4 = (__pyx_v_max_features < 0);
if (!__pyx_t_4) {
__pyx_t_8 = (__pyx_v_max_features == __pyx_v_n_features);
__pyx_t_9 = __pyx_t_8;
} else {
__pyx_t_9 = __pyx_t_4;
}
if (__pyx_t_9) {
/* "sklearn/tree/_tree.pyx":648
* # Features to consider
* if max_features < 0 or max_features == n_features:
* features = np.arange(n_features) # <<<<<<<<<<<<<<
* else:
* features = random_state.permutation(n_features)[:max_features]
*/
__pyx_t_7 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_6 = PyObject_GetAttr(__pyx_t_7, __pyx_n_s__arange); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = PyInt_FromLong(__pyx_v_n_features); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_5));
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_7);
__Pyx_GIVEREF(__pyx_t_7);
__pyx_t_7 = 0;
__pyx_t_7 = PyObject_Call(__pyx_t_6, ((PyObject *)__pyx_t_5), NULL); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 648; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_5)); __pyx_t_5 = 0;
__pyx_v_features = __pyx_t_7;
__pyx_t_7 = 0;
goto __pyx_L7;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":650
* features = np.arange(n_features)
* else:
* features = random_state.permutation(n_features)[:max_features] # <<<<<<<<<<<<<<
*
* # Look for the best split
*/
__pyx_t_7 = PyObject_GetAttr(__pyx_v_random_state, __pyx_n_s__permutation); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 650; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_5 = PyInt_FromLong(__pyx_v_n_features); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 650; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 650; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_6));
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_5);
__Pyx_GIVEREF(__pyx_t_5);
__pyx_t_5 = 0;
__pyx_t_5 = PyObject_Call(__pyx_t_7, ((PyObject *)__pyx_t_6), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 650; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_6)); __pyx_t_6 = 0;
__pyx_t_6 = __Pyx_PySequence_GetSlice(__pyx_t_5, 0, __pyx_v_max_features); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 650; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_v_features = __pyx_t_6;
__pyx_t_6 = 0;
}
__pyx_L7:;
/* "sklearn/tree/_tree.pyx":653
*
* # Look for the best split
* for i in features: # <<<<<<<<<<<<<<
* # Get i-th col of X and X_sorted
* X_i = (<DTYPE_t *>X.data) + X_stride * i
*/
if (PyList_CheckExact(__pyx_v_features) || PyTuple_CheckExact(__pyx_v_features)) {
__pyx_t_6 = __pyx_v_features; __Pyx_INCREF(__pyx_t_6); __pyx_t_10 = 0;
__pyx_t_11 = NULL;
} else {
__pyx_t_10 = -1; __pyx_t_6 = PyObject_GetIter(__pyx_v_features); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 653; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_11 = Py_TYPE(__pyx_t_6)->tp_iternext;
}
for (;;) {
if (PyList_CheckExact(__pyx_t_6)) {
if (__pyx_t_10 >= PyList_GET_SIZE(__pyx_t_6)) break;
__pyx_t_5 = PyList_GET_ITEM(__pyx_t_6, __pyx_t_10); __Pyx_INCREF(__pyx_t_5); __pyx_t_10++;
} else if (PyTuple_CheckExact(__pyx_t_6)) {
if (__pyx_t_10 >= PyTuple_GET_SIZE(__pyx_t_6)) break;
__pyx_t_5 = PyTuple_GET_ITEM(__pyx_t_6, __pyx_t_10); __Pyx_INCREF(__pyx_t_5); __pyx_t_10++;
} else {
__pyx_t_5 = __pyx_t_11(__pyx_t_6);
if (unlikely(!__pyx_t_5)) {
if (PyErr_Occurred()) {
if (likely(PyErr_ExceptionMatches(PyExc_StopIteration))) PyErr_Clear();
else {__pyx_filename = __pyx_f[0]; __pyx_lineno = 653; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
break;
}
__Pyx_GOTREF(__pyx_t_5);
}
__pyx_t_12 = __Pyx_PyInt_AsInt(__pyx_t_5); if (unlikely((__pyx_t_12 == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 653; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_v_i = __pyx_t_12;
/* "sklearn/tree/_tree.pyx":655
* for i in features:
* # Get i-th col of X and X_sorted
* X_i = (<DTYPE_t *>X.data) + X_stride * i # <<<<<<<<<<<<<<
* X_argsorted_i = (<int *>X_argsorted.data) + X_argsorted_stride * i
*
*/
__pyx_v_X_i = (((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_X->data) + (__pyx_v_X_stride * __pyx_v_i));
/* "sklearn/tree/_tree.pyx":656
* # Get i-th col of X and X_sorted
* X_i = (<DTYPE_t *>X.data) + X_stride * i
* X_argsorted_i = (<int *>X_argsorted.data) + X_argsorted_stride * i # <<<<<<<<<<<<<<
*
* # Reset the criterion for this feature
*/
__pyx_v_X_argsorted_i = (((int *)__pyx_v_X_argsorted->data) + (__pyx_v_X_argsorted_stride * __pyx_v_i));
/* "sklearn/tree/_tree.pyx":659
*
* # Reset the criterion for this feature
* criterion.reset() # <<<<<<<<<<<<<<
*
* # Index of smallest sample in X_argsorted_i that is in the sample mask
*/
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->reset(__pyx_v_criterion);
/* "sklearn/tree/_tree.pyx":662
*
* # Index of smallest sample in X_argsorted_i that is in the sample mask
* a = 0 # <<<<<<<<<<<<<<
* while sample_mask_ptr[X_argsorted_i[a]] == 0:
* a = a + 1
*/
__pyx_v_a = 0;
/* "sklearn/tree/_tree.pyx":663
* # Index of smallest sample in X_argsorted_i that is in the sample mask
* a = 0
* while sample_mask_ptr[X_argsorted_i[a]] == 0: # <<<<<<<<<<<<<<
* a = a + 1
*
*/
while (1) {
__pyx_t_9 = ((__pyx_v_sample_mask_ptr[(__pyx_v_X_argsorted_i[__pyx_v_a])]) == 0);
if (!__pyx_t_9) break;
/* "sklearn/tree/_tree.pyx":664
* a = 0
* while sample_mask_ptr[X_argsorted_i[a]] == 0:
* a = a + 1 # <<<<<<<<<<<<<<
*
* # Consider splits between two consecutive samples
*/
__pyx_v_a = (__pyx_v_a + 1);
}
/* "sklearn/tree/_tree.pyx":667
*
* # Consider splits between two consecutive samples
* while True: # <<<<<<<<<<<<<<
* # Find the following larger sample
* b = smallest_sample_larger_than(a, X_i, X_argsorted_i,
*/
while (1) {
if (!1) break;
/* "sklearn/tree/_tree.pyx":670
* # Find the following larger sample
* b = smallest_sample_larger_than(a, X_i, X_argsorted_i,
* sample_mask_ptr, n_total_samples) # <<<<<<<<<<<<<<
* if b == -1:
* break
*/
__pyx_v_b = __pyx_f_7sklearn_4tree_5_tree_smallest_sample_larger_than(__pyx_v_a, __pyx_v_X_i, __pyx_v_X_argsorted_i, __pyx_v_sample_mask_ptr, __pyx_v_n_total_samples);
/* "sklearn/tree/_tree.pyx":671
* b = smallest_sample_larger_than(a, X_i, X_argsorted_i,
* sample_mask_ptr, n_total_samples)
* if b == -1: # <<<<<<<<<<<<<<
* break
*
*/
__pyx_t_9 = (__pyx_v_b == -1);
if (__pyx_t_9) {
/* "sklearn/tree/_tree.pyx":672
* sample_mask_ptr, n_total_samples)
* if b == -1:
* break # <<<<<<<<<<<<<<
*
* # Better split than the best so far?
*/
goto __pyx_L13_break;
goto __pyx_L14;
}
__pyx_L14:;
/* "sklearn/tree/_tree.pyx":675
*
* # Better split than the best so far?
* n_left = criterion.update(a, b, y_ptr, X_argsorted_i, sample_mask_ptr) # <<<<<<<<<<<<<<
*
* # Only consider splits that respect min_leaf
*/
__pyx_v_n_left = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->update(__pyx_v_criterion, __pyx_v_a, __pyx_v_b, __pyx_v_y_ptr, __pyx_v_X_argsorted_i, __pyx_v_sample_mask_ptr);
/* "sklearn/tree/_tree.pyx":678
*
* # Only consider splits that respect min_leaf
* if n_left < min_leaf or (n_samples - n_left) < min_leaf: # <<<<<<<<<<<<<<
* a = b
* continue
*/
__pyx_t_9 = (__pyx_v_n_left < __pyx_v_min_leaf);
if (!__pyx_t_9) {
__pyx_t_4 = ((__pyx_v_n_samples - __pyx_v_n_left) < __pyx_v_min_leaf);
__pyx_t_8 = __pyx_t_4;
} else {
__pyx_t_8 = __pyx_t_9;
}
if (__pyx_t_8) {
/* "sklearn/tree/_tree.pyx":679
* # Only consider splits that respect min_leaf
* if n_left < min_leaf or (n_samples - n_left) < min_leaf:
* a = b # <<<<<<<<<<<<<<
* continue
*
*/
__pyx_v_a = __pyx_v_b;
/* "sklearn/tree/_tree.pyx":680
* if n_left < min_leaf or (n_samples - n_left) < min_leaf:
* a = b
* continue # <<<<<<<<<<<<<<
*
* error = criterion.eval()
*/
goto __pyx_L12_continue;
goto __pyx_L15;
}
__pyx_L15:;
/* "sklearn/tree/_tree.pyx":682
* continue
*
* error = criterion.eval() # <<<<<<<<<<<<<<
*
* if error < best_error:
*/
__pyx_v_error = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->eval(__pyx_v_criterion);
/* "sklearn/tree/_tree.pyx":684
* error = criterion.eval()
*
* if error < best_error: # <<<<<<<<<<<<<<
* t = X_i[X_argsorted_i[a]] + \
* ((X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]]) / 2.0)
*/
__pyx_t_8 = (__pyx_v_error < __pyx_v_best_error);
if (__pyx_t_8) {
/* "sklearn/tree/_tree.pyx":686
* if error < best_error:
* t = X_i[X_argsorted_i[a]] + \
* ((X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]]) / 2.0) # <<<<<<<<<<<<<<
* if t == X_i[X_argsorted_i[b]]:
* t = X_i[X_argsorted_i[a]]
*/
__pyx_v_t = ((__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])]) + (((__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_b])]) - (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])])) / 2.0));
/* "sklearn/tree/_tree.pyx":687
* t = X_i[X_argsorted_i[a]] + \
* ((X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]]) / 2.0)
* if t == X_i[X_argsorted_i[b]]: # <<<<<<<<<<<<<<
* t = X_i[X_argsorted_i[a]]
* best_i = i
*/
__pyx_t_8 = (__pyx_v_t == (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_b])]));
if (__pyx_t_8) {
/* "sklearn/tree/_tree.pyx":688
* ((X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]]) / 2.0)
* if t == X_i[X_argsorted_i[b]]:
* t = X_i[X_argsorted_i[a]] # <<<<<<<<<<<<<<
* best_i = i
* best_t = t
*/
__pyx_v_t = (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])]);
goto __pyx_L17;
}
__pyx_L17:;
/* "sklearn/tree/_tree.pyx":689
* if t == X_i[X_argsorted_i[b]]:
* t = X_i[X_argsorted_i[a]]
* best_i = i # <<<<<<<<<<<<<<
* best_t = t
* best_error = error
*/
__pyx_v_best_i = __pyx_v_i;
/* "sklearn/tree/_tree.pyx":690
* t = X_i[X_argsorted_i[a]]
* best_i = i
* best_t = t # <<<<<<<<<<<<<<
* best_error = error
*
*/
__pyx_v_best_t = __pyx_v_t;
/* "sklearn/tree/_tree.pyx":691
* best_i = i
* best_t = t
* best_error = error # <<<<<<<<<<<<<<
*
* # Proceed to the next interval
*/
__pyx_v_best_error = __pyx_v_error;
goto __pyx_L16;
}
__pyx_L16:;
/* "sklearn/tree/_tree.pyx":694
*
* # Proceed to the next interval
* a = b # <<<<<<<<<<<<<<
*
* return best_i, best_t, best_error, initial_error
*/
__pyx_v_a = __pyx_v_b;
__pyx_L12_continue:;
}
__pyx_L13_break:;
}
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
/* "sklearn/tree/_tree.pyx":696
* a = b
*
* return best_i, best_t, best_error, initial_error # <<<<<<<<<<<<<<
*
* def _find_best_random_split(np.ndarray[DTYPE_t, ndim=2, mode="fortran"] X,
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_6 = PyInt_FromLong(__pyx_v_best_i); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 696; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_5 = PyFloat_FromDouble(__pyx_v_best_t); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 696; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_7 = PyFloat_FromDouble(__pyx_v_best_error); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 696; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_2 = PyFloat_FromDouble(__pyx_v_initial_error); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 696; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_1 = PyTuple_New(4); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 696; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_1));
PyTuple_SET_ITEM(__pyx_t_1, 0, __pyx_t_6);
__Pyx_GIVEREF(__pyx_t_6);
PyTuple_SET_ITEM(__pyx_t_1, 1, __pyx_t_5);
__Pyx_GIVEREF(__pyx_t_5);
PyTuple_SET_ITEM(__pyx_t_1, 2, __pyx_t_7);
__Pyx_GIVEREF(__pyx_t_7);
PyTuple_SET_ITEM(__pyx_t_1, 3, __pyx_t_2);
__Pyx_GIVEREF(__pyx_t_2);
__pyx_t_6 = 0;
__pyx_t_5 = 0;
__pyx_t_7 = 0;
__pyx_t_2 = 0;
__pyx_r = ((PyObject *)__pyx_t_1);
__pyx_t_1 = 0;
goto __pyx_L0;
__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_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_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X_argsorted);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_y);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._find_best_split", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X_argsorted);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_y);
__pyx_L2:;
__Pyx_XDECREF(__pyx_v_features);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "sklearn/tree/_tree.pyx":698
* return best_i, best_t, best_error, initial_error
*
* def _find_best_random_split(np.ndarray[DTYPE_t, ndim=2, mode="fortran"] X, # <<<<<<<<<<<<<<
* np.ndarray[DTYPE_t, ndim=1, mode="c"] y,
* np.ndarray[np.int32_t, ndim=2, mode="fortran"] X_argsorted,
*/
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_6_find_best_random_split(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds); /*proto*/
static char __pyx_doc_7sklearn_4tree_5_tree_6_find_best_random_split[] = "Find the best dimension and threshold that minimises the error.\n\n Parameters\n ----------\n X : ndarray, shape (n_total_samples, n_features), dtype=DTYPE_t\n The feature values.\n\n y : ndarray, shape (n_total_samples,), dtype=float\n The label to predict for each sample.\n\n X_argsorted : ndarray, shape (n_samples, n_features)\n Argsort of cols of `X`. `X_argsorted[0,j]` gives the example\n index of the smallest value of feature `j`.\n\n sample_mask : ndarray, shape (n_samples,), dtype=np.bool\n A mask for the samples to be considered. Only samples `j` for which\n sample_mask[j] != 0 are considered.\n\n n_samples : int\n The number of samples in the current sample_mask\n (i.e. `sample_mask.sum()`).\n\n min_leaf : int\n The minimum number of samples required to be at a leaf node.\n\n max_features : int\n The number of features to consider when looking for the best split.\n\n criterion : Criterion\n The criterion function to be minimized.\n\n random_state : RandomState\n The numpy random state to use.\n\n Returns\n -------\n best_i : int\n The split feature or -1 if criterion not smaller than\n `parent_split_error`.\n\n best_t : DTYPE_t\n The split threshold\n\n best_error : DTYPE_t\n The split error\n\n initial_error : DTYPE_t\n The initial error contained in the node.\n ";
static PyMethodDef __pyx_mdef_7sklearn_4tree_5_tree_6_find_best_random_split = {__Pyx_NAMESTR("_find_best_random_split"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_6_find_best_random_split, METH_VARARGS|METH_KEYWORDS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_6_find_best_random_split)};
static PyObject *__pyx_pf_7sklearn_4tree_5_tree_6_find_best_random_split(PyObject *__pyx_self, PyObject *__pyx_args, PyObject *__pyx_kwds) {
PyArrayObject *__pyx_v_X = 0;
PyArrayObject *__pyx_v_y = 0;
PyArrayObject *__pyx_v_X_argsorted = 0;
PyArrayObject *__pyx_v_sample_mask = 0;
int __pyx_v_n_samples;
int __pyx_v_min_leaf;
int __pyx_v_max_features;
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *__pyx_v_criterion = 0;
PyObject *__pyx_v_random_state = 0;
int __pyx_v_n_total_samples;
int __pyx_v_n_features;
int __pyx_v_i;
int __pyx_v_a;
int __pyx_v_b;
int __pyx_v_best_i;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_t;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_initial_error;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_error;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_best_error;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_v_best_t;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_y_ptr;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *__pyx_v_X_i;
int *__pyx_v_X_argsorted_i;
__pyx_t_7sklearn_4tree_5_tree_BOOL_t *__pyx_v_sample_mask_ptr;
int __pyx_v_X_elem_stride;
int __pyx_v_X_col_stride;
int __pyx_v_X_stride;
int __pyx_v_X_argsorted_elem_stride;
int __pyx_v_X_argsorted_col_stride;
int __pyx_v_X_argsorted_stride;
PyObject *__pyx_v_features = NULL;
PyObject *__pyx_v_c = NULL;
PyObject *__pyx_v_n_left = NULL;
Py_buffer __pyx_bstruct_X;
Py_ssize_t __pyx_bstride_0_X = 0;
Py_ssize_t __pyx_bstride_1_X = 0;
Py_ssize_t __pyx_bshape_0_X = 0;
Py_ssize_t __pyx_bshape_1_X = 0;
Py_buffer __pyx_bstruct_X_argsorted;
Py_ssize_t __pyx_bstride_0_X_argsorted = 0;
Py_ssize_t __pyx_bstride_1_X_argsorted = 0;
Py_ssize_t __pyx_bshape_0_X_argsorted = 0;
Py_ssize_t __pyx_bshape_1_X_argsorted = 0;
Py_buffer __pyx_bstruct_y;
Py_ssize_t __pyx_bstride_0_y = 0;
Py_ssize_t __pyx_bshape_0_y = 0;
PyObject *__pyx_r = NULL;
__Pyx_RefNannyDeclarations
PyObject *__pyx_t_1 = NULL;
PyObject *__pyx_t_2 = NULL;
__pyx_t_7sklearn_4tree_5_tree_DTYPE_t __pyx_t_3;
int __pyx_t_4;
PyObject *__pyx_t_5 = NULL;
PyObject *__pyx_t_6 = NULL;
PyObject *__pyx_t_7 = NULL;
Py_ssize_t __pyx_t_8;
PyObject *(*__pyx_t_9)(PyObject *);
int __pyx_t_10;
int __pyx_t_11;
int __pyx_t_12;
Py_ssize_t __pyx_t_13;
int __pyx_lineno = 0;
const char *__pyx_filename = NULL;
int __pyx_clineno = 0;
static PyObject **__pyx_pyargnames[] = {&__pyx_n_s__X,&__pyx_n_s__y,&__pyx_n_s__X_argsorted,&__pyx_n_s__sample_mask,&__pyx_n_s__n_samples,&__pyx_n_s__min_leaf,&__pyx_n_s__max_features,&__pyx_n_s__criterion,&__pyx_n_s__random_state,0};
__Pyx_RefNannySetupContext("_find_best_random_split");
__pyx_self = __pyx_self;
{
PyObject* values[9] = {0,0,0,0,0,0,0,0,0};
if (unlikely(__pyx_kwds)) {
Py_ssize_t kw_args;
switch (PyTuple_GET_SIZE(__pyx_args)) {
case 9: values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
case 8: values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
case 7: values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
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 (PyTuple_GET_SIZE(__pyx_args)) {
case 0:
values[0] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X);
if (likely(values[0])) kw_args--;
else goto __pyx_L5_argtuple_error;
case 1:
values[1] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__y);
if (likely(values[1])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 1); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 2:
values[2] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__X_argsorted);
if (likely(values[2])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 2); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 3:
values[3] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__sample_mask);
if (likely(values[3])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 3); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 4:
values[4] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__n_samples);
if (likely(values[4])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 4); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 5:
values[5] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__min_leaf);
if (likely(values[5])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 5); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 6:
values[6] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__max_features);
if (likely(values[6])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 6); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 7:
values[7] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__criterion);
if (likely(values[7])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 7); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
case 8:
values[8] = PyDict_GetItem(__pyx_kwds, __pyx_n_s__random_state);
if (likely(values[8])) kw_args--;
else {
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, 8); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
}
if (unlikely(kw_args > 0)) {
if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_pyargnames, 0, values, PyTuple_GET_SIZE(__pyx_args), "_find_best_random_split") < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
}
} else if (PyTuple_GET_SIZE(__pyx_args) != 9) {
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);
values[6] = PyTuple_GET_ITEM(__pyx_args, 6);
values[7] = PyTuple_GET_ITEM(__pyx_args, 7);
values[8] = PyTuple_GET_ITEM(__pyx_args, 8);
}
__pyx_v_X = ((PyArrayObject *)values[0]);
__pyx_v_y = ((PyArrayObject *)values[1]);
__pyx_v_X_argsorted = ((PyArrayObject *)values[2]);
__pyx_v_sample_mask = ((PyArrayObject *)values[3]);
__pyx_v_n_samples = __Pyx_PyInt_AsInt(values[4]); if (unlikely((__pyx_v_n_samples == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 702; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_min_leaf = __Pyx_PyInt_AsInt(values[5]); if (unlikely((__pyx_v_min_leaf == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 703; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_max_features = __Pyx_PyInt_AsInt(values[6]); if (unlikely((__pyx_v_max_features == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 704; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_v_criterion = ((struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *)values[7]);
__pyx_v_random_state = values[8];
}
goto __pyx_L4_argument_unpacking_done;
__pyx_L5_argtuple_error:;
__Pyx_RaiseArgtupleInvalid("_find_best_random_split", 1, 9, 9, PyTuple_GET_SIZE(__pyx_args)); {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L3_error;}
__pyx_L3_error:;
__Pyx_AddTraceback("sklearn.tree._tree._find_best_random_split", __pyx_clineno, __pyx_lineno, __pyx_filename);
__Pyx_RefNannyFinishContext();
return NULL;
__pyx_L4_argument_unpacking_done:;
__pyx_bstruct_X.buf = NULL;
__pyx_bstruct_y.buf = NULL;
__pyx_bstruct_X_argsorted.buf = NULL;
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X), __pyx_ptype_5numpy_ndarray, 1, "X", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __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 = 699; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_X_argsorted), __pyx_ptype_5numpy_ndarray, 1, "X_argsorted", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 700; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_sample_mask), __pyx_ptype_5numpy_ndarray, 1, "sample_mask", 0))) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 701; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 705; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X, (PyObject*)__pyx_v_X, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_F_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X = __pyx_bstruct_X.strides[0]; __pyx_bstride_1_X = __pyx_bstruct_X.strides[1];
__pyx_bshape_0_X = __pyx_bstruct_X.shape[0]; __pyx_bshape_1_X = __pyx_bstruct_X.shape[1];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_y, (PyObject*)__pyx_v_y, &__Pyx_TypeInfo_nn___pyx_t_7sklearn_4tree_5_tree_DTYPE_t, PyBUF_FORMAT| PyBUF_C_CONTIGUOUS, 1, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_y = __pyx_bstruct_y.strides[0];
__pyx_bshape_0_y = __pyx_bstruct_y.shape[0];
{
__Pyx_BufFmt_StackElem __pyx_stack[1];
if (unlikely(__Pyx_GetBufferAndValidate(&__pyx_bstruct_X_argsorted, (PyObject*)__pyx_v_X_argsorted, &__Pyx_TypeInfo_nn___pyx_t_5numpy_int32_t, PyBUF_FORMAT| PyBUF_F_CONTIGUOUS, 2, 0, __pyx_stack) == -1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_bstride_0_X_argsorted = __pyx_bstruct_X_argsorted.strides[0]; __pyx_bstride_1_X_argsorted = __pyx_bstruct_X_argsorted.strides[1];
__pyx_bshape_0_X_argsorted = __pyx_bstruct_X_argsorted.shape[0]; __pyx_bshape_1_X_argsorted = __pyx_bstruct_X_argsorted.shape[1];
/* "sklearn/tree/_tree.pyx":757
* """
* # Variables
* cdef int n_total_samples = X.shape[0] # <<<<<<<<<<<<<<
* cdef int n_features = X.shape[1]
* cdef int i, a, b, best_i = -1
*/
__pyx_v_n_total_samples = (__pyx_v_X->dimensions[0]);
/* "sklearn/tree/_tree.pyx":758
* # Variables
* cdef int n_total_samples = X.shape[0]
* cdef int n_features = X.shape[1] # <<<<<<<<<<<<<<
* cdef int i, a, b, best_i = -1
* cdef DTYPE_t t, initial_error, error
*/
__pyx_v_n_features = (__pyx_v_X->dimensions[1]);
/* "sklearn/tree/_tree.pyx":759
* cdef int n_total_samples = X.shape[0]
* cdef int n_features = X.shape[1]
* cdef int i, a, b, best_i = -1 # <<<<<<<<<<<<<<
* cdef DTYPE_t t, initial_error, error
* cdef DTYPE_t best_error = np.inf, best_t = np.inf
*/
__pyx_v_best_i = -1;
/* "sklearn/tree/_tree.pyx":761
* cdef int i, a, b, best_i = -1
* cdef DTYPE_t t, initial_error, error
* cdef DTYPE_t best_error = np.inf, best_t = np.inf # <<<<<<<<<<<<<<
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef DTYPE_t *X_i = NULL
*/
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 761; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__inf); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 761; __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_PyFloat_AsDouble(__pyx_t_2); if (unlikely((__pyx_t_3 == (npy_float32)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 761; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
__pyx_v_best_error = __pyx_t_3;
__pyx_t_2 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 761; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_1 = PyObject_GetAttr(__pyx_t_2, __pyx_n_s__inf); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 761; __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 == (npy_float32)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 761; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_v_best_t = __pyx_t_3;
/* "sklearn/tree/_tree.pyx":762
* cdef DTYPE_t t, initial_error, error
* cdef DTYPE_t best_error = np.inf, best_t = np.inf
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data # <<<<<<<<<<<<<<
* cdef DTYPE_t *X_i = NULL
* cdef int *X_argsorted_i = NULL
*/
__pyx_v_y_ptr = ((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_y->data);
/* "sklearn/tree/_tree.pyx":763
* cdef DTYPE_t best_error = np.inf, best_t = np.inf
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef DTYPE_t *X_i = NULL # <<<<<<<<<<<<<<
* cdef int *X_argsorted_i = NULL
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
*/
__pyx_v_X_i = NULL;
/* "sklearn/tree/_tree.pyx":764
* cdef DTYPE_t *y_ptr = <DTYPE_t *>y.data
* cdef DTYPE_t *X_i = NULL
* cdef int *X_argsorted_i = NULL # <<<<<<<<<<<<<<
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data
*
*/
__pyx_v_X_argsorted_i = NULL;
/* "sklearn/tree/_tree.pyx":765
* cdef DTYPE_t *X_i = NULL
* cdef int *X_argsorted_i = NULL
* cdef BOOL_t *sample_mask_ptr = <BOOL_t *>sample_mask.data # <<<<<<<<<<<<<<
*
* # Compute the column strides (increment in pointer elements to get
*/
__pyx_v_sample_mask_ptr = ((__pyx_t_7sklearn_4tree_5_tree_BOOL_t *)__pyx_v_sample_mask->data);
/* "sklearn/tree/_tree.pyx":769
* # Compute the column strides (increment in pointer elements to get
* # from column i to i + 1) for `X` and `X_argsorted`
* cdef int X_elem_stride = X.strides[0] # <<<<<<<<<<<<<<
* cdef int X_col_stride = X.strides[1]
* cdef int X_stride = X_col_stride / X_elem_stride
*/
__pyx_v_X_elem_stride = (__pyx_v_X->strides[0]);
/* "sklearn/tree/_tree.pyx":770
* # from column i to i + 1) for `X` and `X_argsorted`
* cdef int X_elem_stride = X.strides[0]
* cdef int X_col_stride = X.strides[1] # <<<<<<<<<<<<<<
* cdef int X_stride = X_col_stride / X_elem_stride
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
*/
__pyx_v_X_col_stride = (__pyx_v_X->strides[1]);
/* "sklearn/tree/_tree.pyx":771
* cdef int X_elem_stride = X.strides[0]
* cdef int X_col_stride = X.strides[1]
* cdef int X_stride = X_col_stride / X_elem_stride # <<<<<<<<<<<<<<
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
* cdef int X_argsorted_col_stride = X_argsorted.strides[1]
*/
__pyx_v_X_stride = (__pyx_v_X_col_stride / __pyx_v_X_elem_stride);
/* "sklearn/tree/_tree.pyx":772
* cdef int X_col_stride = X.strides[1]
* cdef int X_stride = X_col_stride / X_elem_stride
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0] # <<<<<<<<<<<<<<
* cdef int X_argsorted_col_stride = X_argsorted.strides[1]
* cdef int X_argsorted_stride = X_argsorted_col_stride / X_argsorted_elem_stride
*/
__pyx_v_X_argsorted_elem_stride = (__pyx_v_X_argsorted->strides[0]);
/* "sklearn/tree/_tree.pyx":773
* cdef int X_stride = X_col_stride / X_elem_stride
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
* cdef int X_argsorted_col_stride = X_argsorted.strides[1] # <<<<<<<<<<<<<<
* cdef int X_argsorted_stride = X_argsorted_col_stride / X_argsorted_elem_stride
*
*/
__pyx_v_X_argsorted_col_stride = (__pyx_v_X_argsorted->strides[1]);
/* "sklearn/tree/_tree.pyx":774
* cdef int X_argsorted_elem_stride = X_argsorted.strides[0]
* cdef int X_argsorted_col_stride = X_argsorted.strides[1]
* cdef int X_argsorted_stride = X_argsorted_col_stride / X_argsorted_elem_stride # <<<<<<<<<<<<<<
*
* # Compute the initial criterion value
*/
__pyx_v_X_argsorted_stride = (__pyx_v_X_argsorted_col_stride / __pyx_v_X_argsorted_elem_stride);
/* "sklearn/tree/_tree.pyx":777
*
* # Compute the initial criterion value
* X_argsorted_i = <int *>X_argsorted.data # <<<<<<<<<<<<<<
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
* initial_error = criterion.eval()
*/
__pyx_v_X_argsorted_i = ((int *)__pyx_v_X_argsorted->data);
/* "sklearn/tree/_tree.pyx":778
* # Compute the initial criterion value
* X_argsorted_i = <int *>X_argsorted.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples) # <<<<<<<<<<<<<<
* initial_error = criterion.eval()
*
*/
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->init(__pyx_v_criterion, __pyx_v_y_ptr, __pyx_v_sample_mask_ptr, __pyx_v_n_samples, __pyx_v_n_total_samples);
/* "sklearn/tree/_tree.pyx":779
* X_argsorted_i = <int *>X_argsorted.data
* criterion.init(y_ptr, sample_mask_ptr, n_samples, n_total_samples)
* initial_error = criterion.eval() # <<<<<<<<<<<<<<
*
* if initial_error == 0: # break early if the node is pure
*/
__pyx_v_initial_error = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->eval(__pyx_v_criterion);
/* "sklearn/tree/_tree.pyx":781
* initial_error = criterion.eval()
*
* if initial_error == 0: # break early if the node is pure # <<<<<<<<<<<<<<
* return best_i, best_t, best_error, initial_error
*
*/
__pyx_t_4 = (__pyx_v_initial_error == 0.0);
if (__pyx_t_4) {
/* "sklearn/tree/_tree.pyx":782
*
* if initial_error == 0: # break early if the node is pure
* return best_i, best_t, best_error, initial_error # <<<<<<<<<<<<<<
*
* best_error = initial_error
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_1 = PyInt_FromLong(__pyx_v_best_i); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 782; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyFloat_FromDouble(__pyx_v_best_t); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 782; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__pyx_t_5 = PyFloat_FromDouble(__pyx_v_best_error); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 782; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_6 = PyFloat_FromDouble(__pyx_v_initial_error); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 782; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_7 = PyTuple_New(4); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 782; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_7));
PyTuple_SET_ITEM(__pyx_t_7, 0, __pyx_t_1);
__Pyx_GIVEREF(__pyx_t_1);
PyTuple_SET_ITEM(__pyx_t_7, 1, __pyx_t_2);
__Pyx_GIVEREF(__pyx_t_2);
PyTuple_SET_ITEM(__pyx_t_7, 2, __pyx_t_5);
__Pyx_GIVEREF(__pyx_t_5);
PyTuple_SET_ITEM(__pyx_t_7, 3, __pyx_t_6);
__Pyx_GIVEREF(__pyx_t_6);
__pyx_t_1 = 0;
__pyx_t_2 = 0;
__pyx_t_5 = 0;
__pyx_t_6 = 0;
__pyx_r = ((PyObject *)__pyx_t_7);
__pyx_t_7 = 0;
goto __pyx_L0;
goto __pyx_L6;
}
__pyx_L6:;
/* "sklearn/tree/_tree.pyx":784
* return best_i, best_t, best_error, initial_error
*
* best_error = initial_error # <<<<<<<<<<<<<<
*
* # Features to consider
*/
__pyx_v_best_error = __pyx_v_initial_error;
/* "sklearn/tree/_tree.pyx":787
*
* # Features to consider
* if max_features == n_features: # <<<<<<<<<<<<<<
* features = np.arange(n_features)
* else:
*/
__pyx_t_4 = (__pyx_v_max_features == __pyx_v_n_features);
if (__pyx_t_4) {
/* "sklearn/tree/_tree.pyx":788
* # Features to consider
* if max_features == n_features:
* features = np.arange(n_features) # <<<<<<<<<<<<<<
* else:
* features = random_state.permutation(n_features)[:max_features]
*/
__pyx_t_7 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 788; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_6 = PyObject_GetAttr(__pyx_t_7, __pyx_n_s__arange); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 788; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = PyInt_FromLong(__pyx_v_n_features); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 788; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_5 = PyTuple_New(1); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 788; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_5));
PyTuple_SET_ITEM(__pyx_t_5, 0, __pyx_t_7);
__Pyx_GIVEREF(__pyx_t_7);
__pyx_t_7 = 0;
__pyx_t_7 = PyObject_Call(__pyx_t_6, ((PyObject *)__pyx_t_5), NULL); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 788; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_5)); __pyx_t_5 = 0;
__pyx_v_features = __pyx_t_7;
__pyx_t_7 = 0;
goto __pyx_L7;
}
/*else*/ {
/* "sklearn/tree/_tree.pyx":790
* features = np.arange(n_features)
* else:
* features = random_state.permutation(n_features)[:max_features] # <<<<<<<<<<<<<<
*
* # Look for the best random split
*/
__pyx_t_7 = PyObject_GetAttr(__pyx_v_random_state, __pyx_n_s__permutation); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 790; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_5 = PyInt_FromLong(__pyx_v_n_features); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 790; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_6 = PyTuple_New(1); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 790; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_6));
PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_t_5);
__Pyx_GIVEREF(__pyx_t_5);
__pyx_t_5 = 0;
__pyx_t_5 = PyObject_Call(__pyx_t_7, ((PyObject *)__pyx_t_6), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 790; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__Pyx_DECREF(((PyObject *)__pyx_t_6)); __pyx_t_6 = 0;
__pyx_t_6 = __Pyx_PySequence_GetSlice(__pyx_t_5, 0, __pyx_v_max_features); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 790; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_v_features = __pyx_t_6;
__pyx_t_6 = 0;
}
__pyx_L7:;
/* "sklearn/tree/_tree.pyx":793
*
* # Look for the best random split
* for i in features: # <<<<<<<<<<<<<<
* # Get i-th col of X and X_sorted
* X_i = (<DTYPE_t *>X.data) + X_stride * i
*/
if (PyList_CheckExact(__pyx_v_features) || PyTuple_CheckExact(__pyx_v_features)) {
__pyx_t_6 = __pyx_v_features; __Pyx_INCREF(__pyx_t_6); __pyx_t_8 = 0;
__pyx_t_9 = NULL;
} else {
__pyx_t_8 = -1; __pyx_t_6 = PyObject_GetIter(__pyx_v_features); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 793; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_9 = Py_TYPE(__pyx_t_6)->tp_iternext;
}
for (;;) {
if (PyList_CheckExact(__pyx_t_6)) {
if (__pyx_t_8 >= PyList_GET_SIZE(__pyx_t_6)) break;
__pyx_t_5 = PyList_GET_ITEM(__pyx_t_6, __pyx_t_8); __Pyx_INCREF(__pyx_t_5); __pyx_t_8++;
} else if (PyTuple_CheckExact(__pyx_t_6)) {
if (__pyx_t_8 >= PyTuple_GET_SIZE(__pyx_t_6)) break;
__pyx_t_5 = PyTuple_GET_ITEM(__pyx_t_6, __pyx_t_8); __Pyx_INCREF(__pyx_t_5); __pyx_t_8++;
} else {
__pyx_t_5 = __pyx_t_9(__pyx_t_6);
if (unlikely(!__pyx_t_5)) {
if (PyErr_Occurred()) {
if (likely(PyErr_ExceptionMatches(PyExc_StopIteration))) PyErr_Clear();
else {__pyx_filename = __pyx_f[0]; __pyx_lineno = 793; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
break;
}
__Pyx_GOTREF(__pyx_t_5);
}
__pyx_t_10 = __Pyx_PyInt_AsInt(__pyx_t_5); if (unlikely((__pyx_t_10 == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 793; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_v_i = __pyx_t_10;
/* "sklearn/tree/_tree.pyx":795
* for i in features:
* # Get i-th col of X and X_sorted
* X_i = (<DTYPE_t *>X.data) + X_stride * i # <<<<<<<<<<<<<<
* X_argsorted_i = (<int *>X_argsorted.data) + X_argsorted_stride * i
*
*/
__pyx_v_X_i = (((__pyx_t_7sklearn_4tree_5_tree_DTYPE_t *)__pyx_v_X->data) + (__pyx_v_X_stride * __pyx_v_i));
/* "sklearn/tree/_tree.pyx":796
* # Get i-th col of X and X_sorted
* X_i = (<DTYPE_t *>X.data) + X_stride * i
* X_argsorted_i = (<int *>X_argsorted.data) + X_argsorted_stride * i # <<<<<<<<<<<<<<
*
* # Reset the criterion for this feature
*/
__pyx_v_X_argsorted_i = (((int *)__pyx_v_X_argsorted->data) + (__pyx_v_X_argsorted_stride * __pyx_v_i));
/* "sklearn/tree/_tree.pyx":799
*
* # Reset the criterion for this feature
* criterion.reset() # <<<<<<<<<<<<<<
*
* # Find min and max
*/
((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->reset(__pyx_v_criterion);
/* "sklearn/tree/_tree.pyx":802
*
* # Find min and max
* a = 0 # <<<<<<<<<<<<<<
* while sample_mask_ptr[X_argsorted_i[a]] == 0:
* a = a + 1
*/
__pyx_v_a = 0;
/* "sklearn/tree/_tree.pyx":803
* # Find min and max
* a = 0
* while sample_mask_ptr[X_argsorted_i[a]] == 0: # <<<<<<<<<<<<<<
* a = a + 1
*
*/
while (1) {
__pyx_t_4 = ((__pyx_v_sample_mask_ptr[(__pyx_v_X_argsorted_i[__pyx_v_a])]) == 0);
if (!__pyx_t_4) break;
/* "sklearn/tree/_tree.pyx":804
* a = 0
* while sample_mask_ptr[X_argsorted_i[a]] == 0:
* a = a + 1 # <<<<<<<<<<<<<<
*
* b = n_total_samples - 1
*/
__pyx_v_a = (__pyx_v_a + 1);
}
/* "sklearn/tree/_tree.pyx":806
* a = a + 1
*
* b = n_total_samples - 1 # <<<<<<<<<<<<<<
* while sample_mask_ptr[X_argsorted_i[b]] == 0:
* b = b - 1
*/
__pyx_v_b = (__pyx_v_n_total_samples - 1);
/* "sklearn/tree/_tree.pyx":807
*
* b = n_total_samples - 1
* while sample_mask_ptr[X_argsorted_i[b]] == 0: # <<<<<<<<<<<<<<
* b = b - 1
*
*/
while (1) {
__pyx_t_4 = ((__pyx_v_sample_mask_ptr[(__pyx_v_X_argsorted_i[__pyx_v_b])]) == 0);
if (!__pyx_t_4) break;
/* "sklearn/tree/_tree.pyx":808
* b = n_total_samples - 1
* while sample_mask_ptr[X_argsorted_i[b]] == 0:
* b = b - 1 # <<<<<<<<<<<<<<
*
* if b <= a or X_i[X_argsorted_i[a]] == X_i[X_argsorted_i[b]]:
*/
__pyx_v_b = (__pyx_v_b - 1);
}
/* "sklearn/tree/_tree.pyx":810
* b = b - 1
*
* if b <= a or X_i[X_argsorted_i[a]] == X_i[X_argsorted_i[b]]: # <<<<<<<<<<<<<<
* continue
*
*/
__pyx_t_4 = (__pyx_v_b <= __pyx_v_a);
if (!__pyx_t_4) {
__pyx_t_11 = ((__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])]) == (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_b])]));
__pyx_t_12 = __pyx_t_11;
} else {
__pyx_t_12 = __pyx_t_4;
}
if (__pyx_t_12) {
/* "sklearn/tree/_tree.pyx":811
*
* if b <= a or X_i[X_argsorted_i[a]] == X_i[X_argsorted_i[b]]:
* continue # <<<<<<<<<<<<<<
*
* # Draw a random threshold in [a, b)
*/
goto __pyx_L8_continue;
goto __pyx_L14;
}
__pyx_L14:;
/* "sklearn/tree/_tree.pyx":814
*
* # Draw a random threshold in [a, b)
* t = X_i[X_argsorted_i[a]] + random_state.rand() * (X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]]) # <<<<<<<<<<<<<<
* if t == X_i[X_argsorted_i[b]]:
* t = X_i[X_argsorted_i[a]]
*/
__pyx_t_5 = PyFloat_FromDouble((__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])])); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_7 = PyObject_GetAttr(__pyx_v_random_state, __pyx_n_s__rand); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_2 = PyObject_Call(__pyx_t_7, ((PyObject *)__pyx_empty_tuple), NULL); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = PyFloat_FromDouble(((__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_b])]) - (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])]))); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_1 = PyNumber_Multiply(__pyx_t_2, __pyx_t_7); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __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_7); __pyx_t_7 = 0;
__pyx_t_7 = PyNumber_Add(__pyx_t_5, __pyx_t_1); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __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_t_3 = __pyx_PyFloat_AsDouble(__pyx_t_7); if (unlikely((__pyx_t_3 == (npy_float32)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 814; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_v_t = __pyx_t_3;
/* "sklearn/tree/_tree.pyx":815
* # Draw a random threshold in [a, b)
* t = X_i[X_argsorted_i[a]] + random_state.rand() * (X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]])
* if t == X_i[X_argsorted_i[b]]: # <<<<<<<<<<<<<<
* t = X_i[X_argsorted_i[a]]
*
*/
__pyx_t_12 = (__pyx_v_t == (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_b])]));
if (__pyx_t_12) {
/* "sklearn/tree/_tree.pyx":816
* t = X_i[X_argsorted_i[a]] + random_state.rand() * (X_i[X_argsorted_i[b]] - X_i[X_argsorted_i[a]])
* if t == X_i[X_argsorted_i[b]]:
* t = X_i[X_argsorted_i[a]] # <<<<<<<<<<<<<<
*
* # Find the sample just greater than t
*/
__pyx_v_t = (__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_v_a])]);
goto __pyx_L15;
}
__pyx_L15:;
/* "sklearn/tree/_tree.pyx":819
*
* # Find the sample just greater than t
* c = a + 1 # <<<<<<<<<<<<<<
*
* while True:
*/
__pyx_t_7 = PyInt_FromLong((__pyx_v_a + 1)); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 819; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__Pyx_XDECREF(__pyx_v_c);
__pyx_v_c = __pyx_t_7;
__pyx_t_7 = 0;
/* "sklearn/tree/_tree.pyx":821
* c = a + 1
*
* while True: # <<<<<<<<<<<<<<
* if sample_mask_ptr[X_argsorted_i[c]] != 0:
* if X_i[X_argsorted_i[c]] > t or c == b:
*/
while (1) {
if (!1) break;
/* "sklearn/tree/_tree.pyx":822
*
* while True:
* if sample_mask_ptr[X_argsorted_i[c]] != 0: # <<<<<<<<<<<<<<
* if X_i[X_argsorted_i[c]] > t or c == b:
* break
*/
__pyx_t_13 = __Pyx_PyIndex_AsSsize_t(__pyx_v_c); if (unlikely((__pyx_t_13 == (Py_ssize_t)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_12 = ((__pyx_v_sample_mask_ptr[(__pyx_v_X_argsorted_i[__pyx_t_13])]) != 0);
if (__pyx_t_12) {
/* "sklearn/tree/_tree.pyx":823
* while True:
* if sample_mask_ptr[X_argsorted_i[c]] != 0:
* if X_i[X_argsorted_i[c]] > t or c == b: # <<<<<<<<<<<<<<
* break
*
*/
__pyx_t_13 = __Pyx_PyIndex_AsSsize_t(__pyx_v_c); if (unlikely((__pyx_t_13 == (Py_ssize_t)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_12 = ((__pyx_v_X_i[(__pyx_v_X_argsorted_i[__pyx_t_13])]) > __pyx_v_t);
if (!__pyx_t_12) {
__pyx_t_7 = PyInt_FromLong(__pyx_v_b); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_1 = PyObject_RichCompare(__pyx_v_c, __pyx_t_7, Py_EQ); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_4 = __Pyx_PyObject_IsTrue(__pyx_t_1); if (unlikely(__pyx_t_4 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_11 = __pyx_t_4;
} else {
__pyx_t_11 = __pyx_t_12;
}
if (__pyx_t_11) {
/* "sklearn/tree/_tree.pyx":824
* if sample_mask_ptr[X_argsorted_i[c]] != 0:
* if X_i[X_argsorted_i[c]] > t or c == b:
* break # <<<<<<<<<<<<<<
*
* c += 1
*/
goto __pyx_L17_break;
goto __pyx_L19;
}
__pyx_L19:;
goto __pyx_L18;
}
__pyx_L18:;
/* "sklearn/tree/_tree.pyx":826
* break
*
* c += 1 # <<<<<<<<<<<<<<
*
* # Better than the best so far?
*/
__pyx_t_1 = PyNumber_InPlaceAdd(__pyx_v_c, __pyx_int_1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_v_c);
__pyx_v_c = __pyx_t_1;
__pyx_t_1 = 0;
}
__pyx_L17_break:;
/* "sklearn/tree/_tree.pyx":829
*
* # Better than the best so far?
* n_left = criterion.update(0, c, y_ptr, X_argsorted_i, sample_mask_ptr) # <<<<<<<<<<<<<<
* error = criterion.eval()
*
*/
__pyx_t_10 = __Pyx_PyInt_AsInt(__pyx_v_c); if (unlikely((__pyx_t_10 == (int)-1) && PyErr_Occurred())) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_t_1 = PyInt_FromLong(((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->update(__pyx_v_criterion, 0, __pyx_t_10, __pyx_v_y_ptr, __pyx_v_X_argsorted_i, __pyx_v_sample_mask_ptr)); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_v_n_left);
__pyx_v_n_left = __pyx_t_1;
__pyx_t_1 = 0;
/* "sklearn/tree/_tree.pyx":830
* # Better than the best so far?
* n_left = criterion.update(0, c, y_ptr, X_argsorted_i, sample_mask_ptr)
* error = criterion.eval() # <<<<<<<<<<<<<<
*
* if n_left < min_leaf or (n_samples - n_left) < min_leaf:
*/
__pyx_v_error = ((struct __pyx_vtabstruct_7sklearn_4tree_5_tree_Criterion *)__pyx_v_criterion->__pyx_vtab)->eval(__pyx_v_criterion);
/* "sklearn/tree/_tree.pyx":832
* error = criterion.eval()
*
* if n_left < min_leaf or (n_samples - n_left) < min_leaf: # <<<<<<<<<<<<<<
* continue
*
*/
__pyx_t_1 = PyInt_FromLong(__pyx_v_min_leaf); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_7 = PyObject_RichCompare(__pyx_v_n_left, __pyx_t_1, Py_LT); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__pyx_t_11 = __Pyx_PyObject_IsTrue(__pyx_t_7); if (unlikely(__pyx_t_11 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
if (!__pyx_t_11) {
__pyx_t_7 = PyInt_FromLong(__pyx_v_n_samples); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_1 = PyNumber_Subtract(__pyx_t_7, __pyx_v_n_left); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_7 = PyInt_FromLong(__pyx_v_min_leaf); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_5 = PyObject_RichCompare(__pyx_t_1, __pyx_t_7, Py_LT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
__Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
__pyx_t_12 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_12 < 0)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_t_4 = __pyx_t_12;
} else {
__pyx_t_4 = __pyx_t_11;
}
if (__pyx_t_4) {
/* "sklearn/tree/_tree.pyx":833
*
* if n_left < min_leaf or (n_samples - n_left) < min_leaf:
* continue # <<<<<<<<<<<<<<
*
* if error < best_error:
*/
goto __pyx_L8_continue;
goto __pyx_L20;
}
__pyx_L20:;
/* "sklearn/tree/_tree.pyx":835
* continue
*
* if error < best_error: # <<<<<<<<<<<<<<
* best_i = i
* best_t = t
*/
__pyx_t_4 = (__pyx_v_error < __pyx_v_best_error);
if (__pyx_t_4) {
/* "sklearn/tree/_tree.pyx":836
*
* if error < best_error:
* best_i = i # <<<<<<<<<<<<<<
* best_t = t
* best_error = error
*/
__pyx_v_best_i = __pyx_v_i;
/* "sklearn/tree/_tree.pyx":837
* if error < best_error:
* best_i = i
* best_t = t # <<<<<<<<<<<<<<
* best_error = error
*
*/
__pyx_v_best_t = __pyx_v_t;
/* "sklearn/tree/_tree.pyx":838
* best_i = i
* best_t = t
* best_error = error # <<<<<<<<<<<<<<
*
* return best_i, best_t, best_error, initial_error
*/
__pyx_v_best_error = __pyx_v_error;
goto __pyx_L21;
}
__pyx_L21:;
__pyx_L8_continue:;
}
__Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
/* "sklearn/tree/_tree.pyx":840
* best_error = error
*
* return best_i, best_t, best_error, initial_error # <<<<<<<<<<<<<<
*/
__Pyx_XDECREF(__pyx_r);
__pyx_t_6 = PyInt_FromLong(__pyx_v_best_i); if (unlikely(!__pyx_t_6)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_6);
__pyx_t_5 = PyFloat_FromDouble(__pyx_v_best_t); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_7 = PyFloat_FromDouble(__pyx_v_best_error); if (unlikely(!__pyx_t_7)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_7);
__pyx_t_1 = PyFloat_FromDouble(__pyx_v_initial_error); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyTuple_New(4); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_2));
PyTuple_SET_ITEM(__pyx_t_2, 0, __pyx_t_6);
__Pyx_GIVEREF(__pyx_t_6);
PyTuple_SET_ITEM(__pyx_t_2, 1, __pyx_t_5);
__Pyx_GIVEREF(__pyx_t_5);
PyTuple_SET_ITEM(__pyx_t_2, 2, __pyx_t_7);
__Pyx_GIVEREF(__pyx_t_7);
PyTuple_SET_ITEM(__pyx_t_2, 3, __pyx_t_1);
__Pyx_GIVEREF(__pyx_t_1);
__pyx_t_6 = 0;
__pyx_t_5 = 0;
__pyx_t_7 = 0;
__pyx_t_1 = 0;
__pyx_r = ((PyObject *)__pyx_t_2);
__pyx_t_2 = 0;
goto __pyx_L0;
__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_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_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X_argsorted);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_y);
__Pyx_ErrRestore(__pyx_type, __pyx_value, __pyx_tb);}
__Pyx_AddTraceback("sklearn.tree._tree._find_best_random_split", __pyx_clineno, __pyx_lineno, __pyx_filename);
__pyx_r = NULL;
goto __pyx_L2;
__pyx_L0:;
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_X_argsorted);
__Pyx_SafeReleaseBuffer(&__pyx_bstruct_y);
__pyx_L2:;
__Pyx_XDECREF(__pyx_v_features);
__Pyx_XDECREF(__pyx_v_c);
__Pyx_XDECREF(__pyx_v_n_left);
__Pyx_XGIVEREF(__pyx_r);
__Pyx_RefNannyFinishContext();
return __pyx_r;
}
/* "numpy.pxd":190
* # 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.
*/
static CYTHON_UNUSED int __pyx_pf_5numpy_7ndarray___getbuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info, int __pyx_v_flags); /*proto*/
static CYTHON_UNUSED int __pyx_pf_5numpy_7ndarray___getbuffer__(PyObject *__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__");
if (__pyx_v_info != NULL) {
__pyx_v_info->obj = Py_None; __Pyx_INCREF(Py_None);
__Pyx_GIVEREF(__pyx_v_info->obj);
}
/* "numpy.pxd":196
* # of flags
*
* if info == NULL: return # <<<<<<<<<<<<<<
*
* cdef int copy_shape, i, ndim
*/
__pyx_t_1 = (__pyx_v_info == NULL);
if (__pyx_t_1) {
__pyx_r = 0;
goto __pyx_L0;
goto __pyx_L5;
}
__pyx_L5:;
/* "numpy.pxd":199
*
* 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;
/* "numpy.pxd":200
* 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);
/* "numpy.pxd":202
* 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(((PyArrayObject *)__pyx_v_self));
/* "numpy.pxd":204
* 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)));
if (__pyx_t_1) {
/* "numpy.pxd":205
*
* if sizeof(npy_intp) != sizeof(Py_ssize_t):
* copy_shape = 1 # <<<<<<<<<<<<<<
* else:
* copy_shape = 0
*/
__pyx_v_copy_shape = 1;
goto __pyx_L6;
}
/*else*/ {
/* "numpy.pxd":207
* copy_shape = 1
* else:
* copy_shape = 0 # <<<<<<<<<<<<<<
*
* if ((flags & pybuf.PyBUF_C_CONTIGUOUS == pybuf.PyBUF_C_CONTIGUOUS)
*/
__pyx_v_copy_shape = 0;
}
__pyx_L6:;
/* "numpy.pxd":209
* 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);
if (__pyx_t_1) {
/* "numpy.pxd":210
*
* 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(((PyArrayObject *)__pyx_v_self), NPY_C_CONTIGUOUS));
__pyx_t_3 = __pyx_t_2;
} else {
__pyx_t_3 = __pyx_t_1;
}
if (__pyx_t_3) {
/* "numpy.pxd":211
* 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 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_2), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 211; __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[1]; __pyx_lineno = 211; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
goto __pyx_L7;
}
__pyx_L7:;
/* "numpy.pxd":213
* 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);
if (__pyx_t_3) {
/* "numpy.pxd":214
*
* 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(((PyArrayObject *)__pyx_v_self), NPY_F_CONTIGUOUS));
__pyx_t_2 = __pyx_t_1;
} else {
__pyx_t_2 = __pyx_t_3;
}
if (__pyx_t_2) {
/* "numpy.pxd":215
* 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 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_4), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __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[1]; __pyx_lineno = 215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
goto __pyx_L8;
}
__pyx_L8:;
/* "numpy.pxd":217
* 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(((PyArrayObject *)__pyx_v_self));
/* "numpy.pxd":218
*
* 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;
/* "numpy.pxd":219
* 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.
*/
if (__pyx_v_copy_shape) {
/* "numpy.pxd":222
* # 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)));
/* "numpy.pxd":223
* # 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);
/* "numpy.pxd":224
* 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;
/* "numpy.pxd":225
* 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(((PyArrayObject *)__pyx_v_self))[__pyx_v_i]);
/* "numpy.pxd":226
* 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(((PyArrayObject *)__pyx_v_self))[__pyx_v_i]);
}
goto __pyx_L9;
}
/*else*/ {
/* "numpy.pxd":228
* 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(((PyArrayObject *)__pyx_v_self)));
/* "numpy.pxd":229
* 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(((PyArrayObject *)__pyx_v_self)));
}
__pyx_L9:;
/* "numpy.pxd":230
* 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;
/* "numpy.pxd":231
* 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(((PyArrayObject *)__pyx_v_self));
/* "numpy.pxd":232
* info.suboffsets = NULL
* info.itemsize = PyArray_ITEMSIZE(self)
* info.readonly = not PyArray_ISWRITEABLE(self) # <<<<<<<<<<<<<<
*
* cdef int t
*/
__pyx_v_info->readonly = (!PyArray_ISWRITEABLE(((PyArrayObject *)__pyx_v_self)));
/* "numpy.pxd":235
*
* cdef int t
* cdef char* f = NULL # <<<<<<<<<<<<<<
* cdef dtype descr = self.descr
* cdef list stack
*/
__pyx_v_f = NULL;
/* "numpy.pxd":236
* cdef int t
* cdef char* f = NULL
* cdef dtype descr = self.descr # <<<<<<<<<<<<<<
* cdef list stack
* cdef int offset
*/
__Pyx_INCREF(((PyObject *)((PyArrayObject *)__pyx_v_self)->descr));
__pyx_v_descr = ((PyArrayObject *)__pyx_v_self)->descr;
/* "numpy.pxd":240
* cdef int offset
*
* cdef bint hasfields = PyDataType_HASFIELDS(descr) # <<<<<<<<<<<<<<
*
* if not hasfields and not copy_shape:
*/
__pyx_v_hasfields = PyDataType_HASFIELDS(__pyx_v_descr);
/* "numpy.pxd":242
* 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);
if (__pyx_t_2) {
__pyx_t_3 = (!__pyx_v_copy_shape);
__pyx_t_1 = __pyx_t_3;
} else {
__pyx_t_1 = __pyx_t_2;
}
if (__pyx_t_1) {
/* "numpy.pxd":244
* 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_L12;
}
/*else*/ {
/* "numpy.pxd":247
* else:
* # need to call releasebuffer
* info.obj = self # <<<<<<<<<<<<<<
*
* if not hasfields:
*/
__Pyx_INCREF(__pyx_v_self);
__Pyx_GIVEREF(__pyx_v_self);
__Pyx_GOTREF(__pyx_v_info->obj);
__Pyx_DECREF(__pyx_v_info->obj);
__pyx_v_info->obj = __pyx_v_self;
}
__pyx_L12:;
/* "numpy.pxd":249
* info.obj = self
*
* if not hasfields: # <<<<<<<<<<<<<<
* t = descr.type_num
* if ((descr.byteorder == '>' and little_endian) or
*/
__pyx_t_1 = (!__pyx_v_hasfields);
if (__pyx_t_1) {
/* "numpy.pxd":250
*
* if not hasfields:
* t = descr.type_num # <<<<<<<<<<<<<<
* if ((descr.byteorder == '>' and little_endian) or
* (descr.byteorder == '<' and not little_endian)):
*/
__pyx_v_t = __pyx_v_descr->type_num;
/* "numpy.pxd":251
* if not hasfields:
* t = descr.type_num
* if ((descr.byteorder == '>' and little_endian) or # <<<<<<<<<<<<<<
* (descr.byteorder == '<' and not little_endian)):
* raise ValueError(u"Non-native byte order not supported")
*/
__pyx_t_1 = (__pyx_v_descr->byteorder == '>');
if (__pyx_t_1) {
__pyx_t_2 = __pyx_v_little_endian;
} else {
__pyx_t_2 = __pyx_t_1;
}
if (!__pyx_t_2) {
/* "numpy.pxd":252
* t = descr.type_num
* if ((descr.byteorder == '>' and little_endian) or
* (descr.byteorder == '<' 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 == '<');
if (__pyx_t_1) {
__pyx_t_3 = (!__pyx_v_little_endian);
__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) {
/* "numpy.pxd":253
* if ((descr.byteorder == '>' and little_endian) or
* (descr.byteorder == '<' 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 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_6), NULL); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 253; __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[1]; __pyx_lineno = 253; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
goto __pyx_L14;
}
__pyx_L14:;
/* "numpy.pxd":254
* (descr.byteorder == '<' 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_BYTE);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__b;
goto __pyx_L15;
}
/* "numpy.pxd":255
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_UBYTE);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__B;
goto __pyx_L15;
}
/* "numpy.pxd":256
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_SHORT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__h;
goto __pyx_L15;
}
/* "numpy.pxd":257
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_USHORT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__H;
goto __pyx_L15;
}
/* "numpy.pxd":258
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_INT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__i;
goto __pyx_L15;
}
/* "numpy.pxd":259
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_UINT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__I;
goto __pyx_L15;
}
/* "numpy.pxd":260
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_LONG);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__l;
goto __pyx_L15;
}
/* "numpy.pxd":261
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_ULONG);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__L;
goto __pyx_L15;
}
/* "numpy.pxd":262
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_LONGLONG);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__q;
goto __pyx_L15;
}
/* "numpy.pxd":263
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_ULONGLONG);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__Q;
goto __pyx_L15;
}
/* "numpy.pxd":264
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_FLOAT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__f;
goto __pyx_L15;
}
/* "numpy.pxd":265
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_DOUBLE);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__d;
goto __pyx_L15;
}
/* "numpy.pxd":266
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_LONGDOUBLE);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__g;
goto __pyx_L15;
}
/* "numpy.pxd":267
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_CFLOAT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__Zf;
goto __pyx_L15;
}
/* "numpy.pxd":268
* 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"
*/
__pyx_t_1 = (__pyx_v_t == NPY_CDOUBLE);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__Zd;
goto __pyx_L15;
}
/* "numpy.pxd":269
* 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:
*/
__pyx_t_1 = (__pyx_v_t == NPY_CLONGDOUBLE);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__Zg;
goto __pyx_L15;
}
/* "numpy.pxd":270
* 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)
*/
__pyx_t_1 = (__pyx_v_t == NPY_OBJECT);
if (__pyx_t_1) {
__pyx_v_f = __pyx_k__O;
goto __pyx_L15;
}
/*else*/ {
/* "numpy.pxd":272
* 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 = PyInt_FromLong(__pyx_v_t); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 272; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__pyx_t_8 = PyNumber_Remainder(((PyObject *)__pyx_kp_u_7), __pyx_t_4); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 272; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__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[1]; __pyx_lineno = 272; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_4));
PyTuple_SET_ITEM(__pyx_t_4, 0, ((PyObject *)__pyx_t_8));
__Pyx_GIVEREF(((PyObject *)__pyx_t_8));
__pyx_t_8 = 0;
__pyx_t_8 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_t_4), NULL); if (unlikely(!__pyx_t_8)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 272; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_8);
__Pyx_DECREF(((PyObject *)__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[1]; __pyx_lineno = 272; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_L15:;
/* "numpy.pxd":273
* else:
* raise ValueError(u"unknown dtype code in numpy.pxd (%d)" % t)
* info.format = f # <<<<<<<<<<<<<<
* return
* else:
*/
__pyx_v_info->format = __pyx_v_f;
/* "numpy.pxd":274
* 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;
goto __pyx_L13;
}
/*else*/ {
/* "numpy.pxd":276
* return
* else:
* info.format = <char*>stdlib.malloc(_buffer_format_string_len) # <<<<<<<<<<<<<<
* info.format[0] = '^' # Native data types, manual alignment
* offset = 0
*/
__pyx_v_info->format = ((char *)malloc(255));
/* "numpy.pxd":277
* else:
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
* info.format[0] = '^' # Native data types, manual alignment # <<<<<<<<<<<<<<
* offset = 0
* f = _util_dtypestring(descr, info.format + 1,
*/
(__pyx_v_info->format[0]) = '^';
/* "numpy.pxd":278
* info.format = <char*>stdlib.malloc(_buffer_format_string_len)
* info.format[0] = '^' # Native data types, manual alignment
* offset = 0 # <<<<<<<<<<<<<<
* f = _util_dtypestring(descr, info.format + 1,
* info.format + _buffer_format_string_len,
*/
__pyx_v_offset = 0;
/* "numpy.pxd":281
* f = _util_dtypestring(descr, info.format + 1,
* info.format + _buffer_format_string_len,
* &offset) # <<<<<<<<<<<<<<
* f[0] = 0 # Terminate format string
*
*/
__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[1]; __pyx_lineno = 279; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_v_f = __pyx_t_9;
/* "numpy.pxd":282
* info.format + _buffer_format_string_len,
* &offset)
* f[0] = 0 # Terminate format string # <<<<<<<<<<<<<<
*
* def __releasebuffer__(ndarray self, Py_buffer* info):
*/
(__pyx_v_f[0]) = 0;
}
__pyx_L13:;
__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;
}
/* "numpy.pxd":284
* f[0] = 0 # Terminate format string
*
* def __releasebuffer__(ndarray self, Py_buffer* info): # <<<<<<<<<<<<<<
* if PyArray_HASFIELDS(self):
* stdlib.free(info.format)
*/
static CYTHON_UNUSED void __pyx_pf_5numpy_7ndarray_1__releasebuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info); /*proto*/
static CYTHON_UNUSED void __pyx_pf_5numpy_7ndarray_1__releasebuffer__(PyObject *__pyx_v_self, Py_buffer *__pyx_v_info) {
__Pyx_RefNannyDeclarations
int __pyx_t_1;
__Pyx_RefNannySetupContext("__releasebuffer__");
/* "numpy.pxd":285
*
* 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(((PyArrayObject *)__pyx_v_self));
if (__pyx_t_1) {
/* "numpy.pxd":286
* 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_L5;
}
__pyx_L5:;
/* "numpy.pxd":287
* 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)));
if (__pyx_t_1) {
/* "numpy.pxd":288
* 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_L6;
}
__pyx_L6:;
__Pyx_RefNannyFinishContext();
}
/* "numpy.pxd":764
* 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");
/* "numpy.pxd":765
*
* 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[1]; __pyx_lineno = 765; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__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;
}
/* "numpy.pxd":767
* 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");
/* "numpy.pxd":768
*
* 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[1]; __pyx_lineno = 768; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__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;
}
/* "numpy.pxd":770
* 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");
/* "numpy.pxd":771
*
* 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[1]; __pyx_lineno = 771; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__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;
}
/* "numpy.pxd":773
* 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");
/* "numpy.pxd":774
*
* 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[1]; __pyx_lineno = 774; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__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;
}
/* "numpy.pxd":776
* 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");
/* "numpy.pxd":777
*
* 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[1]; __pyx_lineno = 777; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_r = __pyx_t_1;
__pyx_t_1 = 0;
goto __pyx_L0;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
goto __pyx_L0;
__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;
}
/* "numpy.pxd":779
* 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;
PyObject *__pyx_t_5 = NULL;
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");
/* "numpy.pxd":786
* 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;
/* "numpy.pxd":787
* 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);
/* "numpy.pxd":790
* cdef tuple fields
*
* for childname in descr.names: # <<<<<<<<<<<<<<
* fields = descr.fields[childname]
* child, new_offset = fields
*/
if (unlikely(((PyObject *)__pyx_v_descr->names) == Py_None)) {
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable"); {__pyx_filename = __pyx_f[1]; __pyx_lineno = 790; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_t_1 = ((PyObject *)__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;
__pyx_t_3 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_2); __Pyx_INCREF(__pyx_t_3); __pyx_t_2++;
__Pyx_XDECREF(__pyx_v_childname);
__pyx_v_childname = __pyx_t_3;
__pyx_t_3 = 0;
/* "numpy.pxd":791
*
* 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 (!__pyx_t_3) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 791; __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 tuple, got %.200s", Py_TYPE(__pyx_t_3)->tp_name), 0))) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 791; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_XDECREF(((PyObject *)__pyx_v_fields));
__pyx_v_fields = ((PyObject*)__pyx_t_3);
__pyx_t_3 = 0;
/* "numpy.pxd":792
* for childname in descr.names:
* fields = descr.fields[childname]
* child, new_offset = fields # <<<<<<<<<<<<<<
*
* if (end - f) - (new_offset - offset[0]) < 15:
*/
if (likely(PyTuple_CheckExact(((PyObject *)__pyx_v_fields)))) {
PyObject* sequence = ((PyObject *)__pyx_v_fields);
if (unlikely(PyTuple_GET_SIZE(sequence) != 2)) {
if (PyTuple_GET_SIZE(sequence) > 2) __Pyx_RaiseTooManyValuesError(2);
else __Pyx_RaiseNeedMoreValuesError(PyTuple_GET_SIZE(sequence));
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 792; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__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_UnpackTupleError(((PyObject *)__pyx_v_fields), 2);
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 792; __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[1]; __pyx_lineno = 792; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_XDECREF(((PyObject *)__pyx_v_child));
__pyx_v_child = ((PyArray_Descr *)__pyx_t_3);
__pyx_t_3 = 0;
__Pyx_XDECREF(__pyx_v_new_offset);
__pyx_v_new_offset = __pyx_t_4;
__pyx_t_4 = 0;
/* "numpy.pxd":794
* child, new_offset = fields
*
* if (end - f) - (new_offset - offset[0]) < 15: # <<<<<<<<<<<<<<
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
*
*/
__pyx_t_4 = PyInt_FromLong((__pyx_v_end - __pyx_v_f)); if (unlikely(!__pyx_t_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_4);
__pyx_t_3 = PyInt_FromLong((__pyx_v_offset[0])); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyNumber_Subtract(__pyx_v_new_offset, __pyx_t_3); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_3 = PyNumber_Subtract(__pyx_t_4, __pyx_t_5); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
__pyx_t_5 = PyObject_RichCompare(__pyx_t_3, __pyx_int_15, Py_LT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 794; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
/* "numpy.pxd":795
*
* if (end - f) - (new_offset - offset[0]) < 15:
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd") # <<<<<<<<<<<<<<
*
* if ((child.byteorder == '>' and little_endian) or
*/
__pyx_t_5 = PyObject_Call(__pyx_builtin_RuntimeError, ((PyObject *)__pyx_k_tuple_9), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 795; __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[1]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
goto __pyx_L5;
}
__pyx_L5:;
/* "numpy.pxd":797
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd")
*
* if ((child.byteorder == '>' and little_endian) or # <<<<<<<<<<<<<<
* (child.byteorder == '<' and not little_endian)):
* raise ValueError(u"Non-native byte order not supported")
*/
__pyx_t_6 = (__pyx_v_child->byteorder == '>');
if (__pyx_t_6) {
__pyx_t_7 = __pyx_v_little_endian;
} else {
__pyx_t_7 = __pyx_t_6;
}
if (!__pyx_t_7) {
/* "numpy.pxd":798
*
* if ((child.byteorder == '>' and little_endian) or
* (child.byteorder == '<' 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 == '<');
if (__pyx_t_6) {
__pyx_t_8 = (!__pyx_v_little_endian);
__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) {
/* "numpy.pxd":799
* if ((child.byteorder == '>' and little_endian) or
* (child.byteorder == '<' 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_5 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_k_tuple_10), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 799; __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[1]; __pyx_lineno = 799; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
goto __pyx_L6;
}
__pyx_L6:;
/* "numpy.pxd":809
*
* # Output padding bytes
* while offset[0] < new_offset: # <<<<<<<<<<<<<<
* f[0] = 120 # "x"; pad byte
* f += 1
*/
while (1) {
__pyx_t_5 = PyInt_FromLong((__pyx_v_offset[0])); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 809; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_t_5, __pyx_v_new_offset, Py_LT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 809; __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[1]; __pyx_lineno = 809; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (!__pyx_t_6) break;
/* "numpy.pxd":810
* # Output padding bytes
* while offset[0] < new_offset:
* f[0] = 120 # "x"; pad byte # <<<<<<<<<<<<<<
* f += 1
* offset[0] += 1
*/
(__pyx_v_f[0]) = 120;
/* "numpy.pxd":811
* while offset[0] < new_offset:
* f[0] = 120 # "x"; pad byte
* f += 1 # <<<<<<<<<<<<<<
* offset[0] += 1
*
*/
__pyx_v_f = (__pyx_v_f + 1);
/* "numpy.pxd":812
* 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);
}
/* "numpy.pxd":814
* 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);
/* "numpy.pxd":816
* 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));
if (__pyx_t_6) {
/* "numpy.pxd":817
*
* if not PyDataType_HASFIELDS(child):
* t = child.type_num # <<<<<<<<<<<<<<
* if end - f < 5:
* raise RuntimeError(u"Format string allocated too short.")
*/
__pyx_t_3 = PyInt_FromLong(__pyx_v_child->type_num); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 817; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__Pyx_XDECREF(__pyx_v_t);
__pyx_v_t = __pyx_t_3;
__pyx_t_3 = 0;
/* "numpy.pxd":818
* 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);
if (__pyx_t_6) {
/* "numpy.pxd":819
* 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_3 = PyObject_Call(__pyx_builtin_RuntimeError, ((PyObject *)__pyx_k_tuple_12), NULL); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 819; __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[1]; __pyx_lineno = 819; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
goto __pyx_L10;
}
__pyx_L10:;
/* "numpy.pxd":822
*
* # 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_3 = PyInt_FromLong(NPY_BYTE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 822; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 98;
goto __pyx_L11;
}
/* "numpy.pxd":823
* # 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_5 = PyInt_FromLong(NPY_UBYTE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 823; __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[1]; __pyx_lineno = 823; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 66;
goto __pyx_L11;
}
/* "numpy.pxd":824
* 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_3 = PyInt_FromLong(NPY_SHORT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 824; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 824; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 824; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 104;
goto __pyx_L11;
}
/* "numpy.pxd":825
* 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_5 = PyInt_FromLong(NPY_USHORT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 825; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 825; __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[1]; __pyx_lineno = 825; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 72;
goto __pyx_L11;
}
/* "numpy.pxd":826
* 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_3 = PyInt_FromLong(NPY_INT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 826; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 105;
goto __pyx_L11;
}
/* "numpy.pxd":827
* 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_5 = PyInt_FromLong(NPY_UINT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 827; __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[1]; __pyx_lineno = 827; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 73;
goto __pyx_L11;
}
/* "numpy.pxd":828
* 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_3 = PyInt_FromLong(NPY_LONG); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 828; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 108;
goto __pyx_L11;
}
/* "numpy.pxd":829
* 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_5 = PyInt_FromLong(NPY_ULONG); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 829; __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[1]; __pyx_lineno = 829; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 76;
goto __pyx_L11;
}
/* "numpy.pxd":830
* 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_3 = PyInt_FromLong(NPY_LONGLONG); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 830; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 113;
goto __pyx_L11;
}
/* "numpy.pxd":831
* 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_5 = PyInt_FromLong(NPY_ULONGLONG); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 831; __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[1]; __pyx_lineno = 831; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 81;
goto __pyx_L11;
}
/* "numpy.pxd":832
* 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_3 = PyInt_FromLong(NPY_FLOAT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 832; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 102;
goto __pyx_L11;
}
/* "numpy.pxd":833
* 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_5 = PyInt_FromLong(NPY_DOUBLE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 833; __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[1]; __pyx_lineno = 833; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 100;
goto __pyx_L11;
}
/* "numpy.pxd":834
* 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_3 = PyInt_FromLong(NPY_LONGDOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 834; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 103;
goto __pyx_L11;
}
/* "numpy.pxd":835
* 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_5 = PyInt_FromLong(NPY_CFLOAT); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 835; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 835; __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[1]; __pyx_lineno = 835; __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]) = 102;
__pyx_v_f = (__pyx_v_f + 1);
goto __pyx_L11;
}
/* "numpy.pxd":836
* 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_3 = PyInt_FromLong(NPY_CDOUBLE); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 836; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 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;
}
/* "numpy.pxd":837
* 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_5 = PyInt_FromLong(NPY_CLONGDOUBLE); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 837; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__pyx_t_3 = PyObject_RichCompare(__pyx_v_t, __pyx_t_5, Py_EQ); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 837; __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[1]; __pyx_lineno = 837; __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]) = 103;
__pyx_v_f = (__pyx_v_f + 1);
goto __pyx_L11;
}
/* "numpy.pxd":838
* 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_3 = PyInt_FromLong(NPY_OBJECT); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_3);
__pyx_t_5 = PyObject_RichCompare(__pyx_v_t, __pyx_t_3, Py_EQ); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
__pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_5); if (unlikely(__pyx_t_6 < 0)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 838; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
if (__pyx_t_6) {
(__pyx_v_f[0]) = 79;
goto __pyx_L11;
}
/*else*/ {
/* "numpy.pxd":840
* 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_5 = PyNumber_Remainder(((PyObject *)__pyx_kp_u_7), __pyx_v_t); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_5));
__pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_3));
PyTuple_SET_ITEM(__pyx_t_3, 0, ((PyObject *)__pyx_t_5));
__Pyx_GIVEREF(((PyObject *)__pyx_t_5));
__pyx_t_5 = 0;
__pyx_t_5 = PyObject_Call(__pyx_builtin_ValueError, ((PyObject *)__pyx_t_3), NULL); if (unlikely(!__pyx_t_5)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_5);
__Pyx_DECREF(((PyObject *)__pyx_t_3)); __pyx_t_3 = 0;
__Pyx_Raise(__pyx_t_5, 0, 0, 0);
__Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
{__pyx_filename = __pyx_f[1]; __pyx_lineno = 840; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
}
__pyx_L11:;
/* "numpy.pxd":841
* 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*/ {
/* "numpy.pxd":845
* # 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[1]; __pyx_lineno = 845; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_v_f = __pyx_t_11;
}
__pyx_L9:;
}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "numpy.pxd":846
* # so don't output it
* f = _util_dtypestring(child, f, end, offset)
* return f # <<<<<<<<<<<<<<
*
*
*/
__pyx_r = __pyx_v_f;
goto __pyx_L0;
__pyx_r = 0;
goto __pyx_L0;
__pyx_L1_error:;
__Pyx_XDECREF(__pyx_t_1);
__Pyx_XDECREF(__pyx_t_3);
__Pyx_XDECREF(__pyx_t_4);
__Pyx_XDECREF(__pyx_t_5);
__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;
}
/* "numpy.pxd":961
*
*
* 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;
__Pyx_RefNannySetupContext("set_array_base");
/* "numpy.pxd":963
* 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);
if (__pyx_t_1) {
/* "numpy.pxd":964
* 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*/ {
/* "numpy.pxd":966
* 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);
/* "numpy.pxd":967
* 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:;
/* "numpy.pxd":968
* 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);
/* "numpy.pxd":969
* 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;
__Pyx_RefNannyFinishContext();
}
/* "numpy.pxd":971
* 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");
/* "numpy.pxd":972
*
* cdef inline object get_array_base(ndarray arr):
* if arr.base is NULL: # <<<<<<<<<<<<<<
* return None
* else:
*/
__pyx_t_1 = (__pyx_v_arr->base == NULL);
if (__pyx_t_1) {
/* "numpy.pxd":973
* 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;
goto __pyx_L3;
}
/*else*/ {
/* "numpy.pxd":975
* 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;
}
__pyx_L3:;
__pyx_r = Py_None; __Pyx_INCREF(Py_None);
__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, PyObject *a, PyObject *k) {
struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *p;
PyObject *o = (*t->tp_alloc)(t, 0);
if (!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) {
(*Py_TYPE(o)->tp_free)(o);
}
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_Criterion[] = {
{__Pyx_NAMESTR("init_value"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_9Criterion_init_value, METH_NOARGS, __Pyx_DOCSTR(__pyx_doc_7sklearn_4tree_5_tree_9Criterion_init_value)},
{0, 0, 0, 0}
};
static PyNumberMethods __pyx_tp_as_number_Criterion = {
0, /*nb_add*/
0, /*nb_subtract*/
0, /*nb_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_divide*/
#endif
0, /*nb_remainder*/
0, /*nb_divmod*/
0, /*nb_power*/
0, /*nb_negative*/
0, /*nb_positive*/
0, /*nb_absolute*/
0, /*nb_nonzero*/
0, /*nb_invert*/
0, /*nb_lshift*/
0, /*nb_rshift*/
0, /*nb_and*/
0, /*nb_xor*/
0, /*nb_or*/
#if PY_MAJOR_VERSION < 3
0, /*nb_coerce*/
#endif
0, /*nb_int*/
#if PY_MAJOR_VERSION < 3
0, /*nb_long*/
#else
0, /*reserved*/
#endif
0, /*nb_float*/
#if PY_MAJOR_VERSION < 3
0, /*nb_oct*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*nb_hex*/
#endif
0, /*nb_inplace_add*/
0, /*nb_inplace_subtract*/
0, /*nb_inplace_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_inplace_divide*/
#endif
0, /*nb_inplace_remainder*/
0, /*nb_inplace_power*/
0, /*nb_inplace_lshift*/
0, /*nb_inplace_rshift*/
0, /*nb_inplace_and*/
0, /*nb_inplace_xor*/
0, /*nb_inplace_or*/
0, /*nb_floor_divide*/
0, /*nb_true_divide*/
0, /*nb_inplace_floor_divide*/
0, /*nb_inplace_true_divide*/
#if PY_VERSION_HEX >= 0x02050000
0, /*nb_index*/
#endif
};
static PySequenceMethods __pyx_tp_as_sequence_Criterion = {
0, /*sq_length*/
0, /*sq_concat*/
0, /*sq_repeat*/
0, /*sq_item*/
0, /*sq_slice*/
0, /*sq_ass_item*/
0, /*sq_ass_slice*/
0, /*sq_contains*/
0, /*sq_inplace_concat*/
0, /*sq_inplace_repeat*/
};
static PyMappingMethods __pyx_tp_as_mapping_Criterion = {
0, /*mp_length*/
0, /*mp_subscript*/
0, /*mp_ass_subscript*/
};
static PyBufferProcs __pyx_tp_as_buffer_Criterion = {
#if PY_MAJOR_VERSION < 3
0, /*bf_getreadbuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getwritebuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getsegcount*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getcharbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_getbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_releasebuffer*/
#endif
};
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*/
&__pyx_tp_as_number_Criterion, /*tp_as_number*/
&__pyx_tp_as_sequence_Criterion, /*tp_as_sequence*/
&__pyx_tp_as_mapping_Criterion, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
&__pyx_tp_as_buffer_Criterion, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
__Pyx_DOCSTR("Interface for splitting criteria (regression and 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_Criterion, /*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
};
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 (!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;
p->ndarray_label_count_left = Py_None; Py_INCREF(Py_None);
p->ndarray_label_count_right = Py_None; Py_INCREF(Py_None);
p->ndarray_label_count_init = Py_None; Py_INCREF(Py_None);
return o;
}
static void __pyx_tp_dealloc_7sklearn_4tree_5_tree_ClassificationCriterion(PyObject *o) {
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *p = (struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)o;
Py_XDECREF(p->ndarray_label_count_left);
Py_XDECREF(p->ndarray_label_count_right);
Py_XDECREF(p->ndarray_label_count_init);
__pyx_tp_dealloc_7sklearn_4tree_5_tree_Criterion(o);
}
static int __pyx_tp_traverse_7sklearn_4tree_5_tree_ClassificationCriterion(PyObject *o, visitproc v, void *a) {
int e;
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *p = (struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)o;
if (__pyx_ptype_7sklearn_4tree_5_tree_Criterion->tp_traverse) {
e = __pyx_ptype_7sklearn_4tree_5_tree_Criterion->tp_traverse(o, v, a); if (e) return e;
}
if (p->ndarray_label_count_left) {
e = (*v)(p->ndarray_label_count_left, a); if (e) return e;
}
if (p->ndarray_label_count_right) {
e = (*v)(p->ndarray_label_count_right, a); if (e) return e;
}
if (p->ndarray_label_count_init) {
e = (*v)(p->ndarray_label_count_init, a); if (e) return e;
}
return 0;
}
static int __pyx_tp_clear_7sklearn_4tree_5_tree_ClassificationCriterion(PyObject *o) {
struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *p = (struct __pyx_obj_7sklearn_4tree_5_tree_ClassificationCriterion *)o;
PyObject* tmp;
if (__pyx_ptype_7sklearn_4tree_5_tree_Criterion->tp_clear) {
__pyx_ptype_7sklearn_4tree_5_tree_Criterion->tp_clear(o);
}
tmp = ((PyObject*)p->ndarray_label_count_left);
p->ndarray_label_count_left = Py_None; Py_INCREF(Py_None);
Py_XDECREF(tmp);
tmp = ((PyObject*)p->ndarray_label_count_right);
p->ndarray_label_count_right = Py_None; Py_INCREF(Py_None);
Py_XDECREF(tmp);
tmp = ((PyObject*)p->ndarray_label_count_init);
p->ndarray_label_count_init = Py_None; Py_INCREF(Py_None);
Py_XDECREF(tmp);
return 0;
}
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_ClassificationCriterion[] = {
{__Pyx_NAMESTR("init_value"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion_1init_value, METH_NOARGS, __Pyx_DOCSTR(0)},
{0, 0, 0, 0}
};
static PyNumberMethods __pyx_tp_as_number_ClassificationCriterion = {
0, /*nb_add*/
0, /*nb_subtract*/
0, /*nb_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_divide*/
#endif
0, /*nb_remainder*/
0, /*nb_divmod*/
0, /*nb_power*/
0, /*nb_negative*/
0, /*nb_positive*/
0, /*nb_absolute*/
0, /*nb_nonzero*/
0, /*nb_invert*/
0, /*nb_lshift*/
0, /*nb_rshift*/
0, /*nb_and*/
0, /*nb_xor*/
0, /*nb_or*/
#if PY_MAJOR_VERSION < 3
0, /*nb_coerce*/
#endif
0, /*nb_int*/
#if PY_MAJOR_VERSION < 3
0, /*nb_long*/
#else
0, /*reserved*/
#endif
0, /*nb_float*/
#if PY_MAJOR_VERSION < 3
0, /*nb_oct*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*nb_hex*/
#endif
0, /*nb_inplace_add*/
0, /*nb_inplace_subtract*/
0, /*nb_inplace_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_inplace_divide*/
#endif
0, /*nb_inplace_remainder*/
0, /*nb_inplace_power*/
0, /*nb_inplace_lshift*/
0, /*nb_inplace_rshift*/
0, /*nb_inplace_and*/
0, /*nb_inplace_xor*/
0, /*nb_inplace_or*/
0, /*nb_floor_divide*/
0, /*nb_true_divide*/
0, /*nb_inplace_floor_divide*/
0, /*nb_inplace_true_divide*/
#if PY_VERSION_HEX >= 0x02050000
0, /*nb_index*/
#endif
};
static PySequenceMethods __pyx_tp_as_sequence_ClassificationCriterion = {
0, /*sq_length*/
0, /*sq_concat*/
0, /*sq_repeat*/
0, /*sq_item*/
0, /*sq_slice*/
0, /*sq_ass_item*/
0, /*sq_ass_slice*/
0, /*sq_contains*/
0, /*sq_inplace_concat*/
0, /*sq_inplace_repeat*/
};
static PyMappingMethods __pyx_tp_as_mapping_ClassificationCriterion = {
0, /*mp_length*/
0, /*mp_subscript*/
0, /*mp_ass_subscript*/
};
static PyBufferProcs __pyx_tp_as_buffer_ClassificationCriterion = {
#if PY_MAJOR_VERSION < 3
0, /*bf_getreadbuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getwritebuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getsegcount*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getcharbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_getbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_releasebuffer*/
#endif
};
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*/
&__pyx_tp_as_number_ClassificationCriterion, /*tp_as_number*/
&__pyx_tp_as_sequence_ClassificationCriterion, /*tp_as_sequence*/
&__pyx_tp_as_mapping_ClassificationCriterion, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
&__pyx_tp_as_buffer_ClassificationCriterion, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
__Pyx_DOCSTR("Abstract criterion for classification.\n\n Attributes\n ----------\n n_classes : int\n The number of classes.\n\n n_samples : int\n The number of samples.\n\n label_count_left : int*\n The label counts for samples left of splitting point.\n\n label_count_right : int*\n The label counts for samples right of splitting point.\n\n label_count_init : int*\n The initial label counts for samples right of splitting point.\n Used to reset `label_count_right` for each feature.\n\n n_left : int\n The number of samples left of splitting point.\n\n n_right : int\n The number of samples right of splitting point.\n "), /*tp_doc*/
__pyx_tp_traverse_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_traverse*/
__pyx_tp_clear_7sklearn_4tree_5_tree_ClassificationCriterion, /*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*/
__pyx_pf_7sklearn_4tree_5_tree_23ClassificationCriterion___init__, /*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
};
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 (!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 PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_Gini[] = {
{0, 0, 0, 0}
};
static PyNumberMethods __pyx_tp_as_number_Gini = {
0, /*nb_add*/
0, /*nb_subtract*/
0, /*nb_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_divide*/
#endif
0, /*nb_remainder*/
0, /*nb_divmod*/
0, /*nb_power*/
0, /*nb_negative*/
0, /*nb_positive*/
0, /*nb_absolute*/
0, /*nb_nonzero*/
0, /*nb_invert*/
0, /*nb_lshift*/
0, /*nb_rshift*/
0, /*nb_and*/
0, /*nb_xor*/
0, /*nb_or*/
#if PY_MAJOR_VERSION < 3
0, /*nb_coerce*/
#endif
0, /*nb_int*/
#if PY_MAJOR_VERSION < 3
0, /*nb_long*/
#else
0, /*reserved*/
#endif
0, /*nb_float*/
#if PY_MAJOR_VERSION < 3
0, /*nb_oct*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*nb_hex*/
#endif
0, /*nb_inplace_add*/
0, /*nb_inplace_subtract*/
0, /*nb_inplace_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_inplace_divide*/
#endif
0, /*nb_inplace_remainder*/
0, /*nb_inplace_power*/
0, /*nb_inplace_lshift*/
0, /*nb_inplace_rshift*/
0, /*nb_inplace_and*/
0, /*nb_inplace_xor*/
0, /*nb_inplace_or*/
0, /*nb_floor_divide*/
0, /*nb_true_divide*/
0, /*nb_inplace_floor_divide*/
0, /*nb_inplace_true_divide*/
#if PY_VERSION_HEX >= 0x02050000
0, /*nb_index*/
#endif
};
static PySequenceMethods __pyx_tp_as_sequence_Gini = {
0, /*sq_length*/
0, /*sq_concat*/
0, /*sq_repeat*/
0, /*sq_item*/
0, /*sq_slice*/
0, /*sq_ass_item*/
0, /*sq_ass_slice*/
0, /*sq_contains*/
0, /*sq_inplace_concat*/
0, /*sq_inplace_repeat*/
};
static PyMappingMethods __pyx_tp_as_mapping_Gini = {
0, /*mp_length*/
0, /*mp_subscript*/
0, /*mp_ass_subscript*/
};
static PyBufferProcs __pyx_tp_as_buffer_Gini = {
#if PY_MAJOR_VERSION < 3
0, /*bf_getreadbuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getwritebuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getsegcount*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getcharbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_getbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_releasebuffer*/
#endif
};
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*/
&__pyx_tp_as_number_Gini, /*tp_as_number*/
&__pyx_tp_as_sequence_Gini, /*tp_as_sequence*/
&__pyx_tp_as_mapping_Gini, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
&__pyx_tp_as_buffer_Gini, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
__Pyx_DOCSTR("Gini Index splitting criteria.\n\n Gini index = \\sum_{k=0}^{K-1} pmk (1 - pmk)\n = 1 - \\sum_{k=0}^{K-1} pmk ** 2\n "), /*tp_doc*/
__pyx_tp_traverse_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_traverse*/
__pyx_tp_clear_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_clear*/
0, /*tp_richcompare*/
0, /*tp_weaklistoffset*/
0, /*tp_iter*/
0, /*tp_iternext*/
__pyx_methods_7sklearn_4tree_5_tree_Gini, /*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
};
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 (!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 PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_Entropy[] = {
{0, 0, 0, 0}
};
static PyNumberMethods __pyx_tp_as_number_Entropy = {
0, /*nb_add*/
0, /*nb_subtract*/
0, /*nb_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_divide*/
#endif
0, /*nb_remainder*/
0, /*nb_divmod*/
0, /*nb_power*/
0, /*nb_negative*/
0, /*nb_positive*/
0, /*nb_absolute*/
0, /*nb_nonzero*/
0, /*nb_invert*/
0, /*nb_lshift*/
0, /*nb_rshift*/
0, /*nb_and*/
0, /*nb_xor*/
0, /*nb_or*/
#if PY_MAJOR_VERSION < 3
0, /*nb_coerce*/
#endif
0, /*nb_int*/
#if PY_MAJOR_VERSION < 3
0, /*nb_long*/
#else
0, /*reserved*/
#endif
0, /*nb_float*/
#if PY_MAJOR_VERSION < 3
0, /*nb_oct*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*nb_hex*/
#endif
0, /*nb_inplace_add*/
0, /*nb_inplace_subtract*/
0, /*nb_inplace_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_inplace_divide*/
#endif
0, /*nb_inplace_remainder*/
0, /*nb_inplace_power*/
0, /*nb_inplace_lshift*/
0, /*nb_inplace_rshift*/
0, /*nb_inplace_and*/
0, /*nb_inplace_xor*/
0, /*nb_inplace_or*/
0, /*nb_floor_divide*/
0, /*nb_true_divide*/
0, /*nb_inplace_floor_divide*/
0, /*nb_inplace_true_divide*/
#if PY_VERSION_HEX >= 0x02050000
0, /*nb_index*/
#endif
};
static PySequenceMethods __pyx_tp_as_sequence_Entropy = {
0, /*sq_length*/
0, /*sq_concat*/
0, /*sq_repeat*/
0, /*sq_item*/
0, /*sq_slice*/
0, /*sq_ass_item*/
0, /*sq_ass_slice*/
0, /*sq_contains*/
0, /*sq_inplace_concat*/
0, /*sq_inplace_repeat*/
};
static PyMappingMethods __pyx_tp_as_mapping_Entropy = {
0, /*mp_length*/
0, /*mp_subscript*/
0, /*mp_ass_subscript*/
};
static PyBufferProcs __pyx_tp_as_buffer_Entropy = {
#if PY_MAJOR_VERSION < 3
0, /*bf_getreadbuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getwritebuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getsegcount*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getcharbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_getbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_releasebuffer*/
#endif
};
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*/
&__pyx_tp_as_number_Entropy, /*tp_as_number*/
&__pyx_tp_as_sequence_Entropy, /*tp_as_sequence*/
&__pyx_tp_as_mapping_Entropy, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
&__pyx_tp_as_buffer_Entropy, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_GC, /*tp_flags*/
__Pyx_DOCSTR("Entropy splitting criteria.\n\n Cross Entropy = - \\sum_{k=0}^{K-1} pmk log(pmk)\n "), /*tp_doc*/
__pyx_tp_traverse_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_traverse*/
__pyx_tp_clear_7sklearn_4tree_5_tree_ClassificationCriterion, /*tp_clear*/
0, /*tp_richcompare*/
0, /*tp_weaklistoffset*/
0, /*tp_iter*/
0, /*tp_iternext*/
__pyx_methods_7sklearn_4tree_5_tree_Entropy, /*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
};
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 (!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;
return o;
}
static PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_RegressionCriterion[] = {
{__Pyx_NAMESTR("init_value"), (PyCFunction)__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion_1init_value, METH_NOARGS, __Pyx_DOCSTR(0)},
{0, 0, 0, 0}
};
static PyNumberMethods __pyx_tp_as_number_RegressionCriterion = {
0, /*nb_add*/
0, /*nb_subtract*/
0, /*nb_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_divide*/
#endif
0, /*nb_remainder*/
0, /*nb_divmod*/
0, /*nb_power*/
0, /*nb_negative*/
0, /*nb_positive*/
0, /*nb_absolute*/
0, /*nb_nonzero*/
0, /*nb_invert*/
0, /*nb_lshift*/
0, /*nb_rshift*/
0, /*nb_and*/
0, /*nb_xor*/
0, /*nb_or*/
#if PY_MAJOR_VERSION < 3
0, /*nb_coerce*/
#endif
0, /*nb_int*/
#if PY_MAJOR_VERSION < 3
0, /*nb_long*/
#else
0, /*reserved*/
#endif
0, /*nb_float*/
#if PY_MAJOR_VERSION < 3
0, /*nb_oct*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*nb_hex*/
#endif
0, /*nb_inplace_add*/
0, /*nb_inplace_subtract*/
0, /*nb_inplace_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_inplace_divide*/
#endif
0, /*nb_inplace_remainder*/
0, /*nb_inplace_power*/
0, /*nb_inplace_lshift*/
0, /*nb_inplace_rshift*/
0, /*nb_inplace_and*/
0, /*nb_inplace_xor*/
0, /*nb_inplace_or*/
0, /*nb_floor_divide*/
0, /*nb_true_divide*/
0, /*nb_inplace_floor_divide*/
0, /*nb_inplace_true_divide*/
#if PY_VERSION_HEX >= 0x02050000
0, /*nb_index*/
#endif
};
static PySequenceMethods __pyx_tp_as_sequence_RegressionCriterion = {
0, /*sq_length*/
0, /*sq_concat*/
0, /*sq_repeat*/
0, /*sq_item*/
0, /*sq_slice*/
0, /*sq_ass_item*/
0, /*sq_ass_slice*/
0, /*sq_contains*/
0, /*sq_inplace_concat*/
0, /*sq_inplace_repeat*/
};
static PyMappingMethods __pyx_tp_as_mapping_RegressionCriterion = {
0, /*mp_length*/
0, /*mp_subscript*/
0, /*mp_ass_subscript*/
};
static PyBufferProcs __pyx_tp_as_buffer_RegressionCriterion = {
#if PY_MAJOR_VERSION < 3
0, /*bf_getreadbuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getwritebuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getsegcount*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getcharbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_getbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_releasebuffer*/
#endif
};
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_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*/
&__pyx_tp_as_number_RegressionCriterion, /*tp_as_number*/
&__pyx_tp_as_sequence_RegressionCriterion, /*tp_as_sequence*/
&__pyx_tp_as_mapping_RegressionCriterion, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
&__pyx_tp_as_buffer_RegressionCriterion, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER|Py_TPFLAGS_BASETYPE, /*tp_flags*/
__Pyx_DOCSTR("Abstract criterion for regression. Computes variance of the\n target values left and right of the split point.\n\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\n Attributes\n ----------\n n_samples : int\n The number of samples\n\n mean_left : double\n The mean target value of the samples left of the split point.\n\n mean_right : double\n The mean target value of the samples right of the split.\n\n sq_sum_left : double\n The sum of squared target values left of the split point.\n\n sq_sum_right : double\n The sum of squared target values right of the split point.\n\n var_left : double\n The variance of the target values left of the split point.\n\n var_right : double\n The variance of the target values left of the split point.\n\n n_left : int\n number of samples left of split point.\n\n n_right : int\n number of samples right of split point.\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*/
__pyx_pf_7sklearn_4tree_5_tree_19RegressionCriterion___init__, /*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
};
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_Criterion(t, a, k);
if (!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 PyMethodDef __pyx_methods_7sklearn_4tree_5_tree_MSE[] = {
{0, 0, 0, 0}
};
static PyNumberMethods __pyx_tp_as_number_MSE = {
0, /*nb_add*/
0, /*nb_subtract*/
0, /*nb_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_divide*/
#endif
0, /*nb_remainder*/
0, /*nb_divmod*/
0, /*nb_power*/
0, /*nb_negative*/
0, /*nb_positive*/
0, /*nb_absolute*/
0, /*nb_nonzero*/
0, /*nb_invert*/
0, /*nb_lshift*/
0, /*nb_rshift*/
0, /*nb_and*/
0, /*nb_xor*/
0, /*nb_or*/
#if PY_MAJOR_VERSION < 3
0, /*nb_coerce*/
#endif
0, /*nb_int*/
#if PY_MAJOR_VERSION < 3
0, /*nb_long*/
#else
0, /*reserved*/
#endif
0, /*nb_float*/
#if PY_MAJOR_VERSION < 3
0, /*nb_oct*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*nb_hex*/
#endif
0, /*nb_inplace_add*/
0, /*nb_inplace_subtract*/
0, /*nb_inplace_multiply*/
#if PY_MAJOR_VERSION < 3
0, /*nb_inplace_divide*/
#endif
0, /*nb_inplace_remainder*/
0, /*nb_inplace_power*/
0, /*nb_inplace_lshift*/
0, /*nb_inplace_rshift*/
0, /*nb_inplace_and*/
0, /*nb_inplace_xor*/
0, /*nb_inplace_or*/
0, /*nb_floor_divide*/
0, /*nb_true_divide*/
0, /*nb_inplace_floor_divide*/
0, /*nb_inplace_true_divide*/
#if PY_VERSION_HEX >= 0x02050000
0, /*nb_index*/
#endif
};
static PySequenceMethods __pyx_tp_as_sequence_MSE = {
0, /*sq_length*/
0, /*sq_concat*/
0, /*sq_repeat*/
0, /*sq_item*/
0, /*sq_slice*/
0, /*sq_ass_item*/
0, /*sq_ass_slice*/
0, /*sq_contains*/
0, /*sq_inplace_concat*/
0, /*sq_inplace_repeat*/
};
static PyMappingMethods __pyx_tp_as_mapping_MSE = {
0, /*mp_length*/
0, /*mp_subscript*/
0, /*mp_ass_subscript*/
};
static PyBufferProcs __pyx_tp_as_buffer_MSE = {
#if PY_MAJOR_VERSION < 3
0, /*bf_getreadbuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getwritebuffer*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getsegcount*/
#endif
#if PY_MAJOR_VERSION < 3
0, /*bf_getcharbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_getbuffer*/
#endif
#if PY_VERSION_HEX >= 0x02060000
0, /*bf_releasebuffer*/
#endif
};
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_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*/
&__pyx_tp_as_number_MSE, /*tp_as_number*/
&__pyx_tp_as_sequence_MSE, /*tp_as_sequence*/
&__pyx_tp_as_mapping_MSE, /*tp_as_mapping*/
0, /*tp_hash*/
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
&__pyx_tp_as_buffer_MSE, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT|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*/
__pyx_methods_7sklearn_4tree_5_tree_MSE, /*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
};
static PyMethodDef __pyx_methods[] = {
{0, 0, 0, 0}
};
#if PY_MAJOR_VERSION >= 3
static struct PyModuleDef __pyx_moduledef = {
PyModuleDef_HEAD_INIT,
__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_kp_u_1, __pyx_k_1, sizeof(__pyx_k_1), 0, 1, 0, 0},
{&__pyx_kp_u_11, __pyx_k_11, sizeof(__pyx_k_11), 0, 1, 0, 0},
{&__pyx_n_s_13, __pyx_k_13, sizeof(__pyx_k_13), 0, 0, 1, 1},
{&__pyx_n_s_14, __pyx_k_14, sizeof(__pyx_k_14), 0, 0, 1, 1},
{&__pyx_n_s_15, __pyx_k_15, sizeof(__pyx_k_15), 0, 0, 1, 1},
{&__pyx_kp_u_3, __pyx_k_3, sizeof(__pyx_k_3), 0, 1, 0, 0},
{&__pyx_kp_u_5, __pyx_k_5, sizeof(__pyx_k_5), 0, 1, 0, 0},
{&__pyx_kp_u_7, __pyx_k_7, sizeof(__pyx_k_7), 0, 1, 0, 0},
{&__pyx_kp_u_8, __pyx_k_8, sizeof(__pyx_k_8), 0, 1, 0, 0},
{&__pyx_n_s__C, __pyx_k__C, sizeof(__pyx_k__C), 0, 0, 1, 1},
{&__pyx_n_s__DTYPE, __pyx_k__DTYPE, sizeof(__pyx_k__DTYPE), 0, 0, 1, 1},
{&__pyx_n_s__RuntimeError, __pyx_k__RuntimeError, sizeof(__pyx_k__RuntimeError), 0, 0, 1, 1},
{&__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_n_s__X_argsorted, __pyx_k__X_argsorted, sizeof(__pyx_k__X_argsorted), 0, 0, 1, 1},
{&__pyx_n_s____main__, __pyx_k____main__, sizeof(__pyx_k____main__), 0, 0, 1, 1},
{&__pyx_n_s____test__, __pyx_k____test__, sizeof(__pyx_k____test__), 0, 0, 1, 1},
{&__pyx_n_s___apply_tree, __pyx_k___apply_tree, sizeof(__pyx_k___apply_tree), 0, 0, 1, 1},
{&__pyx_n_s___error_at_leaf, __pyx_k___error_at_leaf, sizeof(__pyx_k___error_at_leaf), 0, 0, 1, 1},
{&__pyx_n_s___find_best_split, __pyx_k___find_best_split, sizeof(__pyx_k___find_best_split), 0, 0, 1, 1},
{&__pyx_n_s___predict_tree, __pyx_k___predict_tree, sizeof(__pyx_k___predict_tree), 0, 0, 1, 1},
{&__pyx_n_s___random_sample_mask, __pyx_k___random_sample_mask, sizeof(__pyx_k___random_sample_mask), 0, 0, 1, 1},
{&__pyx_n_s__arange, __pyx_k__arange, sizeof(__pyx_k__arange), 0, 0, 1, 1},
{&__pyx_n_s__asarray, __pyx_k__asarray, sizeof(__pyx_k__asarray), 0, 0, 1, 1},
{&__pyx_n_s__astype, __pyx_k__astype, sizeof(__pyx_k__astype), 0, 0, 1, 1},
{&__pyx_n_s__bool, __pyx_k__bool, sizeof(__pyx_k__bool), 0, 0, 1, 1},
{&__pyx_n_s__children, __pyx_k__children, sizeof(__pyx_k__children), 0, 0, 1, 1},
{&__pyx_n_s__criterion, __pyx_k__criterion, sizeof(__pyx_k__criterion), 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__float32, __pyx_k__float32, sizeof(__pyx_k__float32), 0, 0, 1, 1},
{&__pyx_n_s__inf, __pyx_k__inf, sizeof(__pyx_k__inf), 0, 0, 1, 1},
{&__pyx_n_s__init_value, __pyx_k__init_value, sizeof(__pyx_k__init_value), 0, 0, 1, 1},
{&__pyx_n_s__int32, __pyx_k__int32, sizeof(__pyx_k__int32), 0, 0, 1, 1},
{&__pyx_n_s__int8, __pyx_k__int8, sizeof(__pyx_k__int8), 0, 0, 1, 1},
{&__pyx_n_s__k, __pyx_k__k, sizeof(__pyx_k__k), 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__min_leaf, __pyx_k__min_leaf, sizeof(__pyx_k__min_leaf), 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_samples, __pyx_k__n_samples, sizeof(__pyx_k__n_samples), 0, 0, 1, 1},
{&__pyx_n_s__n_total_in_bag, __pyx_k__n_total_in_bag, sizeof(__pyx_k__n_total_in_bag), 0, 0, 1, 1},
{&__pyx_n_s__n_total_samples, __pyx_k__n_total_samples, sizeof(__pyx_k__n_total_samples), 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__order, __pyx_k__order, sizeof(__pyx_k__order), 0, 0, 1, 1},
{&__pyx_n_s__out, __pyx_k__out, sizeof(__pyx_k__out), 0, 0, 1, 1},
{&__pyx_n_s__permutation, __pyx_k__permutation, sizeof(__pyx_k__permutation), 0, 0, 1, 1},
{&__pyx_n_s__pred, __pyx_k__pred, sizeof(__pyx_k__pred), 0, 0, 1, 1},
{&__pyx_n_s__rand, __pyx_k__rand, sizeof(__pyx_k__rand), 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__sample_mask, __pyx_k__sample_mask, sizeof(__pyx_k__sample_mask), 0, 0, 1, 1},
{&__pyx_n_s__scale, __pyx_k__scale, sizeof(__pyx_k__scale), 0, 0, 1, 1},
{&__pyx_n_s__threshold, __pyx_k__threshold, sizeof(__pyx_k__threshold), 0, 0, 1, 1},
{&__pyx_n_s__values, __pyx_k__values, sizeof(__pyx_k__values), 0, 0, 1, 1},
{&__pyx_n_s__xrange, __pyx_k__xrange, sizeof(__pyx_k__xrange), 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_GetName(__pyx_b, __pyx_n_s__range); if (!__pyx_builtin_range) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 434; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
#if PY_MAJOR_VERSION >= 3
__pyx_builtin_xrange = __Pyx_GetName(__pyx_b, __pyx_n_s__range); if (!__pyx_builtin_xrange) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 452; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
#else
__pyx_builtin_xrange = __Pyx_GetName(__pyx_b, __pyx_n_s__xrange); if (!__pyx_builtin_xrange) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 452; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
#endif
__pyx_builtin_ValueError = __Pyx_GetName(__pyx_b, __pyx_n_s__ValueError); if (!__pyx_builtin_ValueError) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 211; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_builtin_RuntimeError = __Pyx_GetName(__pyx_b, __pyx_n_s__RuntimeError); if (!__pyx_builtin_RuntimeError) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 795; __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");
/* "numpy.pxd":211
* 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_k_tuple_2 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_2)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 211; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_k_tuple_2));
__Pyx_INCREF(((PyObject *)__pyx_kp_u_1));
PyTuple_SET_ITEM(__pyx_k_tuple_2, 0, ((PyObject *)__pyx_kp_u_1));
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_1));
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_2));
/* "numpy.pxd":215
* 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_k_tuple_4 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_4)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 215; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_k_tuple_4));
__Pyx_INCREF(((PyObject *)__pyx_kp_u_3));
PyTuple_SET_ITEM(__pyx_k_tuple_4, 0, ((PyObject *)__pyx_kp_u_3));
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_3));
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_4));
/* "numpy.pxd":253
* if ((descr.byteorder == '>' and little_endian) or
* (descr.byteorder == '<' 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_k_tuple_6 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_6)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 253; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_k_tuple_6));
__Pyx_INCREF(((PyObject *)__pyx_kp_u_5));
PyTuple_SET_ITEM(__pyx_k_tuple_6, 0, ((PyObject *)__pyx_kp_u_5));
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_5));
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_6));
/* "numpy.pxd":795
*
* if (end - f) - (new_offset - offset[0]) < 15:
* raise RuntimeError(u"Format string allocated too short, see comment in numpy.pxd") # <<<<<<<<<<<<<<
*
* if ((child.byteorder == '>' and little_endian) or
*/
__pyx_k_tuple_9 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_9)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 795; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_k_tuple_9));
__Pyx_INCREF(((PyObject *)__pyx_kp_u_8));
PyTuple_SET_ITEM(__pyx_k_tuple_9, 0, ((PyObject *)__pyx_kp_u_8));
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_8));
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_9));
/* "numpy.pxd":799
* if ((child.byteorder == '>' and little_endian) or
* (child.byteorder == '<' 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_k_tuple_10 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_10)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 799; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_k_tuple_10));
__Pyx_INCREF(((PyObject *)__pyx_kp_u_5));
PyTuple_SET_ITEM(__pyx_k_tuple_10, 0, ((PyObject *)__pyx_kp_u_5));
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_5));
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_10));
/* "numpy.pxd":819
* 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_k_tuple_12 = PyTuple_New(1); if (unlikely(!__pyx_k_tuple_12)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 819; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_k_tuple_12));
__Pyx_INCREF(((PyObject *)__pyx_kp_u_11));
PyTuple_SET_ITEM(__pyx_k_tuple_12, 0, ((PyObject *)__pyx_kp_u_11));
__Pyx_GIVEREF(((PyObject *)__pyx_kp_u_11));
__Pyx_GIVEREF(((PyObject *)__pyx_k_tuple_12));
__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_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_15 = PyInt_FromLong(15); if (unlikely(!__pyx_int_15)) {__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;
__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)");
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_binding_PyCFunctionType_USED
if (__pyx_binding_PyCFunctionType_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);
#else
__pyx_m = PyModule_Create(&__pyx_moduledef);
#endif
if (!__pyx_m) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
#if PY_MAJOR_VERSION < 3
Py_INCREF(__pyx_m);
#endif
__pyx_b = PyImport_AddModule(__Pyx_NAMESTR(__Pyx_BUILTIN_MODULE_NAME));
if (!__pyx_b) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;};
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 (__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;};
}
/*--- 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_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *, int, int))__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 = (int (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int, int, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, int *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *))__pyx_f_7sklearn_4tree_5_tree_9Criterion_update;
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.eval = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_9Criterion_eval;
__pyx_vtable_7sklearn_4tree_5_tree_Criterion.init_value = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int __pyx_skip_dispatch))__pyx_f_7sklearn_4tree_5_tree_9Criterion_init_value;
if (PyType_Ready(&__pyx_type_7sklearn_4tree_5_tree_Criterion) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 40; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 40; __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 = 40; __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_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_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *, int, int))__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 = (int (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int, int, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, int *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_update;
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.eval = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_eval;
__pyx_vtable_7sklearn_4tree_5_tree_ClassificationCriterion.__pyx_base.init_value = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int __pyx_skip_dispatch))__pyx_f_7sklearn_4tree_5_tree_23ClassificationCriterion_init_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 = 67; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 67; __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 = 67; __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_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.eval = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_4Gini_eval;
__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 = 181; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 181; __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 = 181; __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_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.eval = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_7Entropy_eval;
__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 = 217; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 217; __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 = 217; __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_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_DTYPE_t *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *, int, int))__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 = (int (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int, int, __pyx_t_7sklearn_4tree_5_tree_DTYPE_t *, int *, __pyx_t_7sklearn_4tree_5_tree_BOOL_t *))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_update;
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.eval = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_eval;
__pyx_vtable_7sklearn_4tree_5_tree_RegressionCriterion.__pyx_base.init_value = (PyArrayObject *(*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *, int __pyx_skip_dispatch))__pyx_f_7sklearn_4tree_5_tree_19RegressionCriterion_init_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 = 245; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 245; __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 = 245; __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.eval = (double (*)(struct __pyx_obj_7sklearn_4tree_5_tree_Criterion *))__pyx_f_7sklearn_4tree_5_tree_3MSE_eval;
__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 = 394; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
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 = 394; __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 = 394; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__pyx_ptype_7sklearn_4tree_5_tree_MSE = &__pyx_type_7sklearn_4tree_5_tree_MSE;
/*--- Type import code ---*/
__pyx_ptype_5numpy_dtype = __Pyx_ImportType("numpy", "dtype", sizeof(PyArray_Descr), 0); if (unlikely(!__pyx_ptype_5numpy_dtype)) {__pyx_filename = __pyx_f[1]; __pyx_lineno = 151; __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[1]; __pyx_lineno = 161; __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[1]; __pyx_lineno = 165; __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[1]; __pyx_lineno = 174; __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[1]; __pyx_lineno = 857; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
/*--- Variable import code ---*/
/*--- Function import code ---*/
/*--- Execution code ---*/
/* "sklearn/tree/_tree.pyx":12
* cimport cython
*
* import numpy as np # <<<<<<<<<<<<<<
* cimport numpy as np
*
*/
__pyx_t_1 = __Pyx_Import(((PyObject *)__pyx_n_s__numpy), 0, -1); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 12; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__np, __pyx_t_1) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 12; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
/* "sklearn/tree/_tree.pyx":16
*
* # Define a datatype for the data array
* DTYPE = np.float32 # <<<<<<<<<<<<<<
* ctypedef np.float32_t DTYPE_t
* ctypedef np.int8_t BOOL_t
*/
__pyx_t_1 = __Pyx_GetName(__pyx_m, __pyx_n_s__np); if (unlikely(!__pyx_t_1)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 16; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_1);
__pyx_t_2 = PyObject_GetAttr(__pyx_t_1, __pyx_n_s__float32); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 16; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
__Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
if (PyObject_SetAttr(__pyx_m, __pyx_n_s__DTYPE, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 16; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":409
*
*
* def _random_sample_mask(int n_total_samples, int n_total_in_bag, random_state): # <<<<<<<<<<<<<<
* """Create a random sample mask where ``n_total_in_bag`` elements are set.
*
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree__random_sample_mask, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s___random_sample_mask, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 409; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":442
*
*
* def _apply_tree(np.ndarray[DTYPE_t, ndim=2] X, # <<<<<<<<<<<<<<
* np.ndarray[np.int32_t, ndim=2] children,
* np.ndarray[np.int32_t, ndim=1] feature,
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_1_apply_tree, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s___apply_tree, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 442; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":463
*
*
* def _predict_tree(np.ndarray[DTYPE_t, ndim=2] X, # <<<<<<<<<<<<<<
* np.ndarray[np.int32_t, ndim=2] children,
* np.ndarray[np.int32_t, ndim=1] feature,
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_2_predict_tree, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s___predict_tree, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 463; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":486
*
*
* def _predict_regression_tree_inplace(np.ndarray[DTYPE_t, ndim=2] X, # <<<<<<<<<<<<<<
* np.ndarray[np.int32_t, ndim=2] children,
* np.ndarray[np.int32_t, ndim=1] feature,
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_3_predict_regression_tree_inplace, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s_14, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 486; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":510
*
*
* def _error_at_leaf(np.ndarray[DTYPE_t, ndim=1, mode="c"] y, # <<<<<<<<<<<<<<
* np.ndarray sample_mask, Criterion criterion,
* int n_samples):
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_4_error_at_leaf, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s___error_at_leaf, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 510; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":557
*
*
* def _find_best_split(np.ndarray[DTYPE_t, ndim=2, mode="fortran"] X, # <<<<<<<<<<<<<<
* np.ndarray[DTYPE_t, ndim=1, mode="c"] y,
* np.ndarray[np.int32_t, ndim=2, mode="fortran"] X_argsorted,
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_5_find_best_split, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s___find_best_split, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 557; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":698
* return best_i, best_t, best_error, initial_error
*
* def _find_best_random_split(np.ndarray[DTYPE_t, ndim=2, mode="fortran"] X, # <<<<<<<<<<<<<<
* np.ndarray[DTYPE_t, ndim=1, mode="c"] y,
* np.ndarray[np.int32_t, ndim=2, mode="fortran"] X_argsorted,
*/
__pyx_t_2 = PyCFunction_NewEx(&__pyx_mdef_7sklearn_4tree_5_tree_6_find_best_random_split, NULL, __pyx_n_s_13); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(__pyx_t_2);
if (PyObject_SetAttr(__pyx_m, __pyx_n_s_15, __pyx_t_2) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 698; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
/* "sklearn/tree/_tree.pyx":1
* # encoding: utf-8 # <<<<<<<<<<<<<<
* # cython: cdivision=True
* # cython: boundscheck=False
*/
__pyx_t_2 = PyDict_New(); if (unlikely(!__pyx_t_2)) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_GOTREF(((PyObject *)__pyx_t_2));
if (PyObject_SetAttr(__pyx_m, __pyx_n_s____test__, ((PyObject *)__pyx_t_2)) < 0) {__pyx_filename = __pyx_f[0]; __pyx_lineno = 1; __pyx_clineno = __LINE__; goto __pyx_L1_error;}
__Pyx_DECREF(((PyObject *)__pyx_t_2)); __pyx_t_2 = 0;
/* "numpy.pxd":971
* 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);
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_GetName(PyObject *dict, PyObject *name) {
PyObject *result;
result = PyObject_GetAttr(dict, name);
if (!result) {
if (dict != __pyx_b) {
PyErr_Clear();
result = PyObject_GetAttr(__pyx_b, name);
}
if (!result) {
PyErr_SetObject(PyExc_NameError, name);
}
}
return result;
}
static CYTHON_INLINE int __Pyx_TypeTest(PyObject *obj, PyTypeObject *type) {
if (unlikely(!type)) {
PyErr_Format(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 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_AS_STRING(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;
} else {
#if PY_MAJOR_VERSION < 3
if (unlikely(!PyString_CheckExact(key)) && unlikely(!PyString_Check(key))) {
#else
if (unlikely(!PyUnicode_CheckExact(key)) && unlikely(!PyUnicode_Check(key))) {
#endif
goto invalid_keyword_type;
} else {
for (name = first_kw_arg; *name; name++) {
#if PY_MAJOR_VERSION >= 3
if (PyUnicode_GET_SIZE(**name) == PyUnicode_GET_SIZE(key) &&
PyUnicode_Compare(**name, key) == 0) break;
#else
if (PyString_GET_SIZE(**name) == PyString_GET_SIZE(key) &&
_PyString_Eq(**name, key)) break;
#endif
}
if (*name) {
values[name-argnames] = value;
} else {
/* unexpected keyword found */
for (name=argnames; name != first_kw_arg; name++) {
if (**name == key) goto arg_passed_twice;
#if PY_MAJOR_VERSION >= 3
if (PyUnicode_GET_SIZE(**name) == PyUnicode_GET_SIZE(key) &&
PyUnicode_Compare(**name, key) == 0) goto arg_passed_twice;
#else
if (PyString_GET_SIZE(**name) == PyString_GET_SIZE(key) &&
_PyString_Eq(**name, key)) goto arg_passed_twice;
#endif
}
if (kwds2) {
if (unlikely(PyDict_SetItem(kwds2, key, value))) goto bad;
} else {
goto invalid_keyword;
}
}
}
}
}
return 0;
arg_passed_twice:
__Pyx_RaiseDoubleKeywordsError(function_name, **name);
goto bad;
invalid_keyword_type:
PyErr_Format(PyExc_TypeError,
"%s() keywords must be strings", function_name);
goto bad;
invalid_keyword:
PyErr_Format(PyExc_TypeError,
#if PY_MAJOR_VERSION < 3
"%s() got an unexpected keyword argument '%s'",
function_name, PyString_AsString(key));
#else
"%s() got an unexpected keyword argument '%U'",
function_name, key);
#endif
bad:
return -1;
}
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,
"%s() takes %s %"PY_FORMAT_SIZE_T"d positional argument%s (%"PY_FORMAT_SIZE_T"d given)",
func_name, more_or_less, num_expected,
(num_expected == 1) ? "" : "s", num_found);
}
static CYTHON_INLINE int __Pyx_IsLittleEndian(void) {
unsigned int n = 1;
return *(unsigned char*)(&n) != 0;
}
typedef struct {
__Pyx_StructField root;
__Pyx_BufFmt_StackElem* head;
size_t fmt_offset;
size_t new_count, enc_count;
int is_complex;
char enc_type;
char new_packmode;
char enc_packmode;
} __Pyx_BufFmt_Context;
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;
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 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 'b': return "'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 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': 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': 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, int is_complex) {
switch (ch) {
case '?': case 'c': case 'b': case 'B': 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;
}
}
static char __Pyx_BufFmt_TypeCharToGroup(char ch, int is_complex) {
switch (ch) {
case 'c': case 'b': case 'h': case 'i': case 'l': case 'q': 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;
if (ctx->enc_type == 0) return 0;
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 (type->size != size || type->typegroup != group) {
if (type->typegroup == 'C' && type->fields != NULL) {
/* special case -- treat as struct rather than complex number */
size_t parent_offset = ctx->head->parent_offset + field->offset;
++ctx->head;
ctx->head->field = type->fields;
ctx->head->parent_offset = parent_offset;
continue;
}
__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 %"PY_FORMAT_SIZE_T"d but %"PY_FORMAT_SIZE_T"d expected",
(Py_ssize_t)ctx->fmt_offset, (Py_ssize_t)offset);
return -1;
}
ctx->fmt_offset += size;
--ctx->enc_count; /* Consume from buffer string */
/* Done checking, move to next field, pushing or popping struct stack if needed */
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 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 10:
case 13:
++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;
ctx->new_count = 1;
++ts;
if (*ts != '{') {
PyErr_SetString(PyExc_ValueError, "Buffer acquisition: Expected '{' after 'T'");
return NULL;
}
++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;
}
break;
case '}': /* end of substruct; either repeat or move on */
++ts;
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;
} /* fall through */
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':
if (ctx->enc_type == *ts && got_Z == ctx->is_complex &&
ctx->enc_packmode == ctx->new_packmode) {
/* Continue pooling same type */
ctx->enc_count += ctx->new_count;
} else {
/* New type */
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;
default:
{
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 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 (%"PY_FORMAT_SIZE_T"d byte%s) does not match size of '%s' (%"PY_FORMAT_SIZE_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) {
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);
}
static CYTHON_INLINE void __Pyx_ErrFetch(PyObject **type, PyObject **value, PyObject **tb) {
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;
}
static CYTHON_INLINE int __Pyx_CheckKeywordStrings(
PyObject *kwdict,
const char* function_name,
int kw_allowed)
{
PyObject* key = 0;
Py_ssize_t pos = 0;
while (PyDict_Next(kwdict, &pos, &key, 0)) {
#if PY_MAJOR_VERSION < 3
if (unlikely(!PyString_CheckExact(key)) && unlikely(!PyString_Check(key)))
#else
if (unlikely(!PyUnicode_CheckExact(key)) && unlikely(!PyUnicode_Check(key)))
#endif
goto invalid_keyword_type;
}
if ((!kw_allowed) && unlikely(key))
goto invalid_keyword;
return 1;
invalid_keyword_type:
PyErr_Format(PyExc_TypeError,
"%s() keywords must be strings", function_name);
return 0;
invalid_keyword:
PyErr_Format(PyExc_TypeError,
#if PY_MAJOR_VERSION < 3
"%s() got an unexpected keyword argument '%s'",
function_name, PyString_AsString(key));
#else
"%s() got an unexpected keyword argument '%U'",
function_name, key);
#endif
return 0;
}
static int __Pyx_ArgTypeTest(PyObject *obj, PyTypeObject *type, int none_allowed,
const char *name, int exact)
{
if (!type) {
PyErr_Format(PyExc_SystemError, "Missing type object");
return 0;
}
if (none_allowed && obj == Py_None) return 1;
else if (exact) {
if (Py_TYPE(obj) == type) return 1;
}
else {
if (PyObject_TypeCheck(obj, type)) return 1;
}
PyErr_Format(PyExc_TypeError,
"Argument '%s' has incorrect type (expected %s, got %s)",
name, type->tp_name, Py_TYPE(obj)->tp_name);
return 0;
}
#if PY_MAJOR_VERSION < 3
static void __Pyx_Raise(PyObject *type, PyObject *value, PyObject *tb, PyObject *cause) {
/* cause is unused */
Py_XINCREF(type);
Py_XINCREF(value);
Py_XINCREF(tb);
/* First, check the traceback argument, replacing None with NULL. */
if (tb == Py_None) {
Py_DECREF(tb);
tb = 0;
}
else if (tb != NULL && !PyTraceBack_Check(tb)) {
PyErr_SetString(PyExc_TypeError,
"raise: arg 3 must be a traceback or None");
goto raise_error;
}
/* Next, replace a missing value with None */
if (value == NULL) {
value = Py_None;
Py_INCREF(value);
}
#if PY_VERSION_HEX < 0x02050000
if (!PyClass_Check(type))
#else
if (!PyType_Check(type))
#endif
{
/* Raising an instance. The value should be a dummy. */
if (value != Py_None) {
PyErr_SetString(PyExc_TypeError,
"instance exception may not have a separate value");
goto raise_error;
}
/* Normalize to raise <class>, <instance> */
Py_DECREF(value);
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) {
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)) {
PyErr_SetString(PyExc_TypeError,
"raise: exception class must be a subclass of BaseException");
goto bad;
}
if (cause) {
PyObject *fixed_cause;
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;
}
if (!value) {
value = PyObject_CallObject(type, NULL);
}
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:
return;
}
#endif
static CYTHON_INLINE void __Pyx_RaiseNeedMoreValuesError(Py_ssize_t index) {
PyErr_Format(PyExc_ValueError,
"need more than %"PY_FORMAT_SIZE_T"d value%s to unpack",
index, (index == 1) ? "" : "s");
}
static CYTHON_INLINE void __Pyx_RaiseTooManyValuesError(Py_ssize_t expected) {
PyErr_Format(PyExc_ValueError,
"too many values to unpack (expected %"PY_FORMAT_SIZE_T"d)", expected);
}
static CYTHON_INLINE void __Pyx_RaiseNoneNotIterableError(void) {
PyErr_SetString(PyExc_TypeError, "'NoneType' object is not iterable");
}
static void __Pyx_UnpackTupleError(PyObject *t, Py_ssize_t index) {
if (t == Py_None) {
__Pyx_RaiseNoneNotIterableError();
} else if (PyTuple_GET_SIZE(t) < index) {
__Pyx_RaiseNeedMoreValuesError(PyTuple_GET_SIZE(t));
} else {
__Pyx_RaiseTooManyValuesError(index);
}
}
#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_pf_5numpy_7ndarray___getbuffer__(obj, view, flags);
else {
PyErr_Format(PyExc_TypeError, "'%100s' does not have the buffer interface", Py_TYPE(obj)->tp_name);
return -1;
}
}
static void __Pyx_ReleaseBuffer(Py_buffer *view) {
PyObject* obj = view->obj;
if (obj) {
#if PY_VERSION_HEX >= 0x02060000
if (PyObject_CheckBuffer(obj)) {PyBuffer_Release(view); return;}
#endif
if (PyObject_TypeCheck(obj, __pyx_ptype_5numpy_ndarray)) __pyx_pf_5numpy_7ndarray_1__releasebuffer__(obj, view);
Py_DECREF(obj);
view->obj = NULL;
}
}
#endif
static PyObject *__Pyx_Import(PyObject *name, PyObject *from_list, long level) {
PyObject *py_import = 0;
PyObject *empty_list = 0;
PyObject *module = 0;
PyObject *global_dict = 0;
PyObject *empty_dict = 0;
PyObject *list;
py_import = __Pyx_GetAttrString(__pyx_b, "__import__");
if (!py_import)
goto bad;
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
{
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
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:
Py_XDECREF(empty_list);
Py_XDECREF(py_import);
Py_XDECREF(empty_dict);
return module;
}
#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
static CYTHON_INLINE unsigned char __Pyx_PyInt_AsUnsignedChar(PyObject* x) {
const unsigned char neg_one = (unsigned char)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(unsigned char) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(unsigned char)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to unsigned char" :
"value too large to convert to unsigned char");
}
return (unsigned char)-1;
}
return (unsigned char)val;
}
return (unsigned char)__Pyx_PyInt_AsUnsignedLong(x);
}
static CYTHON_INLINE unsigned short __Pyx_PyInt_AsUnsignedShort(PyObject* x) {
const unsigned short neg_one = (unsigned short)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(unsigned short) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(unsigned short)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to unsigned short" :
"value too large to convert to unsigned short");
}
return (unsigned short)-1;
}
return (unsigned short)val;
}
return (unsigned short)__Pyx_PyInt_AsUnsignedLong(x);
}
static CYTHON_INLINE unsigned int __Pyx_PyInt_AsUnsignedInt(PyObject* x) {
const unsigned int neg_one = (unsigned int)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(unsigned int) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(unsigned int)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to unsigned int" :
"value too large to convert to unsigned int");
}
return (unsigned int)-1;
}
return (unsigned int)val;
}
return (unsigned int)__Pyx_PyInt_AsUnsignedLong(x);
}
static CYTHON_INLINE char __Pyx_PyInt_AsChar(PyObject* x) {
const char neg_one = (char)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(char) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(char)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to char" :
"value too large to convert to char");
}
return (char)-1;
}
return (char)val;
}
return (char)__Pyx_PyInt_AsLong(x);
}
static CYTHON_INLINE short __Pyx_PyInt_AsShort(PyObject* x) {
const short neg_one = (short)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(short) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(short)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to short" :
"value too large to convert to short");
}
return (short)-1;
}
return (short)val;
}
return (short)__Pyx_PyInt_AsLong(x);
}
static CYTHON_INLINE int __Pyx_PyInt_AsInt(PyObject* x) {
const int neg_one = (int)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(int) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(int)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to int" :
"value too large to convert to int");
}
return (int)-1;
}
return (int)val;
}
return (int)__Pyx_PyInt_AsLong(x);
}
static CYTHON_INLINE signed char __Pyx_PyInt_AsSignedChar(PyObject* x) {
const signed char neg_one = (signed char)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(signed char) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(signed char)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to signed char" :
"value too large to convert to signed char");
}
return (signed char)-1;
}
return (signed char)val;
}
return (signed char)__Pyx_PyInt_AsSignedLong(x);
}
static CYTHON_INLINE signed short __Pyx_PyInt_AsSignedShort(PyObject* x) {
const signed short neg_one = (signed short)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(signed short) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(signed short)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to signed short" :
"value too large to convert to signed short");
}
return (signed short)-1;
}
return (signed short)val;
}
return (signed short)__Pyx_PyInt_AsSignedLong(x);
}
static CYTHON_INLINE signed int __Pyx_PyInt_AsSignedInt(PyObject* x) {
const signed int neg_one = (signed int)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(signed int) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(signed int)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to signed int" :
"value too large to convert to signed int");
}
return (signed int)-1;
}
return (signed int)val;
}
return (signed int)__Pyx_PyInt_AsSignedLong(x);
}
static CYTHON_INLINE int __Pyx_PyInt_AsLongDouble(PyObject* x) {
const int neg_one = (int)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
if (sizeof(int) < sizeof(long)) {
long val = __Pyx_PyInt_AsLong(x);
if (unlikely(val != (long)(int)val)) {
if (!unlikely(val == -1 && PyErr_Occurred())) {
PyErr_SetString(PyExc_OverflowError,
(is_unsigned && unlikely(val < 0)) ?
"can't convert negative value to int" :
"value too large to convert to int");
}
return (int)-1;
}
return (int)val;
}
return (int)__Pyx_PyInt_AsLong(x);
}
static CYTHON_INLINE unsigned long __Pyx_PyInt_AsUnsignedLong(PyObject* x) {
const unsigned long neg_one = (unsigned long)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
#if PY_VERSION_HEX < 0x03000000
if (likely(PyInt_Check(x))) {
long val = PyInt_AS_LONG(x);
if (is_unsigned && unlikely(val < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to unsigned long");
return (unsigned long)-1;
}
return (unsigned long)val;
} else
#endif
if (likely(PyLong_Check(x))) {
if (is_unsigned) {
if (unlikely(Py_SIZE(x) < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to unsigned long");
return (unsigned long)-1;
}
return (unsigned long)PyLong_AsUnsignedLong(x);
} else {
return (unsigned long)PyLong_AsLong(x);
}
} else {
unsigned long val;
PyObject *tmp = __Pyx_PyNumber_Int(x);
if (!tmp) return (unsigned long)-1;
val = __Pyx_PyInt_AsUnsignedLong(tmp);
Py_DECREF(tmp);
return val;
}
}
static CYTHON_INLINE unsigned PY_LONG_LONG __Pyx_PyInt_AsUnsignedLongLong(PyObject* x) {
const unsigned PY_LONG_LONG neg_one = (unsigned PY_LONG_LONG)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
#if PY_VERSION_HEX < 0x03000000
if (likely(PyInt_Check(x))) {
long val = PyInt_AS_LONG(x);
if (is_unsigned && unlikely(val < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to unsigned PY_LONG_LONG");
return (unsigned PY_LONG_LONG)-1;
}
return (unsigned PY_LONG_LONG)val;
} else
#endif
if (likely(PyLong_Check(x))) {
if (is_unsigned) {
if (unlikely(Py_SIZE(x) < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to unsigned PY_LONG_LONG");
return (unsigned PY_LONG_LONG)-1;
}
return (unsigned PY_LONG_LONG)PyLong_AsUnsignedLongLong(x);
} else {
return (unsigned PY_LONG_LONG)PyLong_AsLongLong(x);
}
} else {
unsigned PY_LONG_LONG val;
PyObject *tmp = __Pyx_PyNumber_Int(x);
if (!tmp) return (unsigned PY_LONG_LONG)-1;
val = __Pyx_PyInt_AsUnsignedLongLong(tmp);
Py_DECREF(tmp);
return val;
}
}
static CYTHON_INLINE long __Pyx_PyInt_AsLong(PyObject* x) {
const long neg_one = (long)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
#if PY_VERSION_HEX < 0x03000000
if (likely(PyInt_Check(x))) {
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 (unlikely(Py_SIZE(x) < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to long");
return (long)-1;
}
return (long)PyLong_AsUnsignedLong(x);
} else {
return (long)PyLong_AsLong(x);
}
} else {
long val;
PyObject *tmp = __Pyx_PyNumber_Int(x);
if (!tmp) return (long)-1;
val = __Pyx_PyInt_AsLong(tmp);
Py_DECREF(tmp);
return val;
}
}
static CYTHON_INLINE PY_LONG_LONG __Pyx_PyInt_AsLongLong(PyObject* x) {
const PY_LONG_LONG neg_one = (PY_LONG_LONG)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
#if PY_VERSION_HEX < 0x03000000
if (likely(PyInt_Check(x))) {
long val = PyInt_AS_LONG(x);
if (is_unsigned && unlikely(val < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to PY_LONG_LONG");
return (PY_LONG_LONG)-1;
}
return (PY_LONG_LONG)val;
} else
#endif
if (likely(PyLong_Check(x))) {
if (is_unsigned) {
if (unlikely(Py_SIZE(x) < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to PY_LONG_LONG");
return (PY_LONG_LONG)-1;
}
return (PY_LONG_LONG)PyLong_AsUnsignedLongLong(x);
} else {
return (PY_LONG_LONG)PyLong_AsLongLong(x);
}
} else {
PY_LONG_LONG val;
PyObject *tmp = __Pyx_PyNumber_Int(x);
if (!tmp) return (PY_LONG_LONG)-1;
val = __Pyx_PyInt_AsLongLong(tmp);
Py_DECREF(tmp);
return val;
}
}
static CYTHON_INLINE signed long __Pyx_PyInt_AsSignedLong(PyObject* x) {
const signed long neg_one = (signed long)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
#if PY_VERSION_HEX < 0x03000000
if (likely(PyInt_Check(x))) {
long val = PyInt_AS_LONG(x);
if (is_unsigned && unlikely(val < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to signed long");
return (signed long)-1;
}
return (signed long)val;
} else
#endif
if (likely(PyLong_Check(x))) {
if (is_unsigned) {
if (unlikely(Py_SIZE(x) < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to signed long");
return (signed long)-1;
}
return (signed long)PyLong_AsUnsignedLong(x);
} else {
return (signed long)PyLong_AsLong(x);
}
} else {
signed long val;
PyObject *tmp = __Pyx_PyNumber_Int(x);
if (!tmp) return (signed long)-1;
val = __Pyx_PyInt_AsSignedLong(tmp);
Py_DECREF(tmp);
return val;
}
}
static CYTHON_INLINE signed PY_LONG_LONG __Pyx_PyInt_AsSignedLongLong(PyObject* x) {
const signed PY_LONG_LONG neg_one = (signed PY_LONG_LONG)-1, const_zero = 0;
const int is_unsigned = neg_one > const_zero;
#if PY_VERSION_HEX < 0x03000000
if (likely(PyInt_Check(x))) {
long val = PyInt_AS_LONG(x);
if (is_unsigned && unlikely(val < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to signed PY_LONG_LONG");
return (signed PY_LONG_LONG)-1;
}
return (signed PY_LONG_LONG)val;
} else
#endif
if (likely(PyLong_Check(x))) {
if (is_unsigned) {
if (unlikely(Py_SIZE(x) < 0)) {
PyErr_SetString(PyExc_OverflowError,
"can't convert negative value to signed PY_LONG_LONG");
return (signed PY_LONG_LONG)-1;
}
return (signed PY_LONG_LONG)PyLong_AsUnsignedLongLong(x);
} else {
return (signed PY_LONG_LONG)PyLong_AsLongLong(x);
}
} else {
signed PY_LONG_LONG val;
PyObject *tmp = __Pyx_PyNumber_Int(x);
if (!tmp) return (signed PY_LONG_LONG)-1;
val = __Pyx_PyInt_AsSignedLongLong(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;
}
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_SetItemString(dict, "__pyx_vtable__", ob) < 0)
goto bad;
Py_DECREF(ob);
return 0;
bad:
Py_XDECREF(ob);
return -1;
}
#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_module = __Pyx_ImportModule(module_name);
if (!py_module)
goto bad;
#if PY_MAJOR_VERSION < 3
py_name = PyString_FromString(class_name);
#else
py_name = PyUnicode_FromString(class_name);
#endif
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,
"%s.%s is not a type object",
module_name, class_name);
goto bad;
}
if (!strict && ((PyTypeObject *)result)->tp_basicsize > (Py_ssize_t)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 (((PyTypeObject *)result)->tp_basicsize != (Py_ssize_t)size) {
PyErr_Format(PyExc_ValueError,
"%s.%s 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
#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;
#if PY_MAJOR_VERSION < 3
py_name = PyString_FromString(name);
#else
py_name = PyUnicode_FromString(name);
#endif
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
#include "compile.h"
#include "frameobject.h"
#include "traceback.h"
static void __Pyx_AddTraceback(const char *funcname, int __pyx_clineno,
int __pyx_lineno, const char *__pyx_filename) {
PyObject *py_srcfile = 0;
PyObject *py_funcname = 0;
PyObject *py_globals = 0;
PyCodeObject *py_code = 0;
PyFrameObject *py_frame = 0;
#if PY_MAJOR_VERSION < 3
py_srcfile = PyString_FromString(__pyx_filename);
#else
py_srcfile = PyUnicode_FromString(__pyx_filename);
#endif
if (!py_srcfile) goto bad;
if (__pyx_clineno) {
#if PY_MAJOR_VERSION < 3
py_funcname = PyString_FromFormat( "%s (%s:%d)", funcname, __pyx_cfilenm, __pyx_clineno);
#else
py_funcname = PyUnicode_FromFormat( "%s (%s:%d)", funcname, __pyx_cfilenm, __pyx_clineno);
#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_globals = PyModule_GetDict(__pyx_m);
if (!py_globals) goto bad;
py_code = PyCode_New(
0, /*int argcount,*/
#if PY_MAJOR_VERSION >= 3
0, /*int kwonlyargcount,*/
#endif
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,*/
__pyx_lineno, /*int firstlineno,*/
__pyx_empty_bytes /*PyObject *lnotab*/
);
if (!py_code) 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 = __pyx_lineno;
PyTraceBack_Here(py_frame);
bad:
Py_XDECREF(py_srcfile);
Py_XDECREF(py_funcname);
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;
}
/* Type Conversion Functions */
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_VERSION_HEX < 0x03000000
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_VERSION_HEX < 0x03000000
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_VERSION_HEX < 0x03000000
if (!PyInt_Check(res) && !PyLong_Check(res)) {
#else
if (!PyLong_Check(res)) {
#endif
PyErr_Format(PyExc_TypeError,
"__%s__ returned non-%s (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;
}
static CYTHON_INLINE Py_ssize_t __Pyx_PyIndex_AsSsize_t(PyObject* b) {
Py_ssize_t ival;
PyObject* 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
}
static CYTHON_INLINE size_t __Pyx_PyInt_AsSize_t(PyObject* x) {
unsigned PY_LONG_LONG val = __Pyx_PyInt_AsUnsignedLongLong(x);
if (unlikely(val == (unsigned PY_LONG_LONG)-1 && PyErr_Occurred())) {
return (size_t)-1;
} else if (unlikely(val != (unsigned PY_LONG_LONG)(size_t)val)) {
PyErr_SetString(PyExc_OverflowError,
"value too large to convert to size_t");
return (size_t)-1;
}
return (size_t)val;
}
#endif /* Py_PYTHON_H */
Reference in New Issue View Git Blame Copy Permalink