Merge pull request #18 from vunb/dev-dtype

Dev dtype
vunb · Mar 7, 2020 · b0a8c49 · b0a8c49
2 parents c11c811 + 17a29a7
commit b0a8c49
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -36,7 +36,9 @@ console.log('Tags: ', tags)
 ```js
 const path = require('path')
 const crfsuite = require('crfsuite')
-const trainer = new crfsuite.Trainer()
+const trainer = new crfsuite.Trainer({
+  debug: true
+})
 
 let model_filename = path.resolve('./model.crfsuite')
 

diff --git a/index.d.ts b/index.d.ts
@@ -7,11 +7,17 @@ export interface Options {
   [key: string]: string;
 }
 
+export interface TrainerOptions {
+  [key: string]: any;
+  debug?: boolean;
+}
+
 export interface TrainerCallback {
   (str: string): void;
 }
 
 export declare class Trainer {
+  constructor(opts?: TrainerOptions);
   append(xseq: Array<string[]>, yseq: string[]): void;
   train(model_filename: string): number;
   get_params(options: Options): any;

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "crfsuite",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "NodeJS binding for CRFsuite",
   "main": "index.js",
   "types": "index.d.ts",

diff --git a/test/specs/trainer.js b/test/specs/trainer.js
@@ -4,7 +4,9 @@ const path = require('path');
 const test = require('tape');
 const crfsuite = require('../..');
 
-const trainer = new crfsuite.Trainer();
+const trainer = new crfsuite.Trainer({
+  debug: false,
+});
 
 
 test('crfsuite trainer', function (t) {