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

experiments_controller.rb « ml « projects « controllers « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 85e7f63779c53ead6c74691ebc982327c0c6ca0f (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
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
# frozen_string_literal: true

module Projects
  module Ml
    class ExperimentsController < ::Projects::ApplicationController
      include Projects::Ml::ExperimentsHelper

      before_action :check_read, only: [:show, :index]
      before_action :check_write, only: [:destroy]
      before_action :set_experiment, only: [:show, :destroy]

      feature_category :mlops

      MAX_EXPERIMENTS_PER_PAGE = 20
      MAX_CANDIDATES_PER_PAGE = 30

      def index
        paginator = ::Ml::Experiment.by_project_id(@project.id)
                                    .with_candidate_count
                                    .keyset_paginate(cursor: params[:cursor], per_page: MAX_EXPERIMENTS_PER_PAGE)

        @experiments = paginator.records
        @page_info = page_info(paginator)
      end

      def show
        find_params = params
                        .transform_keys(&:underscore)
                        .permit(:name, :order_by, :sort, :order_by_type)

        finder = CandidateFinder.new(@experiment, find_params)

        respond_to do |format|
          format.csv do
            csv_data = ::Ml::CandidatesCsvPresenter.new(finder.execute).present

            send_data(csv_data, type: 'text/csv; charset=utf-8', filename: 'candidates.csv')
          end

          format.html do
            paginator = finder.execute.keyset_paginate(cursor: params[:cursor], per_page: MAX_CANDIDATES_PER_PAGE)

            @candidates = paginator.records
            @page_info = page_info(paginator)
          end
        end
      end

      def destroy
        @experiment.destroy

        redirect_to project_ml_experiments_path(@project),
          status: :found,
          notice: s_("MlExperimentTracking|Experiment removed")
      end

      private

      def check_read
        render_404 unless can?(current_user, :read_model_experiments, @project)
      end

      def check_write
        render_404 unless can?(current_user, :write_model_experiments, @project)
      end

      def set_experiment
        @experiment = ::Ml::Experiment.by_project_id_and_iid(@project.id, params[:iid])

        render_404 unless @experiment
      end
    end
  end
end