wandb 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f0f0909ffdfe53479bf0c6a09f09f96c7e0aec1dc846f16d9b42c1b1e7f4fae
-  data.tar.gz: f1ffc1d2cdabc05964ef0187f5ab168a71ef38077591e3a07bf960da3b5ae8b5
+  metadata.gz: 17b68d037ed6930865fae0bd24437e1c41d3a746208c69237bf2cd9da3ed2c4e
+  data.tar.gz: be8a03c3963951579dadd5397780cdd1d003f0b2e7e83f6a181b6dda491267de
 SHA512:
-  metadata.gz: 93b0f7df5f4bb660a9d566b3492a75690336eeddbfbe0aad15fc100c292499f5fb93a543f1f9f72b6120b2682d8ee25dd3b5ead6906500efd1f5ccc29f6a7219
-  data.tar.gz: 95e7472bfa9e0322bb353d6ee77bd76a7b6f1c09edd6ef23f10f294ef982b35bae279b5ccbb4beee44991decf774b3bac45ab708ebbab4dbb0dda47917715545
+  metadata.gz: e7e1edea11690670d9d1af572c99caea8d6985fa31bd3e6255602352081d794bc00690a95042bf301588b160952de808be7f242b4f25cf6c54a8d146e966ccdb
+  data.tar.gz: 1c3120cf18ba4ee4cc2cd3e26f61cccc296cc7059b823f0d8892a6b27571a7de120a8d3378151010793b42e99539e722c18a7aae363af465cbc53da26098b7a1

data/lib/wandb/version.rb

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

data/lib/wandb/xgboost_callback.rb

@@ -20,16 +20,18 @@ module Wandb
       end
     end
 
-    attr_accessor :project_name, :api_key, :custom_loggers
+    attr_accessor :project_name, :api_key, :custom_loggers, :history, :sample
 
     def initialize(options = {})
       options = Opts.new(options)
       @log_model = options.default(:log_model, false)
       @log_feature_importance = options.default(:log_feature_importance, true)
       @importance_type = options.default(:importance_type, "gain")
+      @normalize_feature_importance = options.default(:normalize_feature_importance, true)
       @define_metric = options.default(:define_metric, true)
       @api_key = options.default(:api_key, ENV["WANDB_API_KEY"])
       @project_name = options.default(:project_name, nil)
+      @sample = options.default(:sample, 1.0)
       @custom_loggers = options.default(:custom_loggers, [])
     end
 
@@ -80,19 +82,22 @@ module Wandb
     end
 
     def after_iteration(model, epoch, history)
-      history.each do |split, metric_scores|
-        metric = metric_scores.keys.first
-        values = metric_scores.values.last
-        epoch_value = values[epoch]
-
-        define_metric(split, metric) if @define_metric && epoch == 0
-        full_metric_name = "#{split}-#{metric}"
-        Wandb.log({ full_metric_name => epoch_value })
+      log_frequency = (1.0 / @sample).round
+      if epoch % log_frequency == 0
+        history.to_h.each do |split, metric_scores|
+          metric = metric_scores.keys.first
+          values = metric_scores.values.last
+          epoch_value = values[epoch]
+
+          define_metric(split, metric) if @define_metric && epoch == 0
+          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)
       end
-      @custom_loggers.each do |logger|
-        logger.call(model, epoch, history)
-      end
-      Wandb.log("epoch" => epoch)
       false
     end
 
@@ -112,6 +117,12 @@ module Wandb
 
     def log_feature_importance(model)
       fi = model.score(importance_type: @importance_type)
+
+      if @normalize_feature_importance
+        total_importance = fi.values.sum
+        fi = fi.transform_values { |v| v / total_importance }
+      end
+
       fi_data = fi.map { |k, v| [k, v] }
 
       table = Wandb::Table.new(data: fi_data, columns: %w[Feature Importance])

metadata

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