blob: 62e2e75db2a839be3c56659f098dbd8e3481a0bf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
# frozen_string_literal: true
module Ml
class Candidate < ApplicationRecord
enum status: { running: 0, scheduled: 1, finished: 2, failed: 3, killed: 4 }
validates :iid, :experiment, presence: true
validates :status, inclusion: { in: statuses.keys }
belongs_to :experiment, class_name: 'Ml::Experiment'
belongs_to :user
has_many :metrics, class_name: 'Ml::CandidateMetric'
has_many :params, class_name: 'Ml::CandidateParam'
default_value_for(:iid) { SecureRandom.uuid }
def artifact_root
"/ml_candidate_#{iid}/-/"
end
class << self
def with_project_id_and_iid(project_id, iid)
return unless project_id.present? && iid.present?
joins(:experiment).find_by(experiment: { project_id: project_id }, iid: iid)
end
end
end
end
|