Gitiles
Code Review Sign In
korap.ids-mannheim.de / ids-kl / collocatordb / 6663f11ebcbc2b676eba5cf6060af959cdebc0bb
commit6663f11ebcbc2b676eba5cf6060af959cdebc0bb[log] [tgz]
authorMarc Kupietz <kupietz@ids-mannheim.de>Sun Mar 14 09:20:59 2021 +0100
committerMarc Kupietz <kupietz@ids-mannheim.de>Sun Mar 14 14:44:28 2021 +0100
treee63c7ec18e42d5a12b26718c3b446331c9054acc
parentb4a683c0e64ca17c1a20cff91427b35a1d1ca533 [diff]
Add first tests

Change-Id: I7ef95cef912feb2b997e1c7da5eb5a369cbe56b1
389 files changed
tree: e63c7ec18e42d5a12b26718c3b446331c9054acc
  1. examples/
  2. extras/
  3. src/
  4. tests/
  5. .gitignore
  6. Makefile
  7. README.md
README.md

CollocatorDB: Storing and retrieving collocation counts based on RocksDB

Installation

Install RocksdDB and prerequisites

  • install prerequisites on CentOS, FeDora, RHEL
    sudo yum install snappy snappy-devel zlib zlib-devel bzip2 bzip2-devel lz4-devel libzstd-devel libomp-devel

git clone https://github.com/gflags/gflags.git
cd gflags
git checkout v2.0
./configure && make && sudo make install
cd ..
  • install prerequisites on Ubuntu, Debian
    sudo apt-get install libgflags-dev libsnappy-dev zlib1g-dev libbz2-dev liblz4-dev libzstd-dev libomp-dev
  • install RocksDBn v5.11.3
    git clone https://github.com/facebook/rocksdb.git
cd rocksdb
git checkout v5.11.3
make static_lib DISABLE_WARNING_AS_ERROR=1 && sudo make install-static DISABLE_WARNING_AS_ERROR=1
make shared_lib DISABLE_WARNING_AS_ERROR=1 && sudo make install-shared DISABLE_WARNING_AS_ERROR=1

Install CollocatorDB

make && make install

Provided API

typedef struct {
    uint32_t w2;
    uint64_t f2;
    uint64_t raw;
    double pmi;
    double npmi;
    double llr;
    double lfmd;
    double md;
    uint64_t left_raw;
    uint64_t right_raw;
    double left_pmi;
    double right_pmi;
    double dice;
    double logdice;
    double ldaf;
    int window;
    int af_window;
} Collocator ;

COLLOCATORDB *open_collocatordb(const char *s);
COLLOCATORDB *open_collocatordb_for_write(const char *s);
void inc_collocator(COLLOCATORDB *db, uint64_t w1, uint64_t w2, int8_t dist);
void dump_collocators(COLLOCATORDB *db, uint32_t w1, uint32_t w2, int8_t dist);
Collocator *get_collocators(COLLOCATORDB *db, uint32_t w1);
Collocator *get_collocation_scores(COLLOCATORDB *db, uint32_t w1, uint32_t w2);
char *get_collocators_as_json(COLLOCATORDB *db, uint32_t w1);
char *get_collocation_scores_as_json(COLLOCATORDB *db, uint32_t w1, uint32_t w2);
char *get_word(COLLOCATORDB *db, uint32_t w1);
void read_vocab(COLLOCATORDB *db, char *fname);

TODO

  • extend APIs
  • add unit and ci tests
  • improve build process

License

Based on RocksDB, CollocatorDB is dual-licensed under both the GPLv2 (found in the COPYING file in the root directory) and Apache 2.0 License (found in the LICENSE.Apache file in the root directory). You may select, at your option, one of the above-listed licenses.