Welcome to mirror list, hosted at ThFree Co, Russian Federation.

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'app/services/ml/create_candidate_service.rb')
-rw-r--r--app/services/ml/create_candidate_service.rb37
1 files changed, 37 insertions, 0 deletions
diff --git a/app/services/ml/create_candidate_service.rb b/app/services/ml/create_candidate_service.rb
new file mode 100644
index 00000000000..53913c3fb19
--- /dev/null
+++ b/app/services/ml/create_candidate_service.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Ml
+ class CreateCandidateService
+ def initialize(experiment, params = {})
+ @experiment = experiment
+ @name = params[:name]
+ @user = params[:user]
+ @start_time = params[:start_time]
+ @model_version = params[:model_version]
+ end
+
+ def execute
+ Ml::Candidate.create!(
+ experiment: experiment,
+ project: experiment.project,
+ name: candidate_name,
+ start_time: start_time || 0,
+ user: user,
+ model_version: model_version
+ )
+ end
+
+ private
+
+ def candidate_name
+ name.presence || random_candidate_name
+ end
+
+ def random_candidate_name
+ parts = Array.new(3).map { FFaker::Animal.common_name.downcase.delete(' ') } << rand(10000)
+ parts.join('-').truncate(255)
+ end
+
+ attr_reader :name, :user, :experiment, :start_time, :model_version
+ end
+end