blob: 7c1a0ef22945a28ba86777efc116d6c174de6117 (
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
|
// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef MEDIA_CAPABILITIES_LEARNING_HELPER_H_
#define MEDIA_CAPABILITIES_LEARNING_HELPER_H_
#include <memory>
#include "base/macros.h"
#include "base/threading/sequence_bound.h"
#include "media/base/media_export.h"
#include "media/capabilities/video_decode_stats_db.h"
#include "media/learning/impl/feature_provider.h"
#include "media/learning/impl/learning_session_impl.h"
namespace media {
// Helper class to allow MediaCapabilities to log training examples to a
// media::learning LearningTask.
class MEDIA_EXPORT LearningHelper {
public:
// |feature_factory| lets us register FeatureProviders with those
// LearningTasks that include standard features.
LearningHelper(learning::FeatureProviderFactoryCB feature_factory);
~LearningHelper();
// |origin| is sent in separately since it's somewhat hacky. This should be
// provided by the FeatureProvider, but the way LearningHelper is used, it
// doesn't have access to the origin. Per-frame LearningTaskControllers will
// be able to do this much more easily.
void AppendStats(const VideoDecodeStatsDB::VideoDescKey& video_key,
learning::FeatureValue origin,
const VideoDecodeStatsDB::DecodeStatsEntry& new_stats);
private:
// Convenience function to begin and complete an observation.
void AddExample(learning::LearningTaskController* controller,
const learning::LabelledExample& example);
// Learning session for our profile. Normally, we'd not have one of these
// directly, but would instead get one that's connected to a browser profile.
// For now, however, we just instantiate one and assume that we'll be
// destroyed when the profile changes / history is cleared.
std::unique_ptr<learning::LearningSessionImpl> learning_session_;
// Controllers for each task.
std::unique_ptr<learning::LearningTaskController>
base_unweighted_table_controller_;
std::unique_ptr<learning::LearningTaskController>
base_unweighted_tree_controller_;
std::unique_ptr<learning::LearningTaskController>
base_unweighted_tree_200_controller_;
std::unique_ptr<learning::LearningTaskController>
enhanced_unweighted_tree_200_controller_;
};
} // namespace media
#endif // MEDIA_CAPABILITIES_LEARNING_HELPER_H_
|
