Skip to content

Commit 892ab07

Browse files
committed
docs: fix predict example to capture .predictions
1 parent 54196fb commit 892ab07

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ from deepmd.dpa_tools import DPAFineTuner
107107

108108
model = DPAFineTuner(pretrained="DPA-3.1-3M", strategy="frozen_sklearn", predictor="rf")
109109
model.fit(train_data="data/train", target_key="bandgap") # fine-tune on your labeled structures
110-
model.predict("data/new_structures") # predict for new structures
110+
preds = model.predict("data/new_structures").predictions # predict for new structures
111111
```
112112

113113
The same workflow is also available from the command line as `dp dpa fit` / `dp dpa predict`.

0 commit comments

Comments
 (0)