AnoushkaJain3 commited on
Commit
9a9544c
1 Parent(s): 407ee02

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -16
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
- ``` 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
- 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