wandb 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4a00bf9591759e1339bca2ff33aca8dc245f71e8ef8f9e3aa4cd70a59f598a10
-  data.tar.gz: 880bed08a72bb3f03d48d093ab68a94ab8b706b759ec10945008979fd0c044c8
+  metadata.gz: 310ede46ccba30aa00bfbb81fbb8ffdefd818528c7ae09523e7dca9629258892
+  data.tar.gz: 07643a463e807e99db1a4e874bd85b749f6d65b644f15e0423505ead63ec7749
 SHA512:
-  metadata.gz: b6148f71160761a123696f0beac8b2ebe68064a4f64bd55ce1e9ec417c44b5c032a9f60b77394a3ebd0f0535e60c2ce50e53b5606f8acb958bfd35148c0f4a53
-  data.tar.gz: 14a72f4b7b0bc05234d9e0e276c724b1077ac61c07b755f90ce365d4d596c9183928b5df6d9396391a9339d6441b92e1e4c0733f67ee19634ba7fec47acb6818
+  metadata.gz: 00e7a6fe5e888d931c1abbc6047207859a46b024c83a05a6e87feeea8d20576f2f3da1d6a5c851c9dd945e7bd564ecfc56b5c4707ed3b54284aea97ecb4dac46
+  data.tar.gz: d4b2c592d6a4a5e079024d48cb7a97834a883f1e5e12dbe1c856255d4beed84c90f54f629d818cad14513f4778b08657e6781c958ff977fe607d1a74c0a2deec

data/lib/wandb/version.rb

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

data/lib/wandb/xgboost_callback.rb

@@ -19,7 +19,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 +29,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 +56,7 @@ module Wandb
 
       # Log best score and best iteration
       unless model.best_score
-        Wandb.finish
+        finish
         return model
       end
 
@@ -65,18 +64,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 +88,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.6
 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