Class pyAdam

include/bind.h:966–999

};

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

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

        Parameters:
            lr (float, optional): initial learning rate
            weight_decay (float, optional): weight decay (L2 regularization)
            beta1 (float, optional): coefficient for moving average of gradient
            beta2 (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("beta1", &Adam::beta1);
        def_readonly("beta2", &Adam::beta2);
        def_readonly("epsilon", &Adam::epsilon, "");

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

std::function<py::object(const std::string &, py::args, py::kwargs)> optimizer_helper(const py::module &module) {
    return [module](const std::string &type, py::args args, py::kwargs kwargs) {