Update README.md
Browse files
README.md
CHANGED
|
@@ -53,7 +53,7 @@ We recommend using ACE-Mol as an embedding model; the embed method will create a
|
|
| 53 |
embedded = acemol.embed(molecules, tasks, targets)
|
| 54 |
|
| 55 |
# split into train and test
|
| 56 |
-
train, test =
|
| 57 |
|
| 58 |
# use regress method for regression.
|
| 59 |
predictions = acemol.classify(train, test)
|
|
|
|
| 53 |
embedded = acemol.embed(molecules, tasks, targets)
|
| 54 |
|
| 55 |
# split into train and test
|
| 56 |
+
train, test = embedded[:3], embedded[3:]
|
| 57 |
|
| 58 |
# use regress method for regression.
|
| 59 |
predictions = acemol.classify(train, test)
|