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
|
# Distributed inference
|
||||||
prediction_df = est.predict(test_df,
|
prediction_df = est.predict(test_df,
|
||||||
batch_size=batch_size,
|
batch_size=batch_size,
|
||||||
feature_cols=['user', 'item'],
|
feature_cols=['user', 'item'])
|
||||||
steps=val_steps)
|
|
||||||
```
|
```
|
||||||
|
|
||||||
Stop [Orca Context](orca-context.md) after you finish your program:
|
Stop [Orca Context](orca-context.md) after you finish your program:
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue