survival 1.0.10

A high-performance survival analysis library written in Rust with Python bindings
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

from libc.math cimport exp, log
import numpy as np
cimport numpy as cnp

cdef extern from "survS.h":
    double** dmatrix(double* data, int nrows, int ncols)

from libc.stdlib cimport malloc, free
import numpy as np
cimport numpy as cnp

cpdef void agexact(int* maxiter, int* nusedx, int* nvarx, double* start, 
                   double* stop, int* event, double* covar2, double* offset, 
                   int* strata, double* means, double* beta, double* u, 
                   double* imat2, double loglik[2], int* flag, double* work, 
                   int* work2, double* eps, double* tol_chol, 
                   double* sctest, int* nocenter):
    cdef:
        int i, j, k, person
        int iter
        int n = nusedx[0]
        int nvar = nvarx[0]
        double **covar, **imat
        double *a, *newbeta
        double temp
        double *score
        double **cmat
        double *newvar
        int *index
        int deaths, nrisk
        int halving = 0

    covar = <double **>malloc(n * sizeof(double *))
    for i in range(n):
        covar[i] = <double *>malloc(nvar * sizeof(double))

    imat = <double **>malloc(nvar * sizeof(double *))
    for i in range(nvar):
        imat[i] = <double *>malloc(nvar * sizeof(double))
    
    cmat = <double **>malloc(nvar * sizeof(double *))
    for i in range(nvar):
        cmat[i] = <double *>malloc(nvar * sizeof(double))
    
    a = <double *>malloc(nvar * sizeof(double))
    newbeta = <double *>malloc(nvar * sizeof(double))
    score = <double *>malloc(nvar * sizeof(double))
    newvar = <double *>malloc(nvar * sizeof(double))
    index = <int *>malloc(n * sizeof(int))

    for i in range(n):
        free(covar[i])
    free(covar)

    for i in range(nvar):
        free(imat[i])
    free(imat)

    for i in range(nvar):
        free(cmat[i])
    free(cmat)
    
    free(a)
    free(newbeta)
    free(score)
    free(newvar)
    free(index)