| 35 | |
| 36 | public: |
| 37 | bool Init(const DeepConfig& config) override { |
| 38 | resource_.reset(new deep_op::LocalDeepOpResource); |
| 39 | |
| 40 | resource_->set_deep_config(config); |
| 41 | |
| 42 | auto deep_data = DeepData::Create(config); |
| 43 | if (!deep_data) { |
| 44 | return false; |
| 45 | } |
| 46 | resource_->set_deep_data(std::move(deep_data)); |
| 47 | |
| 48 | // The sampler needs to be created only if the freq file is not empty. |
| 49 | if (!config.freq_file().empty()) { |
| 50 | auto sampler_source = NewDeepSamplerSource(resource_->deep_data()); |
| 51 | if (!sampler_source) { |
| 52 | return false; |
| 53 | } |
| 54 | resource_->set_sampler_source(std::move(sampler_source)); |
| 55 | |
| 56 | auto negative_sampler_builder = NewSamplerBuilder( |
| 57 | resource_->sampler_source(), SamplerBuilderEnum::NEGATIVE_SAMPLER, |
| 58 | config.negative_sampler_type(), config.thread_num()); |
| 59 | if (!negative_sampler_builder) { |
| 60 | return false; |
| 61 | } |
| 62 | resource_->set_negative_sampler_builder( |
| 63 | std::move(negative_sampler_builder)); |
| 64 | } |
| 65 | |
| 66 | factory_ = deep_op::LocalDeepOpFactory::GetInstance(); |
| 67 | return factory_->Init(resource_.get()); |
| 68 | } |
| 69 | |
| 70 | bool SharedSampleNegative( |
| 71 | int count, const vec_int_t& nodes, const vec_int_t& excluded_nodes, |
no test coverage detected