Skip to content

Commit 9277fe8

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Remove validation_split for now.
1 parent e7ab63e commit 9277fe8

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1375,16 +1375,14 @@ def create_dataset(raw_text_samples, tokenizer, sample_expansion_batch_size=10)
13751375
return dataset
13761376

13771377
phase_i_b_dataset = create_dataset(raw_text_samples=phase_i_b_samples, tokenizer=tokenizer, sample_expansion_batch_size=10)
1378-
1379-
# To Do: Set .fit() params <------<<<
13801378

13811379

13821380
phase_i_b_history =\
13831381
best_model_found.fit(
13841382
x=phase_i_b_dataset,
13851383
epochs=phase_i_b_epochs,
1386-
batch_size=batch_size,
1387-
validation_split=0.2)
1384+
batch_size=batch_size) # ,
1385+
# validation_split=0.2)
13881386

13891387
phase_i_b_history =\
13901388
pd.DataFrame(phase_i_b_history.history)

0 commit comments

Comments
 (0)