chrishalcrow commited on
Commit
99749b6
·
verified ·
1 Parent(s): 2ff3e73

Upload 6 files

Browse files

Initial upload of model

.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ best_model.skops filter=lfs diff=lfs merge=lfs -text
best_model.skops ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:995a938546e30df25444c5e00c907c79ecf871d2ff97c54ffcee1ecf1fc47712
3
+ size 2870628
labels.csv ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ,0
2
+ 0,good
3
+ 1,good
4
+ 2,good
5
+ 3,good
6
+ 4,good
7
+ 5,bad
8
+ 6,bad
9
+ 7,bad
10
+ 8,bad
11
+ 9,bad
metadata.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"subject_species": "Generated data", "brain_area": "N/A", "probe": {"name": "tetrodes"}}
model_accuracies.csv ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ ,classifier name,imputation_strategy,scaling_strategy,accuracy,precision,recall,model_id,best_params
2
+ 0,RandomForestClassifier,median,StandardScaler(),1.0000,1.0000,1.0000,0,"OrderedDict([('class_weight', 'balanced_subsample'), ('criterion', 'gini'), ('min_samples_leaf', 2), ('min_samples_split', 2), ('n_estimators', 125)])"
model_info.json ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric_params": {
3
+ "analyzer_0": {
4
+ "quality_metric_params": {
5
+ "metric_names": [
6
+ "amplitude_cutoff",
7
+ "amplitude_cv",
8
+ "amplitude_median",
9
+ "d_prime",
10
+ "drift",
11
+ "firing_range",
12
+ "firing_rate",
13
+ "isi_violation",
14
+ "isolation_distance",
15
+ "l_ratio",
16
+ "nearest_neighbor",
17
+ "num_spikes",
18
+ "presence_ratio",
19
+ "rp_violation",
20
+ "sd_ratio",
21
+ "silhouette",
22
+ "sliding_rp_violation",
23
+ "snr",
24
+ "synchrony"
25
+ ],
26
+ "peak_sign": null,
27
+ "seed": null,
28
+ "qm_params": {
29
+ "presence_ratio": {
30
+ "bin_duration_s": 60,
31
+ "mean_fr_ratio_thresh": 0.0
32
+ },
33
+ "snr": {
34
+ "peak_sign": "neg",
35
+ "peak_mode": "extremum"
36
+ },
37
+ "isi_violation": {
38
+ "isi_threshold_ms": 1.5,
39
+ "min_isi_ms": 0
40
+ },
41
+ "rp_violation": {
42
+ "refractory_period_ms": 1.0,
43
+ "censored_period_ms": 0.0
44
+ },
45
+ "sliding_rp_violation": {
46
+ "min_spikes": 0,
47
+ "bin_size_ms": 0.25,
48
+ "window_size_s": 1,
49
+ "exclude_ref_period_below_ms": 0.5,
50
+ "max_ref_period_ms": 10,
51
+ "contamination_values": null
52
+ },
53
+ "synchrony": {
54
+ "synchrony_sizes": [
55
+ 2,
56
+ 4,
57
+ 8
58
+ ]
59
+ },
60
+ "firing_range": {
61
+ "bin_size_s": 5,
62
+ "percentiles": [
63
+ 5,
64
+ 95
65
+ ]
66
+ },
67
+ "amplitude_cv": {
68
+ "average_num_spikes_per_bin": 50,
69
+ "percentiles": [
70
+ 5,
71
+ 95
72
+ ],
73
+ "min_num_bins": 10,
74
+ "amplitude_extension": "spike_amplitudes"
75
+ },
76
+ "amplitude_cutoff": {
77
+ "peak_sign": "neg",
78
+ "num_histogram_bins": 100,
79
+ "histogram_smoothing_value": 3,
80
+ "amplitudes_bins_min_ratio": 5
81
+ },
82
+ "amplitude_median": {
83
+ "peak_sign": "neg"
84
+ },
85
+ "drift": {
86
+ "interval_s": 60,
87
+ "min_spikes_per_interval": 100,
88
+ "direction": "y",
89
+ "min_num_bins": 2
90
+ },
91
+ "nearest_neighbor": {
92
+ "max_spikes": 10000,
93
+ "n_neighbors": 5
94
+ },
95
+ "nn_isolation": {
96
+ "max_spikes": 10000,
97
+ "min_spikes": 10,
98
+ "min_fr": 0.0,
99
+ "n_neighbors": 4,
100
+ "n_components": 10,
101
+ "radius_um": 100,
102
+ "peak_sign": "neg"
103
+ },
104
+ "nn_noise_overlap": {
105
+ "max_spikes": 10000,
106
+ "min_spikes": 10,
107
+ "min_fr": 0.0,
108
+ "n_neighbors": 4,
109
+ "n_components": 10,
110
+ "radius_um": 100,
111
+ "peak_sign": "neg"
112
+ },
113
+ "silhouette": {
114
+ "method": [
115
+ "simplified"
116
+ ]
117
+ }
118
+ },
119
+ "skip_pc_metrics": false
120
+ },
121
+ "template_metric_params": {
122
+ "metric_names": [
123
+ "half_width",
124
+ "num_negative_peaks",
125
+ "num_positive_peaks",
126
+ "peak_to_valley",
127
+ "peak_trough_ratio",
128
+ "recovery_slope",
129
+ "repolarization_slope"
130
+ ],
131
+ "sparsity": null,
132
+ "peak_sign": "neg",
133
+ "upsampling_factor": 10,
134
+ "metrics_kwargs": {
135
+ "recovery_window_ms": 0.7,
136
+ "peak_relative_threshold": 0.2,
137
+ "peak_width_ms": 0.1,
138
+ "depth_direction": "y",
139
+ "min_channels_for_velocity": 5,
140
+ "min_r2_velocity": 0.5,
141
+ "exp_peak_function": "ptp",
142
+ "min_r2_exp_decay": 0.5,
143
+ "spread_threshold": 0.2,
144
+ "spread_smooth_um": 20,
145
+ "column_range": null
146
+ }
147
+ }
148
+ }
149
+ },
150
+ "requirements": {
151
+ "spikeinterface": "0.101.1",
152
+ "scikit-learn": "1.4.0"
153
+ },
154
+ "label_conversion": {
155
+ "1": "good",
156
+ "0": "bad"
157
+ }
158
+ }
training_data.csv ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ,amplitude_cutoff,amplitude_cv_median,amplitude_cv_range,amplitude_median,drift_ptp,drift_std,drift_mad,firing_range,firing_rate,isi_violations_ratio,isi_violations_count,num_spikes,presence_ratio,rp_contamination,rp_violations,sd_ratio,sliding_rp_violation,snr,sync_spike_2,sync_spike_4,sync_spike_8,d_prime,isolation_distance,l_ratio,silhouette,nn_hit_rate,nn_miss_rate,half_width,num_negative_peaks,num_positive_peaks,peak_to_valley,peak_trough_ratio,recovery_slope,repolarization_slope
2
+ 0,,,,-105.55292,,,,1.2599999999999998,14.1,0.0,0.0,141.0,,0.0,0.0,1.4275525575895887,0.13,19.94376038512596,0.014184397163120567,0.0,0.0,14.32472020529736,754.333174814516,3.498809474378825e-05,0.6047805009662675,0.9148936170212766,0.005304101838755304,0.000272,1,1,0.0007,-0.2603113109282029,-34669.6033764812,289954.4413997247
3
+ 1,,,,-33.25789,,,,0.360000000000003,16.4,0.0,0.0,164.0,,0.0,0.0,1.062721238972397,0.1,6.385192930548802,0.012195121951219513,0.0,0.0,5.550849214038404,93.81342566277144,0.025624156435374426,0.3462939490773019,0.8429878048780488,0.01689432063263839,0.000268,1,0,0.000688,-0.1654153151935468,-6686.657520361004,85404.06725637885
4
+ 2,,,,-39.264847,,,,1.4399999999999995,14.6,0.0,0.0,146.0,,0.0,0.0,0.7823625169831282,0.13,7.510754986318674,0.0136986301369863,0.0,0.0,5.174324857350859,70.47972600619985,0.0458272822421763,0.33222679845555547,0.7825342465753424,0.01738821859474805,0.00022,1,0,0.000496,-0.18190749934219932,-9349.811468026372,119624.20539646724
5
+ 3,,,,-150.3613,,,,3.4200000000000017,14.1,0.0,0.0,141.0,,0.0,0.0,1.2420705833860484,0.14,28.31617873073055,0.0070921985815602835,0.0,0.0,20.091786877470362,1368.6948507775646,0.0,0.6278975906072741,0.9539007092198581,0.0031824611032531826,0.000232,1,1,0.000684,-0.28473928913839414,-53956.39289013097,395887.74012533476
6
+ 4,,,,-32.15467,,,,3.7799999999999976,15.3,0.0,0.0,153.0,,0.0,0.0,1.0693804845727417,0.095,6.303967028588283,0.013071895424836602,0.0,0.0,3.3481204356228353,50.179048330926186,0.0616623888316366,0.29690613159496176,0.8529411764705882,0.015335235378031383,0.000256,1,1,0.000564,-0.32801763391376826,-15064.650919764095,124739.97152224768
7
+ 5,,,,0.3231091,,,,0.1800000000000015,15.3,0.0,0.0,153.0,,0.0,0.0,1.7210704419869174,0.11,0.6333596333132606,0.0,0.0,0.0,1.346984042510883,6.256928146523261,3.2203025514597847,-0.05202430885993671,0.1781045751633987,0.09432952924393723,0.001476,2,2,0.000764,-0.6075884125832761,-2393.295952661818,6342.494741897144
8
+ 6,,,,0.30091238,,,,3.7799999999999994,17.5,0.0,0.0,175.0,,0.0,0.0,1.592399979097494,0.09,0.6365438885454862,0.017142857142857144,0.0,0.0,1.3638968138893786,7.5953560548625445,2.608304536554279,-0.007394267230338126,0.18857142857142858,0.10036231884057971,0.001644,2,3,0.001112,-0.37536001353663767,-3332.269394156418,5741.393888220467
9
+ 7,,,,0.26962733,,,,1.259999999999998,16.5,0.0,0.0,165.0,,0.0,0.0,1.752057118043708,0.1,0.5895421305481752,0.0,0.0,0.0,1.448931119036185,7.868476893669308,2.723481862466978,0.0006757284458100525,0.18484848484848485,0.09820143884892087,0.00236,3,2,0.00042,-0.38427887346555645,-1368.9036206312949,11815.0067873371
10
+ 8,,,,-0.6511212,,,,5.399999999999997,17.4,0.0,0.0,174.0,,0.0,0.0,1.2713837562553023,0.085,0.400033710760959,0.0,0.0,0.0,1.4423993470283136,9.103022751770219,2.415844781652313,0.01335317653129366,0.1853448275862069,0.10445329471397538,0.001744,2,3,0.00012,0.05718339921922215,,
11
+ 9,,,,-0.5781461,,,,1.9800000000000022,14.3,0.0,0.0,143.0,,0.0,0.0,1.0529652237146974,0.145,0.4181589882853068,0.013986013986013986,0.0,0.0,1.1875053894770542,6.951082925089799,3.455085804157728,0.0022930177606275298,0.1381118881118881,0.09313031161473088,0.001852,2,3,0.000816,-0.5768069143478995,-2028.997636162123,5328.396822277372