Class pyWordGraph

include/bind.h:190–234

template<class Index = size_t>
class pyWordGraph : public py::class_<graphvite::WordGraph<Index>, graphvite::Graph<Index>> {
public:
    typedef graphvite::WordGraph<Index> WordGraph;
    typedef py::class_<WordGraph, graphvite::Graph<Index>> Base;
    using Base::attr;
    using Base::def;
    using Base::def_readonly;

    template<class... Args>
    pyWordGraph(py::handle scope, const char *name, const Args &...args) :
    Base(scope, signature(name, typeid(Index).name()).c_str(), args...) {
        attr("__doc__") = "WordGraph(index_type=dtype.uint32)"
        R"(
        Normal graphs of word co-occurrences.

        Parameters:
            index_type (dtype): type of node indexes
        )";

        // override instance name with template name
        attr("__name__") = py::internal_str(name);
        attr("__qualname__") = py::internal_str(name);

        // member functions
        def(py::init<>(), py::no_gil());

        def("load", &WordGraph::load_file_compact, py::no_gil(),
            py::arg("file_name"), py::arg("window") = 5, py::arg("min_count") = 5, py::arg("normalization") = false,
            py::arg("delimiters") = " \t\r\n", py::arg("comment") = "#",
            R"(load(file_name, window=5, min_count=5, normalization=False, delimiters=' \\t\\r\\n', comment='#'))"
            R"(
            Load a word graph from a corpus file. Store the graph in an adjacency list.

            Parameters:
                file_name (str): file name
                window (int, optional): word pairs with distance <= window are counted as edges
                min_count (int, optional): words with occurrence <= min_count are discarded
                normalization (bool, optional): normalize the adjacency matrix or not
                delimiters (str, optional): string of delimiter characters
                comment (str, optional): prefix of comment strings
            )");

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

template<class Index = size_t>
class pyKnowledgeGraph : public py::class_<graphvite::KnowledgeGraph<Index>> {