wandb 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f0f0909ffdfe53479bf0c6a09f09f96c7e0aec1dc846f16d9b42c1b1e7f4fae
-  data.tar.gz: f1ffc1d2cdabc05964ef0187f5ab168a71ef38077591e3a07bf960da3b5ae8b5
+  metadata.gz: fe07e6917e230a051db42ac9d95a887397818e5504cf505d836084749de8cee7
+  data.tar.gz: 7d49ebf4b2f85ed7375e6e62fcadce06fe0b3f10aeac5c8d955ea2d4cd537150
 SHA512:
-  metadata.gz: 93b0f7df5f4bb660a9d566b3492a75690336eeddbfbe0aad15fc100c292499f5fb93a543f1f9f72b6120b2682d8ee25dd3b5ead6906500efd1f5ccc29f6a7219
-  data.tar.gz: 95e7472bfa9e0322bb353d6ee77bd76a7b6f1c09edd6ef23f10f294ef982b35bae279b5ccbb4beee44991decf774b3bac45ab708ebbab4dbb0dda47917715545
+  metadata.gz: 1538917b3ea0b8fd0a34af20bcd79c7683bbc70ca126a6ba192c6e8645d6810d6fe6e68f3e36018a19b4722173ee72792c47cb12edaa3ac6bfe86d0a7853363d
+  data.tar.gz: b816436bb73da8935fa3b1c6c7824b1f7b00601da184b27eed806a7782d924c7531b095a4f7fb5d73a6de3f8d2615b7c2da33b1204922669cad29cfb23f04d61

data/lib/wandb/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Wandb
-  VERSION = "0.1.7"
+  VERSION = "0.1.9"
 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])

data/lib/wandb.rb

@@ -73,6 +73,13 @@ module Wandb
     def plot
       Plot
     end
+
+    # Add this new method
+    def run_url
+      raise "No active run. Call Wandb.init() first." unless @current_run
+
+      @current_run.url
+    end
   end
 
   # Run class
@@ -113,6 +120,11 @@ module Wandb
     def log_artifact(artifact)
       @run.log_artifact(artifact.__pyptr__)
     end
+
+    # Add this new method
+    def url
+      @run.get_url
+    end
   end
 
   # Artifact class

metadata

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