Method predict

include/core/solver.h:1564–1584

* @brief Predict on samples in the sample block * @param _head_partition_id id of head partition * @param _tail_partition_id id of tail partition */

     * @param _tail_partition_id id of tail partition
     */
    virtual void predict(int _head_partition_id, int _tail_partition_id) {
        CUDA_CHECK(cudaSetDevice(device_id));
        load_partition(_head_partition_id, _tail_partition_id);

        auto &samples = solver->predict_pool[head_partition_id][tail_partition_id];
        auto &indexes = solver->sample_indexes[head_partition_id][tail_partition_id];
        log_frequency = solver->log_frequency;
        size_t num_sample = samples.size();
        int num_batch = (num_sample + batch_size - 1) / batch_size;
        for (size_t i = 0; i < num_batch; i++) {
            int actual_size = std::min(size_t(batch_size), num_sample - i * batch_size);
            batch.copy(&samples[i * batch_size], actual_size * kSampleSize);
            logits.resize(actual_size);
            predict_batch(solver->predict_batch_id++);
            for (int j = 0; j < actual_size; j++) {
                size_t index = indexes[i * batch_size + j];
                solver->results[index] = logits[j];
            }
        }
        logits.reallocate(0);
    }

    /** Predict a single batch */
    virtual void predict_batch(int batch_id) {