Add roadmap

README.md

@@ -1,3 +1,26 @@
 # ProbeLens
 ### A framework to enable probing of language models.
 ![CI](https://github.com/sharanry/probe-lens/actions/workflows/ci.yaml/badge.svg)
+
+
+## Roadmap
+### Functionalities
+- [x] Add basic linear probe with tests
+- [ ] Add regularization
+- [ ] Add sparsity
+- [ ] Add other kind of probes
+    - [ ] Non-linear probes
+    - [ ] ... ?
+
+### Applications
+- [ ] Reproduce results from SAE-Spelling
+    - [ ] First letter data generation
+- [ ] Add more probing experiments
+    - [ ] ... ?
+
+### Visualization
+- [x] Add Confusion Matrix 
+- [x] Add f2 score and accuracy
+- [ ] Add more visualization experiments
+    - [ ] ... ?
+