updated and added formatted outputs

repetere · Jul 28, 2018 · 62afc93 · 62afc93
1 parent 123ea11
commit 62afc93
Show file tree

Hide file tree

Showing 15 changed files with 2,486 additions and 1,840 deletions.
diff --git a/docs/API.md b/docs/API.md
@@ -6,23 +6,27 @@ Mulitple Linear Regression with Tensorflow
 
 ### `constructor(options: Object, customTF: Object)`
 
+### `yShape: *`
+
+### `xShape: *`
+
 ### `model: *`
 
-### `train(x_matrix: *, y_matrix: *): Object`
+### `train(x_matrix: Array<Array<number>>, y_matrix: Array<Array<number>>): Object`
 
 Asynchronously trains tensorflow model
 
 | Name | Type | Attribute | Description |
 | --- | --- | --- | --- |
-| x_matrix | * |  | independent variables |
-| y_matrix | * |  | dependent variables |
+| x_matrix | Array<Array<number>> |  | independent variables |
+| y_matrix | Array<Array<number>> |  | dependent variables |
 
-### `calculate(matrix: *): Promise`
+### `calculate(matrix: Array<Array<number>>|Array<number>): {data: Promise}`
 
 Predicts new dependent variables
 
 | Name | Type | Attribute | Description |
 | --- | --- | --- | --- |
-| matrix | * |  | new test independent variables |
+| matrix | Array<Array<number>>|Array<number> |  | new test independent variables |
 
 # Function