Class pyAdaGrad

include/bind.h:902–931

};

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

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

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

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

        // 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("epsilon") = 1e-10,
            py::arg("schedule") = "linear");
    }
};

class pyRMSprop : public py::class_<graphvite::RMSprop, graphvite::Optimizer> {
public: