crfsuite/index.d.ts

NodeJS binding for CRFsuite

export declare class Tagger {
  tag(xseq: Array<string[]>): { probability: number; result: string[] };
  open(model_filename: string): boolean;
}

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;
  set_params(options: Options): void;
  set_callback(callback: TrainerCallback): void;
}