Class pyOptimizer

include/bind.h:800–842

};

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

    template<class... Args>
    pyOptimizer(py::handle scope, const char *name, const Args &...args) :
            Base(scope, name, args...) {
        attr("__doc__") = "Optimizer(*args, **kwargs)"
        R"(
        General interface of first-order optimizers.

        This class has 2 constructors

        .. function:: Optimizer(type)
        .. function:: Optimizer(lr=1e-4)

        Parameters:
            type (int): kAuto
            lr (float, optional): initial learning rate
        )";

        // data members
        def_readonly("type", &Optimizer::type);
        def_readonly("lr", &Optimizer::lr);
        def_readonly("weight_decay", &Optimizer::weight_decay);
        def_readonly("schedule", &Optimizer::schedule);

        // member functions
        def(py::init<int>(), py::no_gil(),
            py::arg("type") = graphvite::kAuto);

        def(py::init<float>(), py::no_gil(),
            py::arg("lr") = 1e-4);

        py::implicitly_convertible<int, graphvite::Optimizer>();
        py::implicitly_convertible<float, graphvite::Optimizer>();

        def("__repr__", &Optimizer::info, py::no_gil());
    }
};

class pySGD : public py::class_<graphvite::SGD, graphvite::Optimizer> {
public: