wandb 0.1.5 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4a00bf9591759e1339bca2ff33aca8dc245f71e8ef8f9e3aa4cd70a59f598a10
-  data.tar.gz: 880bed08a72bb3f03d48d093ab68a94ab8b706b759ec10945008979fd0c044c8
+  metadata.gz: 6f0f0909ffdfe53479bf0c6a09f09f96c7e0aec1dc846f16d9b42c1b1e7f4fae
+  data.tar.gz: f1ffc1d2cdabc05964ef0187f5ab168a71ef38077591e3a07bf960da3b5ae8b5
 SHA512:
-  metadata.gz: b6148f71160761a123696f0beac8b2ebe68064a4f64bd55ce1e9ec417c44b5c032a9f60b77394a3ebd0f0535e60c2ce50e53b5606f8acb958bfd35148c0f4a53
-  data.tar.gz: 14a72f4b7b0bc05234d9e0e276c724b1077ac61c07b755f90ce365d4d596c9183928b5df6d9396391a9339d6441b92e1e4c0733f67ee19634ba7fec47acb6818
+  metadata.gz: 93b0f7df5f4bb660a9d566b3492a75690336eeddbfbe0aad15fc100c292499f5fb93a543f1f9f72b6120b2682d8ee25dd3b5ead6906500efd1f5ccc29f6a7219
+  data.tar.gz: 95e7472bfa9e0322bb353d6ee77bd76a7b6f1c09edd6ef23f10f294ef982b35bae279b5ccbb4beee44991decf774b3bac45ab708ebbab4dbb0dda47917715545

data/lib/wandb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Wandb
-  VERSION = "0.1.5"
+  VERSION = "0.1.7"
 end

data/lib/wandb/xgboost_callback.rb

@@ -1,6 +1,7 @@
 require "xgb"
 require "tempfile"
 require "fileutils"
+require "xgboost/training_callback"
 
 module Wandb
   class XGBoostCallback < XGBoost::TrainingCallback
@@ -19,7 +20,7 @@ module Wandb
       end
     end
 
-    attr_accessor :project_name, :api_key
+    attr_accessor :project_name, :api_key, :custom_loggers
 
     def initialize(options = {})
       options = Opts.new(options)
@@ -29,22 +30,21 @@ module Wandb
       @define_metric = options.default(:define_metric, true)
       @api_key = options.default(:api_key, ENV["WANDB_API_KEY"])
       @project_name = options.default(:project_name, nil)
-
-      raise "WANDB_API_KEY required" unless api_key
-      raise "project_name required" unless project_name
-
-      Wandb.login(api_key: api_key)
-      Wandb.init(project: project_name)
-
-      return if Wandb.current_run
-
-      raise "You must call wandb.init() before WandbCallback()"
+      @custom_loggers = options.default(:custom_loggers, [])
     end
 
     def before_training(model)
-      # Update Wandb config with model configuration
-      Wandb.current_run.config = model.params
-      Wandb.log(model.params)
+      Wandb.login(api_key: api_key)
+      Wandb.init(project: project_name)
+      config = JSON.parse(model.save_config)
+      log_conf = {
+        learning_rate: config.dig("learner", "gradient_booster", "tree_train_param", "learning_rate").to_f,
+        max_depth: config.dig("learner", "gradient_booster", "tree_train_param", "max_depth").to_f,
+        n_estimators: model.num_boosted_rounds
+      }
+      Wandb.current_run.config = log_conf
+
+      Wandb.log(log_conf)
       model
     end
 
@@ -57,7 +57,7 @@ module Wandb
 
       # Log best score and best iteration
       unless model.best_score
-        Wandb.finish
+        finish
         return model
       end
 
@@ -65,18 +65,21 @@ module Wandb
         "best_score" => model.best_score.to_f,
         "best_iteration" => model.best_iteration.to_i
       )
-
-      Wandb.finish
-      FileUtils.rm_rf(Rails.root.join("wandb"))
+      finish
 
       model
     end
 
+    def finish
+      Wandb.finish
+      FileUtils.rm_rf(File.join(Dir.pwd, "wandb"))
+    end
+
     def before_iteration(_model, _epoch, _history)
       false
     end
 
-    def after_iteration(_model, epoch, history)
+    def after_iteration(model, epoch, history)
       history.each do |split, metric_scores|
         metric = metric_scores.keys.first
         values = metric_scores.values.last
@@ -86,6 +89,9 @@ module Wandb
         full_metric_name = "#{split}-#{metric}"
         Wandb.log({ full_metric_name => epoch_value })
       end
+      @custom_loggers.each do |logger|
+        logger.call(model, epoch, history)
+      end
       Wandb.log("epoch" => epoch)
       false
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: wandb
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.7
 platform: ruby
 authors:
 - Brett Shollenberger
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-16 00:00:00.000000000 Z
+date: 2024-10-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pycall