(require ffi/vector

xgboost)

(provide run-example)

(define 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 labels (f32vector 3.5 3.5 6.5 2.0 9.0 4.0 7.0 1.0))

(define (train-rounds! b dtrain start rounds)

(for ([iter (in-range start (+ start rounds))])

(booster-update-one-iter! b iter dtrain)))

(define (run-example)

(define dtrain

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

(define b

(train dtrain #:objective "reg:squarederror"

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

(train-rounds! b dtrain 0 5)

(define snapshot (booster->bytes b))

(define snapshot-preds (predict b dtrain #:as 'f32vector))

(define restored (bytes->booster snapshot))

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

(train-rounds! b dtrain 5 5)

(train-rounds! restored dtrain 5 5)

(hash 'snapshot-bytes (bytes-length snapshot)

'matches-immediately?

(equal? (f32vector->list snapshot-preds) (f32vector->list restored-preds))

'matches-after-resume?

(equal? (f32vector->list (predict b dtrain #:as 'f32vector))

(f32vector->list (predict restored dtrain #:as 'f32vector)))))

<r26-require>

<r26-provide>

<r26-helper>

<r26-run>