Uchenna commited on
Commit
bc73213
1 Parent(s): a6c1d68

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -2
README.md CHANGED
@@ -15,34 +15,47 @@ Model
15
  The machine learning model uses a decision tree algorithm to predict student enrollment. The model has been trained on the dataset using 80% of the data for training and 20% for testing. The accuracy of the model is 85%.
16
 
17
  Files
 
18
  This repository contains the following files:
19
 
20
  enrollment_prediction_model.ipynb: Jupyter notebook containing the code for training and testing the model
 
21
  enrollment_prediction_model.pkl: Serialized machine learning model file
 
22
  enrollment_prediction_model_readme.md: Readme file containing information about the machine learning model
 
23
  Usage
 
24
  To use the machine learning model, follow these steps:
25
 
26
  Clone the repository
 
27
  Install the required packages (pandas, numpy, scikit-learn)
 
28
  Load the serialized machine learning model from the enrollment_prediction_model.pkl file
 
29
  Prepare a new dataset with the same columns as the original dataset
 
30
  Use the predict function of the model to predict enrollment for each row in the new dataset
 
31
  Example code:
32
 
33
- python
34
- Copy code
35
  import pandas as pd
 
36
  import pickle
37
 
38
  # Load serialized machine learning model
 
39
  with open('enrollment_prediction_model.pkl', 'rb') as file:
 
40
  model = pickle.load(file)
41
 
42
  # Prepare new dataset
 
43
  new_data = pd.read_csv('new_data.csv')
44
 
45
  # Predict enrollment
 
46
  predictions = model.predict(new_data)
47
 
48
 
 
15
  The machine learning model uses a decision tree algorithm to predict student enrollment. The model has been trained on the dataset using 80% of the data for training and 20% for testing. The accuracy of the model is 85%.
16
 
17
  Files
18
+
19
  This repository contains the following files:
20
 
21
  enrollment_prediction_model.ipynb: Jupyter notebook containing the code for training and testing the model
22
+
23
  enrollment_prediction_model.pkl: Serialized machine learning model file
24
+
25
  enrollment_prediction_model_readme.md: Readme file containing information about the machine learning model
26
+
27
  Usage
28
+
29
  To use the machine learning model, follow these steps:
30
 
31
  Clone the repository
32
+
33
  Install the required packages (pandas, numpy, scikit-learn)
34
+
35
  Load the serialized machine learning model from the enrollment_prediction_model.pkl file
36
+
37
  Prepare a new dataset with the same columns as the original dataset
38
+
39
  Use the predict function of the model to predict enrollment for each row in the new dataset
40
+
41
  Example code:
42
 
 
 
43
  import pandas as pd
44
+
45
  import pickle
46
 
47
  # Load serialized machine learning model
48
+
49
  with open('enrollment_prediction_model.pkl', 'rb') as file:
50
+
51
  model = pickle.load(file)
52
 
53
  # Prepare new dataset
54
+
55
  new_data = pd.read_csv('new_data.csv')
56
 
57
  # Predict enrollment
58
+
59
  predictions = model.predict(new_data)
60
 
61