Class pyMomentum

include/bind.h:871–900

};

class pyMomentum : public py::class_<graphvite::Momentum, graphvite::Optimizer> {
public:
    typedef graphvite::Momentum Momentum;
    typedef py::class_<Momentum, graphvite::Optimizer> Base;
    using Base::def_readonly;
    using Base::def;

    template<class... Args>
    pyMomentum(py::handle scope, const char *name, const Args &...args) :
            Base(scope, name, args...) {
        attr("__doc__") = "Momentum(lr=1e-4, weight_decay=0, momentum=0.999, schedule='linear')"
        R"(
        Momentum optimizer.

        Parameters:
            lr (float, optional): initial learning rate
            weight_decay (float, optional): weight decay (L2 regularization)
            momentum (float, optional): momentum coefficient
            schedule (str or callable, optional): learning rate schedule
        )";

        // data members
        def_readonly("momentum", &Momentum::momentum);

        // member functions
        def(py::init<float, float, float, graphvite::LRSchedule>(), py::no_gil(),
            py::arg("lr") = 1e-4, py::arg("weight_decay") = 0, py::arg("momentum") = 0.999,
            py::arg("schedule") = "linear");
    }
};

class pyAdaGrad : public py::class_<graphvite::AdaGrad, graphvite::Optimizer> {
public: