(require ffi/vector

xgboost)

(provide run-example)

(define dense-features

(f32vector 1.0 2.0 0.5   2.0 1.0 1.5   3.0 0.5 0.0   0.5 3.0 2.0

4.0 2.0 1.0   1.5 1.5 0.5   2.5 3.5 1.5   0.0 1.0 0.0))

(define columnar-features

(list (f32vector 1.0 2.0 3.0 0.5 4.0 1.5 2.5 0.0)

(f32vector 2.0 1.0 0.5 3.0 2.0 1.5 3.5 1.0)

(f32vector 0.5 1.5 0.0 2.0 1.0 0.5 1.5 0.0)))

(define labels (f32vector 3.5 3.5 6.5 2.0 9.0 4.0 7.0 1.0))

(define (f32vector~= a b)

(and (= (f32vector-length a) (f32vector-length b))

(for/and ([i (in-range (f32vector-length a))])

(< (abs (- (f32vector-ref a i) (f32vector-ref b i))) 0.0001))))

(define (make-trained)

(define dtrain (make-dmatrix dense-features #:nrow 8 #:ncol 3 #:missing -1.0 #:labels labels))

(values (train dtrain #:objective "reg:squarederror"

#:max-depth 3 #:eta 0.1 #:verbosity 0 #:rounds 20)

dtrain))

(define (run-example)

(define-values (booster dtrain) (make-trained))

(define dmatrix-preds (predict booster dtrain #:as 'f32vector))

(define inplace-preds

(predict-from-columnar booster columnar-features #:missing -1.0 #:as 'f32vector))

(hash 'prediction-count (f32vector-length inplace-preds)

'matches-dmatrix? (f32vector~= inplace-preds dmatrix-preds)))

<r22-require>

<r22-provide>

<r22-helpers>

<r22-run>