Class SolverMixin

include/core/solver.h:89–888

template<size_t _dim, class _Float, class _Index, template<class> class _Graph, template<class> class _Sampler,
        template<class> class _Worker>
class SolverMixin {
public:
    static const size_t dim = _dim;
    typedef _Float Float;
    typedef _Index Index;

    typedef _Graph<Index> Graph;
    typedef _Sampler<SolverMixin> Sampler;
    typedef _Worker<SolverMixin> Worker;

    typedef Vector<dim, Float> Vector;
    typedef typename Sampler::EdgeSample EdgeSample;
    typedef std::function<void(Sampler *, int, int)> SampleFunction;

    static const int kSampleSize = Sampler::kSampleSize;

    Graph *graph = nullptr;
    Index num_vertex;
    size_t num_edge;
    int num_moment;
    int num_embedding, num_partition, num_negative, sample_batch_size;
    float negative_sample_exponent, negative_weight;
    int num_epoch, episode_size, batch_size, positive_reuse;
    int log_frequency;
    bool shuffle_partition, naive_parallel, resume;
    int assignment_offset;
    std::vector<std::shared_ptr<std::vector<Vector>>> embeddings;
    std::vector<std::shared_ptr<std::vector<std::vector<Vector>>>> moments;
    std::vector<Protocol> protocols;
    Protocol sampler_protocol;
    bool tied_weights;
    std::vector<std::vector<Index>> head_partitions, tail_partitions;
    Index head_partition_size, tail_partition_size;
    std::vector<std::pair<int, Index>> head_locations, tail_locations;
    AliasTable<Float, size_t> edge_table;
    std::vector<std::vector<std::vector<std::vector<EdgeSample>>>> sample_pools;
    // <<<<<<<<<<<<<<< predict <<<<<<<<<<<<<<<
    const std::vector<EdgeSample> *samples;
    const py::array_t<Index> *array;
    std::vector<Float> results;
    std::vector<std::vector<std::vector<EdgeSample>>> predict_pool;
    std::vector<std::vector<std::vector<size_t>>> sample_indexes;
    std::vector<std::vector<std::vector<size_t>>> pool_offsets;
    // >>>>>>>>>>>>>>> predict >>>>>>>>>>>>>>>
    bool is_train;
    int pool_id = 0;
    std::vector<Sampler *> samplers;
    std::vector<Worker *> workers;
    std::string model;
    std::set<std::string> available_models;
    Optimizer optimizer;
    int num_worker, num_sampler, num_thread;
    size_t gpu_memory_limit, gpu_memory_cost;
    volatile std::atomic<int> batch_id, predict_batch_id;
    int num_batch, num_predict_batch;

#define USING_SOLVER_MIXIN(type) \
    using type::dim; \