MCPcopy Create free account
hub / github.com/catboost/catboost / TDatasetSubset

Class TDatasetSubset

catboost/libs/data/loader.h:33–59  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31namespace NCB {
32
33 struct TDatasetSubset {
34 bool HasFeatures = true;
35 TIndexRange<ui64> Range = {0, Max<ui64>()};
36
37 public:
38 ui64 GetSize() const { return Range.GetSize(); }
39
40 static TDatasetSubset MakeRange(ui64 start, ui64 end) {
41 return {true, {start, end}};
42 }
43
44 static TDatasetSubset MakeColumns(bool hasFeatures = true) {
45 return {hasFeatures, {0u, Max<ui64>()}};
46 }
47
48 size_t GetHash() const {
49 return MultiHash(HasFeatures, Range.Begin, Range.End);
50 }
51
52 bool operator==(const TDatasetSubset& rhs) const {
53 return std::tie(HasFeatures, Range) == std::tie(rhs.HasFeatures, rhs.Range);
54 }
55
56 bool operator!=(const TDatasetSubset& rhs) const {
57 return !(rhs == *this);
58 }
59 };
60
61 struct TDatasetLoaderCommonArgs {
62 TPathWithScheme PairsFilePath;

Callers 2

TRawDatasetRowsReaderMethod · 0.85
LoadQuantizedPoolMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected