Wie kann ich das xgboost-Modell speichern und laden?

10

Auf dem Link des XGBoost-Leitfadens ,

  1. Das Modell kann gespeichert werden. %Code%
  2. Das Modell und seine Feature-Map können auch in eine Textdatei ausgegeben werden.
    bst.save_model('0001.model')
  3. Ein gespeichertes Modell kann wie folgt geladen werden: %Code%

Meine Frage ist:

  1. Was ist der Unterschied zwischen bst.dump_model('dump.raw.txt') # dump model bst.dump_model('dump.raw.txt','featmap.txt')# dump model with feature map & amp; %Code%?
  2. Was ist der Unterschied zwischen dem Speichern von bst = xgb.Booster({'nthread':4}) #init model bst.load_model("model.bin") # load data und save_model ?
  3. warum der Modellname zum Laden von dump_model sich von dem Namen unterscheidet, der gespeichert werden soll '0001.model' ?
  4. Angenommen, ich trainiere zwei Modelle 'dump.raw.txt','featmap.txt' und model.bin , ich wollte beide Modelle für zukünftige Verwendung speichern, die 0001.model & amp; model_A Funktion sollte ich verwenden? Könnten Sie helfen, den klaren Prozess zu zeigen?
Pengju Zhao 29.04.2017, 03:05
quelle

1 Antwort

5

Beide Funktionen save_model und dump_model speichern das Modell, der Unterschied ist, dass in dump_model der Merkmalsname gespeichert und der Baum im Textformat gespeichert werden kann.

Der load_model arbeitet mit dem Modell von save_model . Das Modell von dump_model kann beispielsweise mit xgbfi verwendet werden.

Beim Laden des Modells müssen Sie den Pfad angeben, in dem Ihre Modelle gespeichert werden. Im Beispiel wird bst.load_model("model.bin") model aus der Datei model.bin geladen - es ist nur ein Name der Datei mit dem Modell. Viel Glück!

    
pplonski 22.05.2017 13:36
quelle

Tags und Links