bs

Borda · Borda · commit 06cff9d5d697 · 2022-04-15T11:39:14.000+02:00
diff --git a/templates/titanic/tutorial.py b/templates/titanic/tutorial.py
@@ -92,6 +92,7 @@
 dm = TabularClassificationData.from_data_frame(
     predict_data_frame=df_test,
     parameters=datamodule.parameters,
+    batch_size=datamodule.batch_size,
 )
 predictions = trainer.predict(model, datamodule=dm, output="classes")
 print(predictions[0])

Original file line number	Diff line number	Diff line change
`@@ -92,6 +92,7 @@`
`92`	`92`	`dm = TabularClassificationData.from_data_frame(`
`93`	`93`	`predict_data_frame=df_test,`
`94`	`94`	`parameters=datamodule.parameters,`
	`95`	`+ batch_size=datamodule.batch_size,`
`95`	`96`	`)`
`96`	`97`	`predictions = trainer.predict(model, datamodule=dm, output="classes")`
`97`	`98`	`print(predictions[0])`