Update README.md

README.md

@@ -1,7 +1,4 @@
 ---
-license: mit
-base_model:
-- Ultralytics/YOLO11
 ---
 
 # Scientific Context
@@ -25,21 +22,36 @@ base_model:
 
 '''python
 
-### Model Assesment
+model = YOLO('/content/runs/detect/train3/weights/best.pt')
 
-![image/png](https://huggingface.co/OceanCV/Pinniped_Model/resolve/main/F1_curve%20(4).png?download=true)
+#path to Yaml
+dataset_config = '/content/Dataset/data.yaml'
 
-<!-- F1-confidence curves shows us the ideal balance between precision and recall and how that changes with confidence.  My model shows decent F1 scores for CU_non_pup and ZC_non_pup around 0.3 confidence but struggles with all of the other classes.-->
+#Train the model
+results = model.train(
+    data=dataset_config,
+    epochs=50,
+    batch=-1,
+    imgsz=640,
+    plots=True,
+    patience=50,
 
-![Normalized confusion matrrix](https://huggingface.co/OceanCV/Pinniped_Model/resolve/main/confusion_matrix_normalized%20(2).png?download=true)
+)
 
-<!-- Normalized Confusion Matrix shows the percentage of true predictions against all of the other classes in the dataset. Here we can see that my model has difficulties correctly predicting some of my smaller classes as well as the pups.-->
+
+print(results)
+'''
 
 ### Model Assesment
 
-<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app --> This model is to be used by scientist studying Pinniped populations and destributions on the California Coastline. It could be used to show changing population dynamics between eg. California Sea Lions and Northern Fur Seals. 
+![image/png](https://huggingface.co/OceanCV/Pinniped_Model/resolve/main/F1_curve%20(4).png?download=true)
 
+<!-- F1-confidence curves shows us the ideal balance between precision and recall and how that changes with confidence.  My model shows decent F1 scores for CU_non_pup and ZC_non_pup around 0.3 confidence but struggles with all of the other classes.-->
 
+![Normalized confusion matrrix](https://huggingface.co/OceanCV/Pinniped_Model/resolve/main/confusion_matrix_normalized%20(2).png?download=true)
 
+<!-- Normalized Confusion Matrix shows the percentage of true predictions against all of the other classes in the dataset. Here we can see that my model has difficulties correctly predicting some of my smaller classes as well as the pups.-->
 
+### Model Assesment
 
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app --> This model is to be used by scientist studying Pinniped populations and destributions on the California Coastline. It could be used to show changing population dynamics between eg. California Sea Lions and Northern Fur Seals.