69 lines
2.4 KiB
Ruby
69 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Ml
|
|
class Model < ApplicationRecord
|
|
include Presentable
|
|
include Sortable
|
|
include CacheMarkdownField
|
|
|
|
EXPERIMENT_NAME_PREFIX = '[model]'
|
|
|
|
validates :project, :default_experiment, presence: true
|
|
validates :name,
|
|
format: Gitlab::Regex.ml_model_name_regex,
|
|
uniqueness: { scope: :project },
|
|
presence: true,
|
|
length: { maximum: 255 }
|
|
|
|
validate :valid_default_experiment?
|
|
validates :description,
|
|
length: { maximum: 10_000 }
|
|
|
|
has_one :default_experiment, class_name: 'Ml::Experiment'
|
|
belongs_to :project
|
|
belongs_to :user
|
|
has_many :versions, class_name: 'Ml::ModelVersion'
|
|
has_many :candidates, -> { without_model_version }, class_name: 'Ml::Candidate', through: :default_experiment
|
|
has_many :metadata, class_name: 'Ml::ModelMetadata'
|
|
has_one :latest_version, -> { latest_by_model }, class_name: 'Ml::ModelVersion', inverse_of: :model
|
|
|
|
scope :including_latest_version, -> { includes(:latest_version) }
|
|
scope :including_project, -> { includes(:project) }
|
|
scope :with_version_count, -> {
|
|
version_counts = Ml::ModelVersion
|
|
.select(:model_id, 'COUNT(*) as count')
|
|
.group(:model_id)
|
|
|
|
joins("LEFT OUTER JOIN (#{version_counts.to_sql}) as version_counts ON version_counts.model_id = ml_models.id")
|
|
.select('ml_models.*, COALESCE(version_counts.count, 0) as version_count')
|
|
}
|
|
scope :by_name, ->(name) { where("ml_models.name LIKE ?", "%#{sanitize_sql_like(name)}%") } # rubocop:disable GitlabSecurity/SqlInjection
|
|
scope :by_project, ->(project) { where(project_id: project.id) }
|
|
|
|
cache_markdown_field :description
|
|
|
|
def all_packages
|
|
Packages::MlModel::Package.where(project: project, id: versions.select(:package_id))
|
|
end
|
|
|
|
def valid_default_experiment?
|
|
return unless default_experiment
|
|
|
|
errors.add(:default_experiment) unless default_experiment.name == self.class.prefixed_experiment(name)
|
|
errors.add(:default_experiment) unless default_experiment.project_id == project_id
|
|
end
|
|
|
|
def self.by_project_id_and_id(project_id, id)
|
|
find_by(project_id: project_id, id: id)
|
|
end
|
|
|
|
def self.by_project_id_and_name(project_id, name)
|
|
find_by(project_id: project_id, name: name)
|
|
end
|
|
|
|
def self.prefixed_experiment(string)
|
|
"#{EXPERIMENT_NAME_PREFIX}#{string}"
|
|
end
|
|
end
|
|
end
|