AnoushkaJain3 commited on
Commit
53211f9
1 Parent(s): 9a9544c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -13
README.md CHANGED
@@ -32,20 +32,20 @@ There are two tutorial notebooks:
32
  ```
33
  3. Train_new_model.ipynb
34
 
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
 
 
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