Dv04 / Vision_Transformer

A project based on Vision Transformer for Image Regression
MIT License
14 stars 3 forks source link

Data Splitting Mechanism Implemented #6

Closed darshbaxi closed 9 months ago

darshbaxi commented 9 months ago

Issue #2 fix

Changes Made I have implemented a data splitting mechanism with a new 'splitter' function, which allows for flexible dataset splitting. This function is designed to:

  1. Split the entire dataset into a training set and a test set.
  2. Further split the training set into a validation set.

Details of the splitter function The splitter function takes two arguments: -> Train size: The desired size (proportion) of the training set. -> Validation size: The desired size (proportion) of the validation set.

Here's how the splitter function works:

It first splits the entire dataset into a training set and a test set based on the provided train size. Then, it further splits the training set into a validation set based on the provided validation size. The remaining data (if any) after forming the training and validation sets is automatically assigned to the test set.