Drug Classification Using Machine Learning Models
Drug Classification Using Machine Learning Models Data Science Project
Classification in Depth with Scikit-Learn

Drug Classification Using Machine Learning Models

As a medical researcher collecting data for a study, you have gathered information on a group of patients who all suffered from the same illness. Your primary responsibility is to construct a machine model that can identify the most suitable medication for a future patient who has the same illness.

Project Activities

All our Data Science projects include bite-sized activities to test your knowledge and practice in an environment with constant feedback.

All our activities include solutions with explanations on how they work and why we chose them.

codevalidated

Classification Model

Train a Classification model using the training data, and store the model in clf.

For this task, you could use the following models:

- Logistic Regression. 
- Decision Tree Classifier. 
- Naive Bayes Classifier.
- Random Forest Classifier.

However, do not use gradient boosting machines, xgboost or neural networks for this assessment.

Calculate the accuracy of both the training and testing sets, and the macro-average precision, recall and F1-score of the testing set and run the code in a Jupyter Notebook.

Store the results in the variables train_accuracy, test_accuracy, test_precision, test_recall and test_f1score.

Note: The expected evaluation metrics for a simple problem varies depending on the specifics of the problem and data. However, for a well-defined and simple problem with a large and diverse training dataset, a well-trained machine learning model could achieve an precision and recall of over 80% in some cases.

codevalidated

Submission

Now, let's submit your predictions for the test dataset to get a score from the platform.

Drug Classification Using Machine Learning ModelsDrug Classification Using Machine Learning Models
Author

Verónica Barraza

This project is part of

Classification in Depth with Scikit-Learn

Explore other projects