We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 54196fb commit 892ab07Copy full SHA for 892ab07
1 file changed
README.md
@@ -107,7 +107,7 @@ from deepmd.dpa_tools import DPAFineTuner
107
108
model = DPAFineTuner(pretrained="DPA-3.1-3M", strategy="frozen_sklearn", predictor="rf")
109
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
+preds = model.predict("data/new_structures").predictions # predict for new structures
111
```
112
113
The same workflow is also available from the command line as `dp dpa fit` / `dp dpa predict`.
0 commit comments