Skip to content

Instantly share code, notes, and snippets.

@ilivans
ilivans / xgboost_to_solr.py
Created June 10, 2022 22:11
Convert XGBoost model into Solr LTR MultipleAdditiveTreesModel
import json
from xgboost import XGBModel
def dump_xgbmodel(xgb_model: XGBModel) -> dict:
"""
Dump XGBModel instance as a Solr LTR MultipleAdditiveTreesModel model.
Solr LTR MART model specification:
from __future__ import print_function
import os
import gc
from time import sleep
import keras.backend as K
import numpy as np
import psutil
from keras.models import Sequential