AnoushkaJain3
commited on
Commit
•
9a9544c
1
Parent(s):
407ee02
Update README.md
Browse files
README.md
CHANGED
@@ -35,21 +35,17 @@ There are two tutorial notebooks:
|
|
35 |
If you want to create a model based on your own manually curated.
|
36 |
|
37 |
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
best_model = trainer.best_pipeline
|
52 |
-
best_model]
|
53 |
-
)
|
54 |
```
|
55 |
|
|
|
35 |
If you want to create a model based on your own manually curated.
|
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 |
|