()
| 53 | * @returns {tf.Sequential} The linear regression model. |
| 54 | */ |
| 55 | export function linearRegressionModel() { |
| 56 | const model = tf.sequential(); |
| 57 | model.add(tf.layers.dense({inputShape: [bostonData.numFeatures], units: 1})); |
| 58 | |
| 59 | model.summary(); |
| 60 | return model; |
| 61 | }; |
| 62 | |
| 63 | /** |
| 64 | * Builds and returns Multi Layer Perceptron Regression Model |