1eb155ed93
* enable hot-word boosting * more consistent ordering of CLI arguments * progress on review * use map instead of set for hot-words, move string logic to client.cc * typo bug * pointer things? * use map for hotwords, better string splitting * add the boost, not multiply * cleaning up * cleaning whitespace * remove <set> inclusion * change typo set-->map * rename boost_coefficient to boost X-DeepSpeech: NOBUILD * add hot_words to python bindings * missing hot_words * include map in swigwrapper.i * add Map template to swigwrapper.i * emacs intermediate file * map things * map-->unordered_map * typu * typu * use dict() not None * error out if hot_words without scorer * two new functions: remove hot-word and clear all hot-words * starting to work on better error messages X-DeepSpeech: NOBUILD * better error handling + .Net ERR codes * allow for negative boosts:) * adding TC test for hot-words * add hot-words to python client, make TC test hot-words everywhere * only run TC tests for C++ and Python * fully expose API in python bindings * expose API in Java (thanks spectie!) * expose API in dotnet (thanks spectie!) * expose API in javascript (thanks spectie!) * java lol * typo in javascript * commenting * java error codes from swig * java docs from SWIG * java and dotnet issues * add hotword test to android tests * dotnet fixes from carlos * add DS_BINARY_PREFIX to tc-asserts.sh for hotwords command * make sure lm is on android for hotword test * path to android model + nit * path * path
58 lines
1.9 KiB
OpenEdge ABL
58 lines
1.9 KiB
OpenEdge ABL
%module swigwrapper
|
|
|
|
%{
|
|
#include "ctc_beam_search_decoder.h"
|
|
#define SWIG_FILE_WITH_INIT
|
|
#define SWIG_PYTHON_STRICT_BYTE_CHAR
|
|
#include "workspace_status.h"
|
|
%}
|
|
|
|
%include <pyabc.i>
|
|
%include <std_string.i>
|
|
%include <std_vector.i>
|
|
%include <std_shared_ptr.i>
|
|
%include <std_unordered_map.i>
|
|
%include "numpy.i"
|
|
|
|
%init %{
|
|
import_array();
|
|
%}
|
|
|
|
namespace std {
|
|
%template(StringVector) vector<string>;
|
|
%template(UnsignedIntVector) vector<unsigned int>;
|
|
%template(OutputVector) vector<Output>;
|
|
%template(OutputVectorVector) vector<vector<Output>>;
|
|
%template(Map) unordered_map<string, float>;
|
|
}
|
|
|
|
%shared_ptr(Scorer);
|
|
|
|
// Convert NumPy arrays to pointer+lengths
|
|
%apply (double* IN_ARRAY2, int DIM1, int DIM2) {(const double *probs, int time_dim, int class_dim)};
|
|
%apply (double* IN_ARRAY3, int DIM1, int DIM2, int DIM3) {(const double *probs, int batch_size, int time_dim, int class_dim)};
|
|
%apply (int* IN_ARRAY1, int DIM1) {(const int *seq_lengths, int seq_lengths_size)};
|
|
%apply (unsigned int* IN_ARRAY1, int DIM1) {(const unsigned int *input, int length)};
|
|
|
|
%ignore Scorer::dictionary;
|
|
|
|
%include "../alphabet.h"
|
|
%include "output.h"
|
|
%include "scorer.h"
|
|
%include "ctc_beam_search_decoder.h"
|
|
|
|
%constant const char* __version__ = ds_version();
|
|
%constant const char* __git_version__ = ds_git_version();
|
|
|
|
// Import only the error code enum definitions from deepspeech.h
|
|
// We can't just do |%ignore "";| here because it affects this file globally (even
|
|
// files %include'd above). That causes SWIG to lose destructor information and
|
|
// leads to leaks of the wrapper objects.
|
|
// Instead we ignore functions and classes (structs), which are the only other
|
|
// things in deepspeech.h. If we add some new construct to deepspeech.h we need
|
|
// to update the ignore rules here to avoid exposing unwanted APIs in the decoder
|
|
// package.
|
|
%rename("$ignore", %$isfunction) "";
|
|
%rename("$ignore", %$isclass) "";
|
|
%include "../deepspeech.h"
|