(require ffi/vector

racket/file

xgboost)

(provide run-example)

(define (vec=? a b)

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

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

(= (f32vector-ref a i) (f32vector-ref b i)))))

(define (run-example)

(define dtrain

(make-dmatrix (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)

#:nrow 8 #:ncol 3

#:labels (f32vector 3.5 3.5 6.5 2.0 9.0 4.0 7.0 1.0)))

(define booster

(train dtrain #:objective "reg:squarederror"

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

(define baseline (predict booster dtrain #:as 'f32vector))

(define model-path (make-temporary-file "xgbrkt-~a.json"))

(save-model booster model-path)

(define file-bytes (file->bytes model-path))

(define preds-file (predict (load-model model-path) dtrain #:as 'f32vector))

(delete-file model-path)

(define ubj-blob (save-model-to-bytes booster))

(define json-blob (save-model-to-bytes booster #:format "json"))

(define preds-ubj (predict (load-model-from-bytes ubj-blob) dtrain #:as 'f32vector))

(define preds-json (predict (load-model-from-bytes json-blob) dtrain #:as 'f32vector))

(hash 'file-bytes (bytes-length file-bytes)

'ubj-bytes (bytes-length ubj-blob)

'json-bytes (bytes-length json-blob)

'file-match? (vec=? preds-file baseline)

'ubj-match? (vec=? preds-ubj baseline)

'json-match? (vec=? preds-json baseline)

'baseline (f32vector->list baseline)))

<r03-require>

<r03-provide>

<r03-helpers>

<r03-run>