joesterle commited on
Commit
d49b0eb
·
verified ·
1 Parent(s): 3e0fd21

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -3
README.md CHANGED
@@ -1,3 +1,21 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - sklearn
5
+ - random-forest
6
+ - calibrated-classifier
7
+ library_name: sklearn
8
+ ---
9
+
10
+ # Calibrated Random Forest Classifier
11
+
12
+ This is a calibrated Random Forest classifier trained with scikit-learn on two-photon (2p) ganglion cell layer (GCL) responses from the mouse retina.
13
+ The training data is from Baden et al. 2016 (https://datadryad.org/dataset/doi:10.5061/dryad.d9v38).
14
+
15
+ ## Usage
16
+ ```python
17
+ from gcl_classifier import model
18
+
19
+ predictions = model.predict(X_test)
20
+ probabilities = model.predict_proba(X_test)
21
+ ```