Project-10-MCM/project-10-mcm.ipynb

{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":974,"sourceType":"datasetVersion","datasetId":478}],"dockerImageVersionId":30839,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"# **OPEN-ARC**\n---\n\n### Project 10: Mushroom Classification Model:\n**Challenge:** Create an AI model, capable of classifying whether a mushroom is edible or poisonous based on a set of features.\n\n\n### Terms and Use:\nLearn more about the project's [LICENSE](https://github.com/Infinitode/OPEN-ARC/blob/main/LICENSE) and read our [CODE_OF_CONDUCT](https://github.com/Infinitode/OPEN-ARC/blob/main/CODE_OF_CONDUCT) before contributing to the project. You can contribute to this project from here: [https://github.com/Infinitode/OPEN-ARC/](https://github.com/Infinitode/OPEN-ARC/).\n\n---","metadata":{}},{"cell_type":"markdown","source":"Please fill out this performance sheet to help others quickly see your model's performance **(optional)**:\n\n### Performance Sheet:\n| Contributor | Architecture Type | Platform | Base Model | Dataset | Accuracy | Link |\n|-------------|-------------------|----------|------------|---------|----------|------|\n| Infinitode  | RandomForestClassifier  | Kaggle   | ✔  | Mushroom Classification | 91.1% (CV)    | [Notebook](https://github.com/Infinitode/OPEN-ARC/blob/main/Project-10-MCM/project-10-mcm.ipynb) |\n| Username  | Unknown  | Kaggle   | ✗/✔  | Mushroom Classification | Score    | [Notebook](https://github.com) |\n\n---","metadata":{}},{"cell_type":"markdown","source":"**Disclaimer:** This model is for **educational purposes only** and should not be used for real-life mushroom classification or any decision-making processes related to the consumption of mushrooms. While the model performs well on the provided dataset, it has not been thoroughly validated for real-world scenarios and may not accurately detect poisonous mushrooms in all conditions. Always consult an expert or use trusted resources when identifying mushrooms.","metadata":{}},{"cell_type":"markdown","source":"### Using `RandomForestClassifier`\nFor this dataset, we decided to use a simple `RandomForestClassifier` along with automatic class balancing.","metadata":{}},{"cell_type":"code","source":"import pandas as pd\nimport numpy as np\nfrom sklearn.model_selection import train_test_split\nfrom sklearn.ensemble import RandomForestClassifier\nfrom sklearn.metrics import classification_report\nimport joblib\n\n# Load the dataset from our inputs\nurl = '/kaggle/input/mushroom-classification/mushrooms.csv'\ndf = pd.read_csv(url)\n\n# Display basic dataset information\nprint(df.info())\nprint(df.head())\n\n# Define the mappings for each feature\nmappings = {\n    'class': {'e': 0, 'p': 1},\n    'cap-shape': {'b': 0, 'c': 1, 'x': 2, 'f': 3, 'k': 4, 's': 5},\n    'cap-surface': {'f': 0, 'g': 1, 'y': 2, 's': 3},\n    'cap-color': {'n': 0, 'b': 1, 'c': 2, 'g': 3, 'r': 4, 'p': 5, 'u': 6, 'e': 7, 'w': 8, 'y': 9},\n    'bruises': {'t': 1, 'f': 0},\n    'odor': {'a': 0, 'l': 1, 'c': 2, 'y': 3, 'f': 4, 'm': 5, 'n': 6, 'p': 7, 's': 8},\n    'gill-attachment': {'a': 0, 'd': 1, 'f': 2, 'n': 3},\n    'gill-spacing': {'c': 0, 'w': 1, 'd': 2},\n    'gill-size': {'b': 0, 'n': 1},\n    'gill-color': {'k': 0, 'n': 1, 'b': 2, 'h': 3, 'g': 4, 'r': 5, 'o': 6, 'p': 7, 'u': 8, 'e': 9, 'w': 10, 'y': 11},\n    'stalk-shape': {'e': 0, 't': 1},\n    'stalk-root': {'b': 0, 'c': 1, 'u': 2, 'e': 3, 'z': 4, 'r': 5, '?': 6},\n    'stalk-surface-above-ring': {'f': 0, 'y': 1, 'k': 2, 's': 3},\n    'stalk-surface-below-ring': {'f': 0, 'y': 1, 'k': 2, 's': 3},\n    'stalk-color-above-ring': {'n': 0, 'b': 1, 'c': 2, 'g': 3, 'o': 4, 'p': 5, 'e': 6, 'w': 7, 'y': 8},\n    'stalk-color-below-ring': {'n': 0, 'b': 1, 'c': 2, 'g': 3, 'o': 4, 'p': 5, 'e': 6, 'w': 7, 'y': 8},\n    'veil-type': {'p': 0, 'u': 1},\n    'veil-color': {'n': 0, 'o': 1, 'w': 2, 'y': 3},\n    'ring-number': {'n': 0, 'o': 1, 't': 2},\n    'ring-type': {'c': 0, 'e': 1, 'f': 2, 'l': 3, 'n': 4, 'p': 5, 's': 6, 'z': 7},\n    'spore-print-color': {'k': 0, 'n': 1, 'b': 2, 'h': 3, 'r': 4, 'o': 5, 'u': 6, 'w': 7, 'y': 8},\n    'population': {'a': 0, 'c': 1, 'n': 2, 's': 3, 'v': 4, 'y': 5},\n    'habitat': {'g': 0, 'l': 1, 'm': 2, 'p': 3, 'u': 4, 'w': 5, 'd': 6}\n}\n\n# Map the values to numerical values\nfor column, mapping in mappings.items():\n    df[column] = df[column].map(mapping)\n\n# Separate the target from features\nX = df.drop('class', axis=1)\ny = df['class']\n\n# Split the data\nX_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)\n\n# Initialize the default model with class weight balancing\nmodel = RandomForestClassifier(class_weight='balanced', random_state=42)\n\n# Train the model\nmodel.fit(X_train, y_train)\n\n# Make predictions\ny_pred = model.predict(X_test)\n\n# Evaluate the model\nprint(classification_report(y_test, y_pred))\n\n# Save the model\njoblib.dump(model, 'mushroom_classifier.pkl')\n\n# Save the mappings\njoblib.dump(mappings, 'mappings.pkl')","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true,"execution":{"iopub.status.busy":"2025-02-05T08:40:29.924795Z","iopub.execute_input":"2025-02-05T08:40:29.925204Z","iopub.status.idle":"2025-02-05T08:40:30.410639Z","shell.execute_reply.started":"2025-02-05T08:40:29.925174Z","shell.execute_reply":"2025-02-05T08:40:30.409735Z"}},"outputs":[{"name":"stdout","text":"<class 'pandas.core.frame.DataFrame'>\nRangeIndex: 8124 entries, 0 to 8123\nData columns (total 23 columns):\n #   Column                    Non-Null Count  Dtype \n---  ------                    --------------  ----- \n 0   class                     8124 non-null   object\n 1   cap-shape                 8124 non-null   object\n 2   cap-surface               8124 non-null   object\n 3   cap-color                 8124 non-null   object\n 4   bruises                   8124 non-null   object\n 5   odor                      8124 non-null   object\n 6   gill-attachment           8124 non-null   object\n 7   gill-spacing              8124 non-null   object\n 8   gill-size                 8124 non-null   object\n 9   gill-color                8124 non-null   object\n 10  stalk-shape               8124 non-null   object\n 11  stalk-root                8124 non-null   object\n 12  stalk-surface-above-ring  8124 non-null   object\n 13  stalk-surface-below-ring  8124 non-null   object\n 14  stalk-color-above-ring    8124 non-null   object\n 15  stalk-color-below-ring    8124 non-null   object\n 16  veil-type                 8124 non-null   object\n 17  veil-color                8124 non-null   object\n 18  ring-number               8124 non-null   object\n 19  ring-type                 8124 non-null   object\n 20  spore-print-color         8124 non-null   object\n 21  population                8124 non-null   object\n 22  habitat                   8124 non-null   object\ndtypes: object(23)\nmemory usage: 1.4+ MB\nNone\n  class cap-shape cap-surface cap-color bruises odor gill-attachment  \\\n0     p         x           s         n       t    p               f   \n1     e         x           s         y       t    a               f   \n2     e         b           s         w       t    l               f   \n3     p         x           y         w       t    p               f   \n4     e         x           s         g       f    n               f   \n\n  gill-spacing gill-size gill-color  ... stalk-surface-below-ring  \\\n0            c         n          k  ...                        s   \n1            c         b          k  ...                        s   \n2            c         b          n  ...                        s   \n3            c         n          n  ...                        s   \n4            w         b          k  ...                        s   \n\n  stalk-color-above-ring stalk-color-below-ring veil-type veil-color  \\\n0                      w                      w         p          w   \n1                      w                      w         p          w   \n2                      w                      w         p          w   \n3                      w                      w         p          w   \n4                      w                      w         p          w   \n\n  ring-number ring-type spore-print-color population habitat  \n0           o         p                 k          s       u  \n1           o         p                 n          n       g  \n2           o         p                 n          n       m  \n3           o         p                 k          s       u  \n4           o         e                 n          a       g  \n\n[5 rows x 23 columns]\n              precision    recall  f1-score   support\n\n           0       1.00      1.00      1.00       843\n           1       1.00      1.00      1.00       782\n\n    accuracy                           1.00      1625\n   macro avg       1.00      1.00      1.00      1625\nweighted avg       1.00      1.00      1.00      1625\n\n","output_type":"stream"},{"execution_count":3,"output_type":"execute_result","data":{"text/plain":"['mappings.pkl']"},"metadata":{}}],"execution_count":3},{"cell_type":"markdown","source":"Since our classifcation report showed perfect results, with `precision`, `recall`, and `F1` all being 1, with an accuracy of `100%`, we'll use cross-validation testing to evaluate how the model performs on certain parts of the dataset.","metadata":{}},{"cell_type":"code","source":"from sklearn.model_selection import cross_val_score\n\n# Perform 5-fold cross-validation\ncv_scores = cross_val_score(model, X, y, cv=5)\n\nprint(f\"Cross-validation scores: {cv_scores}\")\nprint(f\"Average cross-validation score: {cv_scores.mean()}\")","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T08:40:30.412129Z","iopub.execute_input":"2025-02-05T08:40:30.412478Z","iopub.status.idle":"2025-02-05T08:40:32.090975Z","shell.execute_reply.started":"2025-02-05T08:40:30.412451Z","shell.execute_reply":"2025-02-05T08:40:32.089801Z"}},"outputs":[{"name":"stdout","text":"Cross-validation scores: [0.84246154 1.         1.         1.         0.71059113]\nAverage cross-validation score: 0.910610534293293\n","output_type":"stream"}],"execution_count":4},{"cell_type":"markdown","source":"A cross-validation score of `0.91` or `91.06%` is quite remarkable, but as shown in the cross-validation testing, the model struggles with certain splits of the dataset. This indicates that these parts of the dataset might contain more difficult problems. It can also show that our model has potentially overfitted on the simpler examples from our dataset.","metadata":{}},{"cell_type":"markdown","source":"### Testing the model\nWe'll now define feature maps, to make it easier to understand the model's input data.","metadata":{}},{"cell_type":"code","source":"feature_options = {\n    'cap-shape': {'b': 'bell', 'c': 'conical', 'x': 'convex', 'f': 'flat', 'k': 'knobbed', 's': 'sunken'},\n    'cap-surface': {'f': 'fibrous', 'g': 'grooves', 'y': 'scaly', 's': 'smooth'},\n    'cap-color': {'n': 'brown', 'b': 'buff', 'c': 'cinnamon', 'g': 'gray', 'r': 'green', 'p': 'pink', 'u': 'purple', 'e': 'red', 'w': 'white', 'y': 'yellow'},\n    'bruises': {'t': 'bruises', 'f': 'no'},\n    'odor': {'a': 'almond', 'l': 'anise', 'c': 'creosote', 'y': 'fishy', 'f': 'foul', 'm': 'musty', 'n': 'none', 'p': 'pungent', 's': 'spicy'},\n    'gill-attachment': {'a': 'attached', 'd': 'descending', 'f': 'free', 'n': 'notched'},\n    'gill-spacing': {'c': 'close', 'w': 'crowded', 'd': 'distant'},\n    'gill-size': {'b': 'broad', 'n': 'narrow'},\n    'gill-color': {'k': 'black', 'n': 'brown', 'b': 'buff', 'h': 'chocolate', 'g': 'gray', 'r': 'green', 'o': 'orange', 'p': 'pink', 'u': 'purple', 'e': 'red', 'w': 'white', 'y': 'yellow'},\n    'stalk-shape': {'e': 'enlarging', 't': 'tapering'},\n    'stalk-root': {'b': 'bulbous', 'c': 'club', 'u': 'cup', 'e': 'equal', 'z': 'rhizomorphs', 'r': 'rooted', '?': 'missing'},\n    'stalk-surface-above-ring': {'f': 'fibrous', 'y': 'scaly', 'k': 'silky', 's': 'smooth'},\n    'stalk-surface-below-ring': {'f': 'fibrous', 'y': 'scaly', 'k': 'silky', 's': 'smooth'},\n    'stalk-color-above-ring': {'n': 'brown', 'b': 'buff', 'c': 'cinnamon', 'g': 'gray', 'o': 'orange', 'p': 'pink', 'e': 'red', 'w': 'white', 'y': 'yellow'},\n    'stalk-color-below-ring': {'n': 'brown', 'b': 'buff', 'c': 'cinnamon', 'g': 'gray', 'o': 'orange', 'p': 'pink', 'e': 'red', 'w': 'white', 'y': 'yellow'},\n    'veil-type': {'p': 'partial', 'u': 'universal'},\n    'veil-color': {'n': 'brown', 'o': 'orange', 'w': 'white', 'y': 'yellow'},\n    'ring-number': {'n': 'none', 'o': 'one', 't': 'two'},\n    'ring-type': {'c': 'cobwebby', 'e': 'evanescent', 'f': 'flaring', 'l': 'large', 'n': 'none', 'p': 'pendant', 's': 'sheathing', 'z': 'zone'},\n    'spore-print-color': {'k': 'black', 'n': 'brown', 'b': 'buff', 'h': 'chocolate', 'r': 'green', 'o': 'orange', 'u': 'purple', 'w': 'white', 'y': 'yellow'},\n    'population': {'a': 'abundant', 'c': 'clustered', 'n': 'numerous', 's': 'scattered', 'v': 'several', 'y': 'solitary'},\n    'habitat': {'g': 'grasses', 'l': 'leaves', 'm': 'meadows', 'p': 'paths', 'u': 'urban', 'w': 'waste', 'd': 'woods'}\n}\n\ndef get_user_input():\n    \"\"\"\n    Collects user input for each mushroom feature.\n\n    Returns:\n    dict: A dictionary containing the user's input for each feature.\n    \"\"\"\n    user_input = {}\n    print(\"Please provide the following mushroom characteristics:\")\n    for feature, options in feature_options.items():\n        print(f\"\\n{feature.replace('-', ' ').capitalize()}:\")\n        for key, value in options.items():\n            print(f\"  {key}: {value}\")\n        while True:\n            choice = input(f\"Enter the corresponding letter for {feature}: \").strip().lower()\n            if choice in options:\n                user_input[feature] = choice\n                break\n            else:\n                print(\"Invalid input. Please enter one of the listed letters.\")\n    return user_input\n\nuser_input = get_user_input()\n\ndef predict_mushroom(features):\n    \"\"\"\n    Predict whether a mushroom is edible or poisonous based on its features.\n    \n    Parameters:\n    features (dict): A dictionary of mushroom features with feature names as keys and corresponding categorical values.\n    \n    Returns:\n    str: 'Edible' or 'Poisonous'\n    \"\"\"\n    # Load the trained model and mappings\n    model = joblib.load('mushroom_classifier.pkl')\n    mappings = joblib.load('mappings.pkl')\n    \n    # Initialize a dictionary to hold the numerical features\n    numerical_features = {}\n    \n    # Map each feature to its numerical value\n    for feature, value in features.items():\n        if feature in mappings:\n            if value in mappings[feature]:\n                numerical_features[feature] = mappings[feature][value]\n            else:\n                raise ValueError(f\"Invalid value '{value}' for feature '{feature}'.\")\n        else:\n            raise ValueError(f\"Feature '{feature}' is not recognized.\")\n    \n    # Convert the numerical features into a DataFrame\n    input_df = pd.DataFrame([numerical_features])\n    \n    # Predict using the trained model\n    prediction = model.predict(input_df)\n    \n    # Interpret the prediction\n    if prediction[0] == 0:\n        return 'Edible'\n    else:\n        return 'Poisonous'\n\n# Predict edibility\ntry:\n    result = predict_mushroom(user_input)\n    print(f\"\\nThe mushroom is likely: {result}\")\nexcept ValueError as e:\n    print(f\"Error: {e}\")","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T08:40:32.092956Z","iopub.execute_input":"2025-02-05T08:40:32.093366Z","iopub.status.idle":"2025-02-05T08:41:34.148651Z","shell.execute_reply.started":"2025-02-05T08:40:32.093329Z","shell.execute_reply":"2025-02-05T08:41:34.147645Z"}},"outputs":[{"name":"stdout","text":"Please provide the following mushroom characteristics:\n\nCap shape:\n  b: bell\n  c: conical\n  x: convex\n  f: flat\n  k: knobbed\n  s: sunken\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for cap-shape:  b\n"},{"name":"stdout","text":"\nCap surface:\n  f: fibrous\n  g: grooves\n  y: scaly\n  s: smooth\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for cap-surface:  f\n"},{"name":"stdout","text":"\nCap color:\n  n: brown\n  b: buff\n  c: cinnamon\n  g: gray\n  r: green\n  p: pink\n  u: purple\n  e: red\n  w: white\n  y: yellow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for cap-color:  n\n"},{"name":"stdout","text":"\nBruises:\n  t: bruises\n  f: no\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for bruises:  t\n"},{"name":"stdout","text":"\nOdor:\n  a: almond\n  l: anise\n  c: creosote\n  y: fishy\n  f: foul\n  m: musty\n  n: none\n  p: pungent\n  s: spicy\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for odor:  a\n"},{"name":"stdout","text":"\nGill attachment:\n  a: attached\n  d: descending\n  f: free\n  n: notched\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for gill-attachment:  a\n"},{"name":"stdout","text":"\nGill spacing:\n  c: close\n  w: crowded\n  d: distant\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for gill-spacing:  c\n"},{"name":"stdout","text":"\nGill size:\n  b: broad\n  n: narrow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for gill-size:  b\n"},{"name":"stdout","text":"\nGill color:\n  k: black\n  n: brown\n  b: buff\n  h: chocolate\n  g: gray\n  r: green\n  o: orange\n  p: pink\n  u: purple\n  e: red\n  w: white\n  y: yellow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for gill-color:  k\n"},{"name":"stdout","text":"\nStalk shape:\n  e: enlarging\n  t: tapering\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for stalk-shape:  e\n"},{"name":"stdout","text":"\nStalk root:\n  b: bulbous\n  c: club\n  u: cup\n  e: equal\n  z: rhizomorphs\n  r: rooted\n  ?: missing\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for stalk-root:  b\n"},{"name":"stdout","text":"\nStalk surface above ring:\n  f: fibrous\n  y: scaly\n  k: silky\n  s: smooth\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for stalk-surface-above-ring:  f\n"},{"name":"stdout","text":"\nStalk surface below ring:\n  f: fibrous\n  y: scaly\n  k: silky\n  s: smooth\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for stalk-surface-below-ring:  f\n"},{"name":"stdout","text":"\nStalk color above ring:\n  n: brown\n  b: buff\n  c: cinnamon\n  g: gray\n  o: orange\n  p: pink\n  e: red\n  w: white\n  y: yellow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for stalk-color-above-ring:  n\n"},{"name":"stdout","text":"\nStalk color below ring:\n  n: brown\n  b: buff\n  c: cinnamon\n  g: gray\n  o: orange\n  p: pink\n  e: red\n  w: white\n  y: yellow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for stalk-color-below-ring:  n\n"},{"name":"stdout","text":"\nVeil type:\n  p: partial\n  u: universal\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for veil-type:  p\n"},{"name":"stdout","text":"\nVeil color:\n  n: brown\n  o: orange\n  w: white\n  y: yellow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for veil-color:  n\n"},{"name":"stdout","text":"\nRing number:\n  n: none\n  o: one\n  t: two\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for ring-number:  n\n"},{"name":"stdout","text":"\nRing type:\n  c: cobwebby\n  e: evanescent\n  f: flaring\n  l: large\n  n: none\n  p: pendant\n  s: sheathing\n  z: zone\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for ring-type:  c\n"},{"name":"stdout","text":"\nSpore print color:\n  k: black\n  n: brown\n  b: buff\n  h: chocolate\n  r: green\n  o: orange\n  u: purple\n  w: white\n  y: yellow\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for spore-print-color:  k\n"},{"name":"stdout","text":"\nPopulation:\n  a: abundant\n  c: clustered\n  n: numerous\n  s: scattered\n  v: several\n  y: solitary\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for population:  a\n"},{"name":"stdout","text":"\nHabitat:\n  g: grasses\n  l: leaves\n  m: meadows\n  p: paths\n  u: urban\n  w: waste\n  d: woods\n","output_type":"stream"},{"output_type":"stream","name":"stdin","text":"Enter the corresponding letter for habitat:  g\n"},{"name":"stdout","text":"\nThe mushroom is likely: Edible\n","output_type":"stream"}],"execution_count":5},{"cell_type":"markdown","source":"You can also just run the code cell below to use predefined inputs and run predictions on them. The second set of input features, are from an **Amanita Phalloides** mushroom, also known as a **Death Cap** mushroom, a very poisonous and toxic mushroom.","metadata":{}},{"cell_type":"code","source":"edible_mushroom = {\n    'cap-shape': 'x',  # convex\n    'cap-surface': 's',  # smooth\n    'cap-color': 'n',  # brown\n    'bruises': 't',  # bruises\n    'odor': 'a',  # almond\n    'gill-attachment': 'a',  # attached\n    'gill-spacing': 'c',  # close\n    'gill-size': 'b',  # broad\n    'gill-color': 'k',  # black\n    'stalk-shape': 'e',  # enlarging\n    'stalk-root': 'b',  # bulbous\n    'stalk-surface-above-ring': 'f',  # fibrous\n    'stalk-surface-below-ring': 'f',  # fibrous\n    'stalk-color-above-ring': 'n',  # brown\n    'stalk-color-below-ring': 'n',  # brown\n    'veil-type': 'p',  # partial\n    'veil-color': 'n',  # brown\n    'ring-number': 'o',  # one\n    'ring-type': 'c',  # cobwebby\n    'spore-print-color': 'k',  # black\n    'population': 'a',  # abundant\n    'habitat': 'g'  # grasses\n}\n\namanita_phalloides_features = {\n    'cap-shape': 'b',  # bell\n    'cap-surface': 's',  # smooth\n    'cap-color': 'w',  # white\n    'bruises': 'f',  # no\n    'odor': 'n',  # none\n    'gill-attachment': 'f',  # free\n    'gill-spacing': 'w',  # crowded\n    'gill-size': 'n',  # narrow\n    'gill-color': 'w',  # white\n    'stalk-shape': 't',  # tapering\n    'stalk-root': 'e',  # equal\n    'stalk-surface-above-ring': 's',  # smooth\n    'stalk-surface-below-ring': 's',  # smooth\n    'stalk-color-above-ring': 'w',  # white\n    'stalk-color-below-ring': 'w',  # white\n    'veil-type': 'p',  # partial\n    'veil-color': 'w',  # white\n    'ring-number': 'o',  # one\n    'ring-type': 'p',  # pendant\n    'spore-print-color': 'w',  # white\n    'population': 's',  # scattered\n    'habitat': 'd'  # woods\n}\n\n\nprint(f\"This mushroom is {predict_mushroom(edible_mushroom)}\")\nprint(f\"This mushroom is {predict_mushroom(amanita_phalloides_features)}\")","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-02-05T08:41:51.371876Z","iopub.execute_input":"2025-02-05T08:41:51.372221Z","iopub.status.idle":"2025-02-05T08:41:51.462680Z","shell.execute_reply.started":"2025-02-05T08:41:51.372194Z","shell.execute_reply":"2025-02-05T08:41:51.461690Z"}},"outputs":[{"name":"stdout","text":"This mushroom is Edible\nThis mushroom is Poisonous\n","output_type":"stream"}],"execution_count":7},{"cell_type":"markdown","source":"### The End:\n\nThis is the end of this project notebook, make sure to experiment and contribute to help improve the model and implementation. You can browse more of the open-source free projects on our GitHub repository: https://github.com/Infinitode/OPEN-ARC. If you like this project, make sure to star the repo and contribute your implementation, or help others in the community.\n\n~ Infinitode","metadata":{}}]}