Commit dcddf57

committed

clean up nn example

1 parent 3aae6d2 commit dcddf57Copy full SHA for dcddf57

File tree

-5

lines changed

-5

lines changed

Lines changed: 5 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -7,9 +7,9 @@ const mlflow = new MLflow('http://localhost:5001');`
`7`	`7`
`8`	`8`	`const HYPERPARAMETER_SPACE = {`
`9`	`9`	`networkArchitectures: [`
`10`		`- [16, 8],// Small network`
`11`		`- [32, 16],// Medium network`
`12`		`- [64, 32],// Larger network`
	`10`	`+ [16, 8],`
	`11`	`+ [32, 16],`
	`12`	`+ [64, 32],`
`13`	`13`	`],`
`14`	`14`	`learningRates: [0.001, 0.01],`
`15`	`15`	`batchSizes: [32, 64],`
`@@ -23,7 +23,7 @@ const TRAINING_CONFIG = {`
`23`	`23`	`datasetSize: 2000,`
`24`	`24`	`inputFeatures: 5,`
`25`	`25`	`outputClasses: 3,`
`26`		`- minibatchSize: 128,// Added for faster training`
	`26`	`+ minibatchSize: 128,`
`27`	`27`	`};`
`28`	`28`
`29`	`29`	`// Data generation`
`@@ -252,7 +252,7 @@ async function main() {`
`252`	`252`	`try {`
`253`	`253`	`console.time('Total Execution Time');`
`254`	`254`
`255`		`- const experimentName = 'Neural_Network_Hyperparameter_Tuning_Fast';`
	`255`	`+ const experimentName = 'Neural_Network_Hyperparameter_Tuning';`
`256`	`256`	`let experimentId;`
`257`	`257`	`try {`
`258`	`258`	`const experiment = await mlflow.getExperimentByName(experimentName);`

Comments

(0)