Sha256: 4c12180279118c76b72d7de2f39ae0b224a8d902a2f826af0134f2b2a87de7d5

Contents?: true

Size: 1.21 KB

Versions: 11

Compression:

Stored size: 1.21 KB

Contents

class CreateEasyMLTunerJobs < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
  def change
    unless table_exists?(:easy_ml_tuner_jobs)
      create_table :easy_ml_tuner_jobs do |t|
        t.json :config, null: false
        t.bigint :best_tuner_run_id
        t.bigint :model_id, null: false
        t.string :status
        t.string :direction, default: 'minimize'
        t.datetime :started_at
        t.datetime :completed_at
        t.jsonb :metadata
        t.string :wandb_url

        t.timestamps

        t.index :status
        t.index :started_at
        t.index :completed_at
        t.index :model_id
        t.index :best_tuner_run_id
        t.index :wandb_url
      end
    end

    unless table_exists?(:easy_ml_tuner_runs)
      create_table :easy_ml_tuner_runs do |t|
        t.bigint :tuner_job_id, null: false
        t.json :hyperparameters, null: false
        t.float :value
        t.integer :trial_number
        t.string :status
        t.string :wandb_url

        t.timestamps

        t.index [:tuner_job_id, :value]
        t.index [:tuner_job_id, :trial_number], name: "idx_tuner_runs_and_trial_number"
        t.index :status
        t.index :wandb_url
      end
    end
  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
easy_ml-0.2.0.pre.rc52 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc51 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc50 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc49 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc48 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc47 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc46 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc45 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc44 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc43 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt
easy_ml-0.2.0.pre.rc41 lib/easy_ml/railtie/templates/migration/create_easy_ml_tuner_jobs.rb.tt