Class pyRMSprop

include/bind.h:933–964

};

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

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

        Parameters:
            lr (float, optional): initial learning rate
            weight_decay (float, optional): weight decay (L2 regularization)
            alpha (float, optional): coefficient for moving average of squared gradient
            epsilon (float, optional): smooth term for numerical stability
            schedule (str or callable, optional): learning rate schedule
        )";

        // data members
        def_readonly("alpha", &RMSprop::alpha);
        def_readonly("epsilon", &RMSprop::epsilon);

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

class pyAdam : public py::class_<graphvite::Adam, graphvite::Optimizer> {
public: