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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
# frozen_string_literal: true
module Ml
module ExperimentTracking
class CandidateRepository
attr_accessor :project, :user, :experiment, :candidate
def initialize(project, user)
@project = project
@user = user
end
def by_iid(iid)
::Ml::Candidate.with_project_id_and_iid(project.id, iid)
end
def create!(experiment, start_time)
experiment.candidates.create!(
user: user,
start_time: start_time || 0
)
end
def update(candidate, status, end_time)
candidate.status = status.downcase if status
candidate.end_time = end_time if end_time
candidate.save
end
def add_metric!(candidate, name, value, tracked_at, step)
candidate.metrics.create!(
name: name,
value: value,
tracked_at: tracked_at,
step: step
)
end
def add_param!(candidate, name, value)
candidate.params.create!(name: name, value: value)
end
def add_metrics(candidate, metric_definitions)
return unless candidate.present?
metrics = metric_definitions.map do |metric|
{
candidate_id: candidate.id,
name: metric[:key],
value: metric[:value],
tracked_at: metric[:timestamp],
step: metric[:step],
**timestamps
}
end
::Ml::CandidateMetric.insert_all(metrics, returning: false) unless metrics.empty?
end
def add_params(candidate, param_definitions)
return unless candidate.present?
parameters = param_definitions.map do |p|
{
candidate_id: candidate.id,
name: p[:key],
value: p[:value],
**timestamps
}
end
::Ml::CandidateParam.insert_all(parameters, returning: false) unless parameters.empty?
end
private
def timestamps
current_time = Time.zone.now
{ created_at: current_time, updated_at: current_time }
end
end
end
end
|