wandb 0.1.9 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fe07e6917e230a051db42ac9d95a887397818e5504cf505d836084749de8cee7
-  data.tar.gz: 7d49ebf4b2f85ed7375e6e62fcadce06fe0b3f10aeac5c8d955ea2d4cd537150
+  metadata.gz: ed086eee277a0665a3281f93764c8042edf254c30bded00812f3f332085e0ece
+  data.tar.gz: 4c2e7443bbc914c38701eb3baa00f858f2e7a9d25afcef274c198404512681cc
 SHA512:
-  metadata.gz: 1538917b3ea0b8fd0a34af20bcd79c7683bbc70ca126a6ba192c6e8645d6810d6fe6e68f3e36018a19b4722173ee72792c47cb12edaa3ac6bfe86d0a7853363d
-  data.tar.gz: b816436bb73da8935fa3b1c6c7824b1f7b00601da184b27eed806a7782d924c7531b095a4f7fb5d73a6de3f8d2615b7c2da33b1204922669cad29cfb23f04d61
+  metadata.gz: 52c8315c8d711a06263cbd98d296b7fca5030ee1ed155e5603a607c04348a36115595cd2cfa9ea9b4a76547d3d31cce07f2835496bd26c78cbf6f3cb824253aa
+  data.tar.gz: caa96acc54494baf471d5b06dc74dfdde85fb7d54e35ebdfee485b3c02cbfca044c2c3793161b6eed2b0816b1e43a2568818d143e180a9af7697edfb7728b73c

data/lib/wandb/version.rb

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

data/lib/wandb/xgboost_callback.rb

@@ -20,7 +20,9 @@ module Wandb
       end
     end
 
-    attr_accessor :project_name, :api_key, :custom_loggers, :history, :sample
+    attr_accessor :project_name, :api_key, :custom_loggers, :history, :sample,
+                  :log_model, :log_feature_importance, :importance_type, :define_metric,
+                  :normalize_feature_importance
 
     def initialize(options = {})
       options = Opts.new(options)
@@ -35,6 +37,19 @@ module Wandb
       @custom_loggers = options.default(:custom_loggers, [])
     end
 
+    def as_json
+      {
+        log_model: @log_model,
+        log_feature_importance: @log_feature_importance,
+        importance_type: @importance_type,
+        define_metric: @define_metric,
+        normalize_feature_importance: @normalize_feature_importance,
+        sample: @sample,
+        project_name: @project_name,
+        callback_type: :wandb,
+      }
+    end
+
     def before_training(model)
       Wandb.login(api_key: api_key)
       Wandb.init(project: project_name)
@@ -42,10 +57,8 @@ module Wandb
       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
+        n_estimators: model.num_boosted_rounds,
       }
-      Wandb.current_run.config = log_conf
-
       Wandb.log(log_conf)
       model
     end
@@ -65,7 +78,7 @@ module Wandb
 
       Wandb.log(
         "best_score" => model.best_score.to_f,
-        "best_iteration" => model.best_iteration.to_i
+        "best_iteration" => model.best_iteration.to_i,
       )
       finish
 
@@ -93,9 +106,6 @@ 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)
       end
       false
@@ -126,7 +136,7 @@ module Wandb
       fi_data = fi.map { |k, v| [k, v] }
 
       table = Wandb::Table.new(data: fi_data, columns: %w[Feature Importance])
-      bar_plot = Wandb::Plot.bar(table.table, "Feature", "Importance", title: "Feature Importance")
+      bar_plot = Wandb::Plot.bar(table.table, label: "Feature", value: "Importance", title: "Feature Importance")
       Wandb.log({ "Feature Importance" => bar_plot.__pyptr__ })
     end
 

data/lib/wandb.rb

@@ -48,6 +48,14 @@ module Wandb
       @current_run = Run.new(run)
     end
 
+    def latest_run=(run)
+      @latest_run = run
+    end
+
+    def latest_run
+      @latest_run
+    end
+
     # Get the current run
     attr_reader :current_run
 
@@ -74,7 +82,6 @@ module Wandb
       Plot
     end
 
-    # Add this new method
     def run_url
       raise "No active run. Call Wandb.init() first." unless @current_run
 
@@ -244,8 +251,14 @@ module Wandb
   # Plot class
   class Plot
     class << self
-      def bar(table, x_key, y_key, title: nil)
-        py_plot = Wandb.__pyptr__.plot.bar(table.__pyptr__, x_key, y_key, title: title)
+      def bar(table, label:, value:, title: nil, split_table: false)
+        py_plot = Wandb.__pyptr__.plot.bar(
+          table: table.__pyptr__,
+          label: label,
+          value: value,
+          title: title,
+          split_table: split_table,
+        )
         new(py_plot)
       end
 
@@ -270,7 +283,7 @@ module Wandb
     end
 
     def __pyptr__
-      @plot
+      @plot.__pyptr__
     end
   end
 end

metadata

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