AnoushkaJain3
commited on
Commit
•
53211f9
1
Parent(s):
9a9544c
Update README.md
Browse files
README.md
CHANGED
@@ -32,20 +32,20 @@ There are two tutorial notebooks:
|
|
32 |
```
|
33 |
3. Train_new_model.ipynb
|
34 |
|
35 |
-
If you
|
36 |
|
37 |
|
38 |
-
|
39 |
-
|
40 |
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
|
|
|
32 |
```
|
33 |
3. Train_new_model.ipynb
|
34 |
|
35 |
+
If already have your own manually curated on other species you can use this notebook to create your own model.
|
36 |
|
37 |
|
38 |
+
```python
|
39 |
+
from spikeinterface.curation.train_manual_curation import train_model
|
40 |
|
41 |
+
trainer = train_model(mode = "analyzers",
|
42 |
+
labels = labels,
|
43 |
+
analyzers = [labelled_analyzer, labelled_analyzer],
|
44 |
+
output_folder = str(output_folder),
|
45 |
+
imputation_strategies = None,
|
46 |
+
scaling_techniques = None,
|
47 |
+
classifiers = None, # Default to Random Forest only. Other classifiers you can try [ "AdaBoostClassifier","GradientBoostingClassifier",
|
48 |
+
# "LogisticRegression","MLPClassifier"]
|
49 |
+
)
|
50 |
+
```
|
51 |
|