Class pySGD

include/bind.h:844–869

};

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

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

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

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

class pyMomentum : public py::class_<graphvite::Momentum, graphvite::Optimizer> {
public: