vhad/DefinitelyTyped
1
0
Fork 0
mirror of https://github.com/gosticks/DefinitelyTyped.git synced 2026-06-30 23:30:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added classification definition to classifiers

Browse Source
This commit is contained in:
Yonatan Bentzur
2019-04-10 23:11:40 -07:00
parent 55ece09ae5
commit 91beabb556
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
types/natural/index.d.ts vendored
View File

@@ -1,4 +1,4 @@
// Type definitions for Natural 0.2.2
// Type definitions for Natural v0.6.3
// Project: https://github.com/NaturalNode/natural
// Definitions by: Dylan R. E. Moonfire <https://github.com/dmoonfire>
// Definitions: https://github.com/DefinitelyTyped/DefinitelyTyped
@@ -67,6 +67,7 @@ declare var LancasterStemmer: {
stem(token: string): string;
}
interface BayesClassifierClassification { label: string, value: number }
interface BayesClassifierCallback { (err: any, classifier: any): void }
declare class BayesClassifier {
events: events.EventEmitter;
@@ -74,12 +75,13 @@ declare class BayesClassifier {
addDocument(text: string[], stem: string): void;
train(): void;
classify(observation: string): string;
getClassifications(observation: string): string[];
getClassifications(observation: string): BayesClassifierClassification[];
save(filename: string, callback: BayesClassifierCallback): void;
static load(filename: string, stemmer: Stemmer, callback: BayesClassifierCallback): void;
static restore(classifier: any, stemmer?: Stemmer): BayesClassifier;
}
interface LogisticRegressionClassifierClassification { label: string, value: number }
interface LogisticRegressionClassifierCallback { (err: any, classifier: any): void }
declare class LogisticRegressionClassifier {
events: events.EventEmitter;
@@ -87,7 +89,7 @@ declare class LogisticRegressionClassifier {
addDocument(text: string[], stem: string): void;
train(): void;
classify(observation: string): string;
getClassifications(observation: string): string[];
getClassifications(observation: string): LogisticRegressionClassifierClassification[];
save(filename: string, callback: LogisticRegressionClassifierCallback): void;
static load(filename: string, stemmer: Stemmer, callback: LogisticRegressionClassifierCallback): void;
static restore(classifier: any, stemmer?: Stemmer): LogisticRegressionClassifier;