Doc: update orca-in-5-mins prediction. (#6554)
* fix: update orca-in-5-mins dataset volume. * fix: update predict stage.
This commit is contained in:
parent
93351ffd0a
commit
2904af55ed
1 changed files with 1 additions and 2 deletions
|
|
@ -94,8 +94,7 @@ est.fit(data=train_df,
|
|||
# Distributed inference
|
||||
prediction_df = est.predict(test_df,
|
||||
batch_size=batch_size,
|
||||
feature_cols=['user', 'item'],
|
||||
steps=val_steps)
|
||||
feature_cols=['user', 'item'])
|
||||
```
|
||||
|
||||
Stop [Orca Context](orca-context.md) after you finish your program:
|
||||
|
|
|
|||
Loading…
Reference in a new issue