wandb 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 310ede46ccba30aa00bfbb81fbb8ffdefd818528c7ae09523e7dca9629258892
-  data.tar.gz: 07643a463e807e99db1a4e874bd85b749f6d65b644f15e0423505ead63ec7749
+  metadata.gz: 17b68d037ed6930865fae0bd24437e1c41d3a746208c69237bf2cd9da3ed2c4e
+  data.tar.gz: be8a03c3963951579dadd5397780cdd1d003f0b2e7e83f6a181b6dda491267de
 SHA512:
-  metadata.gz: 00e7a6fe5e888d931c1abbc6047207859a46b024c83a05a6e87feeea8d20576f2f3da1d6a5c851c9dd945e7bd564ecfc56b5c4707ed3b54284aea97ecb4dac46
-  data.tar.gz: d4b2c592d6a4a5e079024d48cb7a97834a883f1e5e12dbe1c856255d4beed84c90f54f629d818cad14513f4778b08657e6781c958ff977fe607d1a74c0a2deec
+  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.6"
+  VERSION = "0.1.8"
 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,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
 
@@ -79,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
 
@@ -111,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.6
+  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